Upload rag_api.py with huggingface_hub

rag_api.py

@@ -36,6 +36,9 @@ TEMP_CACHE_DIR = "/tmp/huggingface_cache"
 os.environ.setdefault("TRANSFORMERS_CACHE", TEMP_CACHE_DIR)
 os.environ.setdefault("HF_HOME", TEMP_CACHE_DIR)
 os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", TEMP_CACHE_DIR)
+# Evita la ruta "Xet" del Hub, que a veces responde sin la cabecera
+# X-Repo-Commit y hace fallar la descarga del índice.
+os.environ.setdefault("HF_HUB_DISABLE_XET", "1")
 os.makedirs(TEMP_CACHE_DIR, exist_ok=True)
 
 # --------------------------------------------------------
@@ -149,20 +152,47 @@ def ensure_index() -> str:
     """Descarga index.faiss e index.pkl desde el repo del Space a /tmp.
 
     Los Spaces Docker no materializan los archivos del repo en disco (en especial
-    si están en Xet). Con `hf_hub_download` sí se descargan correctamente. Son
-    archivos pequeños (~1 MB), así que el arranque no se penaliza.
+    si están en Xet), por eso se descargan al arrancar.
+
+    Se intenta primero con `hf_hub_download` y, si el Hub responde sin la
+    cabecera `X-Repo-Commit` (típico cuando la petición pasa por un proxy),
+    se cae a una descarga HTTP directa, que no depende de esa cabecera.
     """
-    os.makedirs(INDEX_LOCAL_DIR, exist_ok=True)
-    from huggingface_hub import hf_hub_download
+    target_dir = os.path.join(INDEX_LOCAL_DIR, "db_faiss")
+    os.makedirs(target_dir, exist_ok=True)
 
     for fname in ["index.faiss", "index.pkl"]:
-        hf_hub_download(
-            repo_id=SPACE_REPO_ID,
-            filename=f"db_faiss/{fname}",
-            repo_type="space",
-            local_dir=INDEX_LOCAL_DIR,
+        dest = os.path.join(target_dir, fname)
+
+        # 1) Vía normal: huggingface_hub (usa HF_TOKEN si está disponible)
+        try:
+            from huggingface_hub import hf_hub_download
+
+            hf_hub_download(
+                repo_id=SPACE_REPO_ID,
+                filename=f"db_faiss/{fname}",
+                repo_type="space",
+                local_dir=INDEX_LOCAL_DIR,
+                token=os.environ.get("HF_TOKEN") or None,
+            )
+            print(f"✓ {fname} descargado con hf_hub_download")
+            continue
+        except Exception as e:  # noqa: BLE001
+            print(f"⚠ hf_hub_download falló para {fname}: {type(e).__name__}: {e}")
+
+        # 2) Fallback: descarga HTTP directa (no usa la cabecera X-Repo-Commit)
+        import urllib.request
+
+        url = (
+            f"https://huggingface.co/spaces/{SPACE_REPO_ID}"
+            f"/resolve/main/db_faiss/{fname}"
         )
-    return os.path.join(INDEX_LOCAL_DIR, "db_faiss")
+        print(f"↓ Descargando por HTTP directo: {url}")
+        with urllib.request.urlopen(url, timeout=60) as r, open(dest, "wb") as f:
+            f.write(r.read())
+        print(f"✓ {fname} descargado por HTTP directo")
+
+    return target_dir
 
 
 def load_rag():