#!/bin/bash
  #
  # Start Local Embedding Service
  #
  # This service exposes:
  # - POST /embed/text
  # - POST /embed/image
  #
  # Defaults are defined in `embeddings/config.py`
  #
  set -e
  
  cd "$(dirname "$0")/.."
  

  source ./activate.sh

  DEFAULT_EMBEDDING_SERVICE_HOST=$(python -c "from embeddings.config import CONFIG; print(CONFIG.HOST)")
  DEFAULT_EMBEDDING_SERVICE_PORT=$(python -c "from embeddings.config import CONFIG; print(CONFIG.PORT)")

  USE_CLIP_AS_SERVICE=$(python -c "from embeddings.config import CONFIG; print('1' if CONFIG.USE_CLIP_AS_SERVICE else '0')")

  
  EMBEDDING_SERVICE_HOST="${EMBEDDING_HOST:-${DEFAULT_EMBEDDING_SERVICE_HOST}}"
  EMBEDDING_SERVICE_PORT="${EMBEDDING_PORT:-${DEFAULT_EMBEDDING_SERVICE_PORT}}"

  if [[ "${USE_CLIP_AS_SERVICE}" == "1" ]]; then
    if ! python - <<'PY'
  try:
      import pkg_resources  # noqa: F401
  except Exception:
      raise SystemExit(1)
  PY
    then
      echo "ERROR: clip-as-service image embedding requires pkg_resources, but current venv is missing it." >&2
      echo "Fix:" >&2
      echo "  python -m pip install 'setuptools<82'" >&2
      echo "Then restart: ./scripts/start_embedding_service.sh" >&2
      exit 1
    fi
  fi
  

  echo "========================================"
  echo "Starting Local Embedding Service"
  echo "========================================"
  echo "Host: ${EMBEDDING_SERVICE_HOST}"
  echo "Port: ${EMBEDDING_SERVICE_PORT}"
  echo
  echo "Tips:"
  echo "  - Use a single worker (GPU models cannot be safely duplicated across workers)."
  echo "  - Clients can set EMBEDDING_SERVICE_URL=http://localhost:${EMBEDDING_SERVICE_PORT}"
  echo
  
  exec python -m uvicorn embeddings.server:app \
    --host "${EMBEDDING_SERVICE_HOST}" \
    --port "${EMBEDDING_SERVICE_PORT}" \
    --workers 1