Update utils/database.py

utils/database.py

@@ -175,6 +175,42 @@ def create_tables(conn: sqlite3.Connection) -> None:
     except Exception as e:
         st.error(f"Unexpected error in create_tables: {e}")
         raise
+
+def create_chat_tables(conn):
+    """Create necessary tables for chat management."""
+    try:
+        cursor = conn.cursor()
+        
+        # Create chats table
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS chats (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                title TEXT NOT NULL,
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                last_updated TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                collection_id INTEGER,
+                FOREIGN KEY (collection_id) REFERENCES collections (id) ON DELETE SET NULL
+            )
+        ''')
+        
+        # Create chat messages table
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS chat_messages (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                chat_id INTEGER NOT NULL,
+                role TEXT NOT NULL,
+                content TEXT NOT NULL,
+                timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                metadata TEXT,  -- Store metadata as JSON string
+                FOREIGN KEY (chat_id) REFERENCES chats (id) ON DELETE CASCADE
+            )
+        ''')
+        
+        conn.commit()
+        
+    except sqlite3.Error as e:
+        st.error(f"Error creating chat tables: {e}")
+        raise
 def search_documents_in_collection(conn: sqlite3.Connection, collection_id: int, query: str) -> List[Dict]:
     """Search for documents within a collection."""
     try: