Update app.py

app.py

@@ -17,7 +17,7 @@ META_FILE = "navy_metadata.pkl"
 
 st.set_page_config(page_title="Navy Policy Architect", layout="wide", page_icon="⚓")
 
-# --- CLOUD SYNC MANAGER (ROBUST) ---
+# --- CLOUD SYNC MANAGER (FIXED) ---
 class SyncManager:
     """Handles downloading/uploading the Database & Index to Hugging Face"""
     
@@ -26,6 +26,7 @@ class SyncManager:
         if not HF_TOKEN: return []
         try:
             api = HfApi(token=HF_TOKEN)
+            # This worked because we specified repo_type="dataset" here
             files = api.list_repo_files(repo_id=DATASET_REPO_ID, repo_type="dataset")
             dbs = [f for f in files if f.endswith(".db")]
             return dbs
@@ -38,20 +39,32 @@ class SyncManager:
             st.error("HF_TOKEN missing.")
             return False
         try:
-            # FIX: Remove "if os.path.exists" check. 
-            # We ALWAYS let hf_hub_download check for updates/integrity.
+            # FIX: Added repo_type="dataset"
             hf_hub_download(
                 repo_id=DATASET_REPO_ID, 
                 filename=db_filename, 
                 local_dir=".", 
                 token=HF_TOKEN,
-                force_download=False # It will only download if the cloud version is newer
+                repo_type="dataset", # <--- THE MISSING LINK
+                force_download=False 
             )
             
             # Download Index (Best effort)
             try:
-                hf_hub_download(repo_id=DATASET_REPO_ID, filename=INDEX_FILE, local_dir=".", token=HF_TOKEN)
-                hf_hub_download(repo_id=DATASET_REPO_ID, filename=META_FILE, local_dir=".", token=HF_TOKEN)
+                hf_hub_download(
+                    repo_id=DATASET_REPO_ID, 
+                    filename=INDEX_FILE, 
+                    local_dir=".", 
+                    token=HF_TOKEN,
+                    repo_type="dataset" # <--- Added here too
+                )
+                hf_hub_download(
+                    repo_id=DATASET_REPO_ID, 
+                    filename=META_FILE, 
+                    local_dir=".", 
+                    token=HF_TOKEN,
+                    repo_type="dataset" # <--- And here
+                )
             except:
                 pass 
             
@@ -65,12 +78,15 @@ class SyncManager:
         if not HF_TOKEN: return
         api = HfApi(token=HF_TOKEN)
         try:
+            # This was already working because we had repo_type="dataset"
             api.upload_file(path_or_fileobj=db_filename, path_in_repo=db_filename, repo_id=DATASET_REPO_ID, repo_type="dataset")
             api.upload_file(path_or_fileobj=INDEX_FILE, path_in_repo=INDEX_FILE, repo_id=DATASET_REPO_ID, repo_type="dataset")
             api.upload_file(path_or_fileobj=META_FILE, path_in_repo=META_FILE, repo_id=DATASET_REPO_ID, repo_type="dataset")
             st.toast("Cloud Sync Complete!", icon="☁️")
         except Exception as e:
             st.error(f"Sync Error (Push): {e}")
+
+            
 # --- SIDEBAR: KNOWLEDGE BASE SELECTOR ---
 with st.sidebar:
     st.header("🗄️ Knowledge Base")