Update utils/database.py

utils/database.py

@@ -95,6 +95,31 @@ def get_documents(conn):
     except Error as e:
         st.error(f"Error retrieving documents: {e}")
         return [], []
+
+        def insert_document(conn, name, content):
+    """Insert a new document into the database.
+    
+    Args:
+        conn: SQLite database connection
+        name (str): Name of the document
+        content (str): Content of the document
+        
+    Returns:
+        int: ID of the inserted document, or None if insertion failed
+    """
+    try:
+        cursor = conn.cursor()
+        sql = '''INSERT INTO documents (name, content)
+                VALUES (?, ?)'''
+        cursor.execute(sql, (name, content))
+        conn.commit()
+        return cursor.lastrowid
+        
+    except Error as e:
+        st.error(f"Error inserting document: {e}")
+        return None
+
+        
 # QA system initialization function with necessary updates
 def initialize_qa_system(vector_store):
     """Initialize QA system with proper chat handling."""