Spaces:
Running
Running
| # Boot: fetch the Qdrant snapshot from the HF dataset, start Qdrant, recover the collection, | |
| # then serve the API + SPA on 7860. Free HF Spaces have no persistent disk, so this runs on every | |
| # cold start (~a few minutes to download the ~3.7GB snapshot + load models). | |
| set -euo pipefail | |
| SNAP_DIR="${QDRANT__STORAGE__SNAPSHOTS_PATH:-/home/user/app/qdrant_data/snapshots}" | |
| STORE_DIR="${QDRANT__STORAGE__STORAGE_PATH:-/home/user/app/qdrant_data/storage}" | |
| mkdir -p "$SNAP_DIR" "$STORE_DIR" | |
| status_of() { # -> collection status string ("" if absent) | |
| curl -s localhost:6333/collections/reviewsearch 2>/dev/null \ | |
| | python -c "import sys,json;print(json.load(sys.stdin).get('result',{}).get('status',''))" 2>/dev/null || echo "" | |
| } | |
| echo "[start] fetching Qdrant snapshot from HF dataset..." | |
| python - <<PY | |
| import os, shutil | |
| from huggingface_hub import hf_hub_download | |
| dst = os.path.join("$SNAP_DIR", "reviewsearch.snapshot") | |
| if not os.path.exists(dst): | |
| p = hf_hub_download("yjoonjang/reviewsearch-index", "reviewsearch.snapshot", repo_type="dataset") | |
| shutil.copy(p, dst) | |
| print("[start] snapshot ready:", dst, os.path.getsize(dst), "bytes", flush=True) | |
| PY | |
| echo "[start] launching Qdrant..." | |
| ./qdrant > /home/user/app/qdrant.log 2>&1 & | |
| for _ in $(seq 1 60); do curl -sf localhost:6333/healthz >/dev/null 2>&1 && break; sleep 1; done | |
| echo "[start] Qdrant healthy" | |
| if [ "$(status_of)" != "green" ]; then | |
| echo "[start] recovering collection from snapshot..." | |
| curl -s -X PUT localhost:6333/collections/reviewsearch/snapshots/recover \ | |
| -H 'content-type: application/json' \ | |
| -d "{\"location\":\"file://$SNAP_DIR/reviewsearch.snapshot\"}" | |
| echo | |
| fi | |
| for _ in $(seq 1 300); do [ "$(status_of)" = "green" ] && break; sleep 1; done | |
| echo "[start] collection status: $(status_of)" | |
| echo "[start] launching uvicorn on :7860" | |
| exec uvicorn reviewsearch.serve.app:app --host 0.0.0.0 --port 7860 | |