Update app.py

app.py

@@ -239,29 +239,30 @@ def create_indexed_database():
             log_progress(f"Download failed: {e}", "ERROR")
             log_progress("Will create locally", "INFO")
     
-    # Check for partial progress
-    if completed_indices or analyzed_tables:
-        log_progress("Resuming from checkpoint:", "INFO")
+    # NEW: If all indices complete OR partial work done, try to download indexed DB
+    if completed_indices or analyzed_tables or database_uploaded:
+        log_progress("Checkpoint detected - attempting to download indexed database...", "INFO")
         log_progress(f"  Completed indices: {sorted(completed_indices)}", "INFO")
         log_progress(f"  Analyzed tables: {sorted(analyzed_tables)}", "INFO")
-    
-    # Get or create local database
-    if os.path.exists(LOCAL_DB_PATH) and (completed_indices or analyzed_tables):
-        log_progress("Using existing local database", "SUCCESS")
-    elif database_uploaded:
-        log_progress("Downloading partial database from HF...", "INFO")
+        
         try:
-            remote_db = hf_hub_download(
+            indexed_path = hf_hub_download(
                 repo_id=INDEXED_REPO_ID,
                 filename=INDEXED_DB_FILENAME,
                 repo_type="dataset",
                 token=HF_TOKEN
             )
-            shutil.copy2(remote_db, LOCAL_DB_PATH)
-            log_progress("Downloaded partial database", "SUCCESS")
+            
+            # Copy to local path
+            log_progress(f"Downloaded indexed DB, copying to {LOCAL_DB_PATH}...", "INFO")
+            shutil.copy2(indexed_path, LOCAL_DB_PATH)
+            log_progress("Using downloaded indexed database", "SUCCESS")
+            
         except Exception as e:
-            log_progress(f"No partial database: {e}", "INFO")
+            log_progress(f"Could not download indexed DB: {e}", "WARN")
+            log_progress("Will need to start from original and create indices", "INFO")
     
+    # Only download original if we don't have local DB yet
     if not os.path.exists(LOCAL_DB_PATH):
         # Download and copy original
         log_progress("Downloading original database...", "INFO")
@@ -285,6 +286,11 @@ def create_indexed_database():
         shutil.copy2(original_path, LOCAL_DB_PATH)
         elapsed = time.time() - start
         log_progress(f"Copied in {elapsed:.1f}s", "SUCCESS")
+        
+        # Clear completed indices since we're starting from scratch
+        log_progress("Starting from original DB - clearing completed indices list", "WARN")
+        completed_indices = set()
+        analyzed_tables = set()
     
     # Connect to database
     conn = sqlite3.connect(LOCAL_DB_PATH)