#!/usr/bin/env bash
set -euo pipefail

ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "$ROOT"

unset VIRTUAL_ENV

PYTHON="${PYTHON:-3.12}"
VENV=".venv-api"

echo "==> Creating PaddleOCR-VL pipeline API env (${VENV}) with Python ${PYTHON}"
uv venv "${VENV}" --python "${PYTHON}"

echo "==> Installing PaddlePaddle GPU (CUDA 12.6)"
uv pip install --python "${VENV}" \
  paddlepaddle-gpu \
  -i https://www.paddlepaddle.org.cn/packages/stable/cu126/

echo "==> Installing PaddleOCR doc-parser stack"
uv pip install --python "${VENV}" "paddleocr[doc-parser]>=3.3.0"

echo "==> Installing PaddleX serving dependencies"
uv pip install --python "${VENV}" \
  "aiohttp>=3.9" "filetype>=1.2" "fastapi>=0.110" "starlette>=0.36" "uvicorn>=0.16"

echo "==> Done. Start pipeline API with: ./scripts/serve-pipeline.sh"