Update main.py

main.py

@@ -3,10 +3,9 @@
 FastAPI + Gradio : service d’indexation asynchrone avec FAISS.
 Ce fichier a été corrigé pour :
 
-* importer correctement `JobState`
-* garantir que toutes les dépendances (typing, pathlib…) sont disponibles
-* exposer les routes attendues par le client
-* garder la même logique que la version originale.
+* importer correctement `JobState` (import relatif)
+* garantir que le répertoire `app` est dans le PYTHONPATH lorsqu’on lance le script
+* conserver toutes les fonctionnalités précédentes (indexation, recherche, UI)
 """
 
 from __future__ import annotations
@@ -18,6 +17,7 @@ import time
 import hashlib
 import logging
 import tarfile
+import sys
 from pathlib import Path
 from typing import List, Dict, Any, Tuple, Optional
 
@@ -32,6 +32,16 @@ from pydantic import BaseModel
 
 import gradio as gr
 
+# --------------------------------------------------------------------------- #
+#   RÉGLAGE DU PYTHONPATH (pour que les imports relatifs fonctionnent)
+# --------------------------------------------------------------------------- #
+# Si le script est lancé depuis le répertoire `app/`, le package `app` n’est pas
+# découvert automatiquement. On ajoute le répertoire parent au sys.path.
+CURRENT_DIR = Path(__file__).resolve().parent
+PROJECT_ROOT = CURRENT_DIR.parent
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
 # --------------------------------------------------------------------------- #
 #   LOGGING
 # --------------------------------------------------------------------------- #
@@ -100,11 +110,11 @@ def _setup_cache_dirs() -> Dict[str, str]:
 CACHE_PATHS = _setup_cache_dirs()
 
 # --------------------------------------------------------------------------- #
-#   IMPORT DE LA CLASSE DE STATE (c’est ce qui manquait)
+#   IMPORT DE LA CLASSE DE STATE (corrigé : import relatif)
 # --------------------------------------------------------------------------- #
-# La classe `JobState` se trouve dans `app/core/index_state.py`.
-# On l’importe ici afin qu’elle soit disponible dans tout le module.
-from app.core.index_state import JobState  # <-- IMPORT CORRIGÉ
+# Le fichier `index_state.py` se trouve dans `app/core/`.
+# En étant dans le répertoire `app`, on peut l’importer via le package `core`.
+from core.index_state import JobState   # <-- IMPORT CORRIGÉ
 
 # --------------------------------------------------------------------------- #
 #   GLOBALS
@@ -255,9 +265,7 @@ def _save_faiss(fx_dir: str, xb: np.ndarray, meta: Dict[str, Any]) -> None:
     os.makedirs(fx_dir, exist_ok=True)
     idx_path = os.path.join(fx_dir, "emb.faiss")
 
-    # ------------------------------------------------------------------- #
-    #   Index quantisé (IVF‑PQ) – optimisation mémoire / disque
-    # ------------------------------------------------------------------- #
+    # ------------------- INDEX QUANTISÉ (IVF‑PQ) ------------------- #
     quantizer = faiss.IndexFlatIP(xb.shape[1])          # inner‑product (cosine si normalisé)
     index = faiss.IndexIVFPQ(quantizer, xb.shape[1], 100, 8, 8)  # nlist=100, m=8, nbits=8
 
@@ -338,7 +346,7 @@ def _do_index_job(
         else:
             xb = _emb_hf(texts)
 
-        # ------------------- 3️⃣ Réduction de dimension (PCA) -------------------
+        # ------------------- 3️⃣ Réduction PCA (si besoin) -------------------
         if xb.shape[1] != EMB_DIM:
             from sklearn.decomposition import PCA
             pca = PCA(n_components=EMB_DIM, random_state=0)