Update utils/database.py

utils/database.py

@@ -67,7 +67,34 @@ def create_tables(conn):
         conn.execute(sql_create_annotations_table)
     except Error as e:
         st.error(f"Error: {e}")
-
+# Add this function to your database.py file
+
+def get_documents(conn):
+    """Retrieve all documents from the database.
+    
+    Args:
+        conn: SQLite database connection
+        
+    Returns:
+        tuple: (list of document contents, list of document names)
+    """
+    try:
+        cursor = conn.cursor()
+        cursor.execute("SELECT content, name FROM documents")
+        results = cursor.fetchall()
+        
+        if not results:
+            return [], []
+            
+        # Separate contents and names
+        document_contents = [row[0] for row in results]
+        document_names = [row[1] for row in results]
+        
+        return document_contents, document_names
+        
+    except Error as e:
+        st.error(f"Error retrieving documents: {e}")
+        return [], []
 # QA system initialization function with necessary updates
 def initialize_qa_system(vector_store):
     """Initialize QA system with proper chat handling."""