Download FAISS index at runtime (robust HF setup)

.gitignore

@@ -19,3 +19,7 @@ models/*.gguf
 
 # OS
 .DS_Store
+
+#faiss et pkl
+db/faiss_code_edu_by_article/index.faiss
+db/faiss_code_edu_by_article/index.pkl

app.py

@@ -9,6 +9,34 @@ import sys
 import traceback
 import gradio as gr
 from huggingface_hub import hf_hub_download
+import os, shutil
+
+def ensure_faiss_index_present():
+    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")
+
+    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)."
+        )
+
+    token = os.environ.get("HF_TOKEN")
+
+    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)
+
+    shutil.copyfile(f_faiss, faiss_path)
+    shutil.copyfile(f_pkl, pkl_path)
+
+ensure_faiss_index_present()
+
 
 
 def ensure_model_present():