Update utils/database.py

utils/database.py

@@ -35,6 +35,27 @@ def create_connection(db_file):
         st.error("Failed to connect to database. Please try again or contact support.")
     return None
 
+
+
+# Add this function to your database.py file
+def get_db_connection():
+    """Get a thread-safe database connection."""
+    try:
+        data_dir = Path("data")
+        data_dir.mkdir(exist_ok=True)
+        db_path = data_dir / 'rfp_analysis.db'
+        
+        # Create new connection
+        conn = sqlite3.connect(str(db_path))
+        
+        # Create tables if they don't exist
+        create_tables(conn)
+        
+        return conn
+    except Exception as e:
+        st.error(f"Database connection error: {str(e)}")
+        return None
+
 def create_tables(conn):
     """Create necessary tables in the database."""
     try:
@@ -75,8 +96,28 @@ def create_tables(conn):
     except Error as e:
         st.error(f"Error: {e}")
 
-# Add this function to your database.py file
-
+def insert_document(name, content):
+    """Insert a document with thread-safe connection."""
+    try:
+        conn = get_db_connection()
+        if conn is None:
+            return None
+            
+        cursor = conn.cursor()
+        cursor.execute(
+            "INSERT INTO documents (name, content) VALUES (?, ?)",
+            (name, content)
+        )
+        conn.commit()
+        doc_id = cursor.lastrowid
+        conn.close()
+        return doc_id
+    except Exception as e:
+        st.error(f"Error inserting document: {str(e)}")
+        if conn:
+            conn.rollback()
+            conn.close()
+        return None
 def get_documents(conn):
     """Retrieve all documents from the database.