Fix: download FAISS from dataset repo (repo_type=dataset)

app.py

@@ -10,29 +10,15 @@ import traceback
 import gradio as gr
 from huggingface_hub import hf_hub_download
 import os, shutil
-
+from pathlib import Path
 
 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 HF qui contient l'index (DATASET, pas model, pas space)
-    repo_id = os.getenv("FAISS_REPO_ID", "FabIndy/code-education-faiss-index")
+    repo_id = os.environ.get("FAISS_REPO_ID", "FabIndy/code-education-faiss-index")
+    token = os.environ.get("HF_TOKEN")  # optionnel si dataset public
 
-    # Token optionnel (uniquement si repo privé/gated)
-    token = os.getenv("HF_TOKEN")
-    if not token or not token.startswith("hf_"):
-        token = None
+    local_dir = Path("db/faiss_code_edu_by_article")
+    local_dir.mkdir(parents=True, exist_ok=True)
 
-    # Téléchargement depuis un dataset repo
     f_faiss = hf_hub_download(
         repo_id=repo_id,
         repo_type="dataset",
@@ -46,9 +32,9 @@ def ensure_faiss_index_present():
         token=token,
     )
 
-    # Copie vers le chemin local attendu
-    shutil.copyfile(f_faiss, faiss_path)
-    shutil.copyfile(f_pkl, pkl_path)
+    shutil.copyfile(f_faiss, local_dir / "index.faiss")
+    shutil.copyfile(f_pkl, local_dir / "index.pkl")
+