Update app.py

app.py

@@ -114,25 +114,32 @@ with st.sidebar:
     # If the DB has changed or isn't loaded, load it now
     if 'current_db_name' not in st.session_state or st.session_state.current_db_name != selected_db:
         
-        # We use an empty container to hold the error if it happens
-        error_container = st.empty()
+        # We use an empty container to hold messages
+        msg_container = st.empty()
         
-        with st.spinner(f"Loading {selected_db} from Cloud..."):
+        with st.spinner(f"Syncing {selected_db}..."):
             
             # 1. Attempt the Pull
             result = SyncManager.pull_data(selected_db)
             
             # 2. Check the Result
             if result is True:
-                # Success! Initialize normally
+                # Success! Cloud file found.
+                msg_container.success(f"Loaded {selected_db} from Cloud.")
+            else:
+                # Failure! (File deleted or new setup)
+                # INSTEAD OF STOPPING, we warn and create a fresh local DB.
+                msg_container.warning(f"Could not find {selected_db} in cloud. Creating new local database.")
+                # We do NOT run st.stop() here anymore.
+
+            # 3. Initialize the Database Manager (Either with the downloaded file or a new blank one)
+            try:
                 st.session_state.db = DatabaseManager(selected_db)
                 st.session_state.search_engine = SearchEngine()
                 st.session_state.current_db_name = selected_db
-                st.rerun()
-            else:
-                # Failure! Show the error and STOP.
-                # Do NOT create an empty database. Do NOT rerun.
-                error_container.error(f"CRITICAL ERROR: Could not load database.\n\nDetails: {result}")
+                # We intentionally do NOT rerun immediately here to let the warning show
+            except Exception as e:
+                st.error(f"Failed to initialize database: {e}")
                 st.stop()
 
     # 3. Upload Section