Fix FAISS index download from HF dataset at startup

app.py

@@ -11,30 +11,47 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 import os, shutil
 
+
 def ensure_faiss_index_present():
+    # Dossier attendu par le projet (FAISS local)
     base_dir = "db/faiss_code_edu_by_article"
     os.makedirs(base_dir, exist_ok=True)
 
     faiss_path = os.path.join(base_dir, "index.faiss")
     pkl_path   = os.path.join(base_dir, "index.pkl")
 
+    # Déjà présent -> on ne retélécharge pas
     if os.path.exists(faiss_path) and os.path.exists(pkl_path):
         return
 
-    repo_id = os.environ.get("INDEX_REPO_ID")
-    if not repo_id:
-        raise RuntimeError(
-            "INDEX_REPO_ID non défini (repo HF contenant l’index FAISS)."
-        )
+    # Repo HF qui contient l'index (DATASET, pas model, pas space)
+    repo_id = os.getenv("FAISS_REPO_ID", "FabIndy/code-education-faiss-index")
 
-    token = os.environ.get("HF_TOKEN")
+    # Token optionnel (uniquement si repo privé/gated)
+    token = os.getenv("HF_TOKEN")
+    if not token or not token.startswith("hf_"):
+        token = None
 
-    f_faiss = hf_hub_download(repo_id=repo_id, filename="index.faiss", token=token)
-    f_pkl   = hf_hub_download(repo_id=repo_id, filename="index.pkl", token=token)
+    # Téléchargement depuis un dataset repo
+    f_faiss = hf_hub_download(
+        repo_id=repo_id,
+        repo_type="dataset",
+        filename="index.faiss",
+        token=token,
+    )
+    f_pkl = hf_hub_download(
+        repo_id=repo_id,
+        repo_type="dataset",
+        filename="index.pkl",
+        token=token,
+    )
 
+    # Copie vers le chemin local attendu
     shutil.copyfile(f_faiss, faiss_path)
     shutil.copyfile(f_pkl, pkl_path)
 
+
+
 ensure_faiss_index_present()