Update download_assets.py

download_assets.py

@@ -3,13 +3,16 @@ import os
 
 def download_assets():
     """Download necessary assets from Hugging Face Hub."""
+    # ensure local dirs exist
     os.makedirs("data", exist_ok=True)
     os.makedirs("vectorstore/db_faiss", exist_ok=True)
 
     repo_id   = "MoizK/mindmedic-assets"
     repo_type = "dataset"
-    token     = os.getenv("HUGGINGFACE_API_TOKEN")  # optional
+    # allow either env var name
+    token     = os.getenv("HUGGINGFACE_HUB_TOKEN") or os.getenv("HUGGINGFACE_API_TOKEN")
 
+    # list of PDF filenames *at the repo root*
     pdf_files = [
         "71763-gale-encyclopedia-of-medicine.-vol.-1.-2nd-ed.pdf",
         "Depression-NIM-2024.pdf",
@@ -22,31 +25,31 @@ def download_assets():
 
     for fname in pdf_files:
         try:
-            hf_hub_download(
+            path = hf_hub_download(
                 repo_id=repo_id,
                 repo_type=repo_type,
-                filename=f"data/{fname}",
-                local_dir=".",
-                local_dir_use_symlinks=False,
+                filename=fname,            # no "data/" prefix
+                local_dir="data",          # drop into ./data/
                 token=token,
             )
-            print(f"Downloaded {fname}")
+            print(f"✅ Downloaded {fname} → {path}")
         except Exception as e:
             print(f"⚠️ Failed to download {fname}: {e}")
 
-    for idx in ("index.faiss", "index.pkl"):
+    # FAISS index files also at the repo root
+    index_files = ["index.faiss", "index.pkl"]
+    for fname in index_files:
         try:
-            hf_hub_download(
+            path = hf_hub_download(
                 repo_id=repo_id,
                 repo_type=repo_type,
-                filename=f"vectorstore/db_faiss/{idx}",
-                local_dir=".",
-                local_dir_use_symlinks=False,
+                filename=fname,                   # root filename
+                local_dir="vectorstore/db_faiss", # into ./vectorstore/db_faiss/
                 token=token,
             )
-            print(f"Downloaded {idx}")
+            print(f"✅ Downloaded {fname} → {path}")
         except Exception as e:
-            print(f"⚠️ Failed to download {idx}: {e}")
+            print(f"⚠️ Failed to download {fname}: {e}")
 
 if __name__ == "__main__":
     download_assets()