Update main.py

main.py

@@ -3,24 +3,25 @@
 HF Space - main.py de substitution pour tests Qdrant / indexation minimale
 
 Endpoints:
-- POST /wipe?project_id=XXX
-- POST /index
-- GET  /status/{job_id}
-- GET  /collections/{project_id}/count
-- POST /query
-- GET  /health        <-- healthcheck OK
-
-UI Gradio montée sur "/" pour tests sans console.
+- GET  /                      → redirige vers UI_PATH (défaut: /ui)
+- GET  /ui (UI_PATH)          → UI Gradio
+- GET  /health                → healthcheck
+- GET  /api                   → infos service
+- POST /wipe?project_id=XXX   → supprime la collection Qdrant
+- POST /index                 → lance un job d'indexation
+- GET  /status/{job_id}       → état + logs du job
+- GET  /collections/{proj}/count → count points dans Qdrant
+- POST /query                 → recherche sémantique
 
 ENV attendues :
-- QDRANT_URL            : https://...qdrant.io:6333
-- QDRANT_API_KEY        : clé Qdrant
-- COLLECTION_PREFIX     : "proj_" par défaut
-- EMB_PROVIDER          : "hf" (défaut) ou "dummy"
-- HF_EMBED_MODEL        : "BAAI/bge-m3" par défaut
+- QDRANT_URL, QDRANT_API_KEY (requis pour upsert)
+- COLLECTION_PREFIX (défaut "proj_")
+- EMB_PROVIDER ("hf" par défaut, "dummy" sinon)
+- HF_EMBED_MODEL (défaut "BAAI/bge-m3")
 - HUGGINGFACEHUB_API_TOKEN (si EMB_PROVIDER=hf)
-- LOG_LEVEL             : DEBUG (défaut)
-- PORT                  : 7860 (fourni par HF)
+- LOG_LEVEL (défaut DEBUG)
+- PORT (fourni par HF, défaut 7860)
+- UI_PATH (défaut "/ui")
 
 Dépendances suggérées :
 fastapi>=0.111, uvicorn>=0.30, httpx>=0.27, pydantic>=2.7, gradio>=4.43, numpy>=2.0
@@ -41,6 +42,7 @@ import uvicorn
 from pydantic import BaseModel, Field, ValidationError
 from fastapi import FastAPI, HTTPException, Query
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import RedirectResponse, JSONResponse
 import gradio as gr
 
 # ------------------------------------------------------------------------------
@@ -61,6 +63,8 @@ EMB_PROVIDER = os.getenv("EMB_PROVIDER", "hf").lower()  # "hf" | "dummy"
 HF_EMBED_MODEL = os.getenv("HF_EMBED_MODEL", "BAAI/bge-m3")
 HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN", "")
 
+UI_PATH = os.getenv("UI_PATH", "/ui")  # UI montée ici par défaut
+
 if not QDRANT_URL or not QDRANT_API_KEY:
     LOG.warning("QDRANT_URL / QDRANT_API_KEY non fournis : l'upsert échouera.")
 
@@ -332,9 +336,14 @@ fastapi_app.add_middleware(
 async def health():
     return {"status": "ok"}
 
+@fastapi_app.get("/api")
+async def api_info():
+    return {"ok": True, "service": "remote-indexer-min", "qdrant": bool(QDRANT_URL), "emb_provider": EMB_PROVIDER, "ui_path": UI_PATH}
+
+# Redirige "/" → UI_PATH (ex.: /ui). Ça évite tout conflit avec la route racine.
 @fastapi_app.get("/")
-async def root():
-    return {"ok": True, "service": "remote-indexer-min", "qdrant": bool(QDRANT_URL), "emb_provider": EMB_PROVIDER}
+async def root_redirect():
+    return RedirectResponse(url=UI_PATH, status_code=307)
 
 @fastapi_app.post("/wipe")
 async def wipe(project_id: str = Query(..., min_length=1)):
@@ -491,11 +500,10 @@ with gr.Blocks(title="Remote Indexer - Minimal Test", analytics_enabled=False) a
     count_btn.click(ui_count, inputs=[project_tb], outputs=[out_log])
     query_btn.click(ui_query, inputs=[project_tb, query_tb, topk], outputs=[query_out])
 
-# Monte l'UI Gradio sur la FastAPI
-app = gr.mount_gradio_app(fastapi_app, ui, path="/")
+# Monte l'UI Gradio sur la FastAPI au chemin UI_PATH
+app = gr.mount_gradio_app(fastapi_app, ui, path=UI_PATH)
 
 if __name__ == "__main__":
-    # Démarre Uvicorn pour les Spaces Docker (CMD: python -u /app/main.py)
     port = int(os.getenv("PORT", "7860"))
-    LOG.info(f"Démarrage Uvicorn sur 0.0.0.0:{port}")
+    LOG.info(f"Démarrage Uvicorn sur 0.0.0.0:{port} (UI_PATH={UI_PATH})")
     uvicorn.run(app, host="0.0.0.0", port=port)