Mount Gradio under /ui for HF Spaces, API now publicly accessible

main.py

@@ -433,3 +433,26 @@ if __name__ == "__main__":
     import uvicorn
 
     uvicorn.run(app, host="0.0.0.0", port=8000)
+
+
+# ============== MOUNT GRADIO FOR HF SPACES ==============
+def mount_gradio_ui():
+    """Mount Gradio UI under /ui path for HF Spaces deployment"""
+    try:
+        import gradio as gr
+        # Lazy import to avoid circular dependency
+        import importlib.util
+        spec = importlib.util.spec_from_file_location("app_gradio", "app_gradio.py")
+        app_gradio = importlib.util.module_from_spec(spec)
+        spec.loader.exec_module(app_gradio)
+        
+        # Mount Gradio under /ui
+        app.mount("/ui", gr.mount_gradio_app(app, app_gradio.blocks, path="/"))
+        print("✅ Gradio UI mounted at /ui")
+    except Exception as e:
+        print(f"⚠️  Could not mount Gradio: {e}")
+
+
+# Auto-mount when running in HF Spaces (PORT env var set)
+if os.getenv("PORT"):
+    mount_gradio_ui()

start.sh

@@ -3,23 +3,17 @@ set -euo pipefail
 
 cd "$(dirname "$0")"
 
-API_HOST="${API_HOST:-127.0.0.1}"
-API_PORT="${API_PORT:-8000}"
-UI_HOST="${UI_HOST:-0.0.0.0}"
-UI_PORT="${PORT:-7860}"
+API_HOST="${API_HOST:-0.0.0.0}"
+API_PORT="${PORT:-7860}"
 
 : "${DATABASE_URL:?DATABASE_URL is not set. Add it as a Hugging Face Space Secret (Settings → Repository secrets).}"
 
-echo "Starting FastAPI on http://${API_HOST}:${API_PORT} ..."
-uv run uvicorn main:app --host "${API_HOST}" --port "${API_PORT}" &
-API_PID=$!
+echo "Starting FastAPI + Gradio (mounted at /ui) on http://${API_HOST}:${API_PORT} ..."
+echo "UI: http://localhost:${API_PORT}/ui"
+echo "API Docs: http://localhost:${API_PORT}/docs"
 
-cleanup() {
-  echo "Stopping FastAPI..."
-  kill "${API_PID}" 2>/dev/null || true
-}
-trap cleanup EXIT
+# Internal API URL for Gradio to use
+export KEYVAULT_BASE_URL="http://127.0.0.1:8000"
 
-echo "Starting Gradio UI on http://${UI_HOST}:${UI_PORT} (API -> http://${API_HOST}:${API_PORT}) ..."
-export KEYVAULT_BASE_URL="http://${API_HOST}:${API_PORT}"
-uv run python app_gradio.py
+# Run on PORT (7860 for HF Spaces) - FastAPI will mount Gradio at /ui
+uv run uvicorn main:app --host "${API_HOST}" --port "${API_PORT}"