Update app.py

app.py

@@ -8,7 +8,7 @@ from datetime import datetime
 from threading import Lock
 
 from utils.database import (
-    # Base database operations
+   # Base database operations
     create_connection,
     create_tables,
     get_all_documents,
@@ -20,6 +20,7 @@ from utils.database import (
     remove_from_collection,
     update_collection,
     get_collection_documents,
+    handle_document_upload,
     
     # Search functionality
     search_documents
@@ -30,6 +31,28 @@ from components.chat import display_chat_interface
 # Create locks for thread-safe operations
 conn_lock = Lock()
 
+def verify_database_tables(conn):
+    """Verify that all required tables exist."""
+    try:
+        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])
+        
+        # If collections table doesn't exist, recreate tables
+        if 'collections' not in [table[0] for table in tables]:
+            st.warning("Collections table not found. Recreating tables...")
+            create_tables(conn)
+            
+            # 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])
+            
+    except Exception as e:
+        st.error(f"Error verifying tables: {e}")
+
 def initialize_database():
     """Initialize database with persistent storage."""
     try:
@@ -59,10 +82,15 @@ def initialize_database():
                 # Create all tables (includes collection tables)
                 create_tables(conn)
                 st.session_state.db_conn = conn
+                
+                # Verify tables were created
+                verify_database_tables(conn)
                 return True
             else:
                 return False
         else:
+            # Verify tables exist in existing connection
+            verify_database_tables(st.session_state.db_conn)
             return True
 
     except Exception as e: