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

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

# Avoid picking up another project's active venv
unset VIRTUAL_ENV

PYTHON="${PYTHON:-3.12}"
FLASH_ATTN_WHEEL="${FLASH_ATTN_WHEEL:-}"

echo "==> Using uv with Python ${PYTHON}"
uv python pin "${PYTHON}"
uv sync

if [[ -n "${FLASH_ATTN_WHEEL}" ]]; then
  echo "==> Installing prebuilt flash-attn wheel"
  uv pip install --python .venv "${FLASH_ATTN_WHEEL}"
else
  echo "==> Installing flash-attn (may require a prebuilt wheel; set FLASH_ATTN_WHEEL)"
  uv pip install --python .venv "flash-attn==2.8.2"
fi

echo "==> Verifying paddleocr genai_server entrypoint"
uv run paddleocr genai_server --help >/dev/null

echo "==> Done. Start the service with: ./scripts/serve.sh"