Update app.py

app.py

@@ -47,24 +47,29 @@ def verify_database_tables(conn):
         cursor = conn.cursor()
         # Get list of all tables
         cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
-        tables = cursor.fetchall()
-        st.write("Existing tables:", [table[0] for table in tables])
+        tables = [table[0] for table in cursor.fetchall()]
+        
+        # Check for required tables
+        required_tables = {
+            'documents', 'queries', 'annotations', 
+            'collections', 'document_collections', 'chats', 'chat_messages'
+        }
+        
+        missing_tables = required_tables - set(tables)
         
         # If collections table doesn't exist, force recreate it
-        if 'collections' not in [table[0] for table in tables]:
-            st.warning("Collections table not found. Force recreating collections tables...")
+        if 'collections' not in tables:
             if force_recreate_collections_tables(conn):
-                st.success("Collections tables recreated successfully!")
+                st.session_state.show_debug = True
             else:
-                st.error("Failed to recreate collections tables!")
-            
-            # Verify again
-            cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
-            tables = cursor.fetchall()
-            st.write("Tables after recreation:", [table[0] for table in tables])
+                st.error("Failed to create required database tables.")
+                return False
+                
+        return True
             
     except Exception as e:
-        st.error(f"Error verifying tables: {e}")
+        st.error(f"Error verifying database tables: {e}")
+        return False
 
 def initialize_database():
     """Initialize database with persistent storage."""