Update utils/database.py

utils/database.py

@@ -28,6 +28,10 @@ import io
 import tempfile
 from sqlite3 import Error
 
+from threading import Lock
+
+# Create a lock for database connection
+conn_lock = Lock()
 
 def create_connection(db_file):
     """
@@ -40,7 +44,7 @@ def create_connection(db_file):
     """
     conn = None
     try:
-        conn = sqlite3.connect(db_file)
+        conn = sqlite3.connect(db_file, check_same_thread=False)
         return conn
     except Error as e:
         st.error("Failed to connect to database. Please try again or contact support.")
@@ -133,9 +137,10 @@ def get_documents(conn):
         tuple: (list of document contents, list of document names).
     """
     try:
-        cursor = conn.cursor()
-        cursor.execute("SELECT content, name FROM documents")
-        results = cursor.fetchall()
+        with conn_lock:
+            cursor = conn.cursor()
+            cursor.execute("SELECT content, name FROM documents")
+            results = cursor.fetchall()
 
         if not results:
             return [], []
@@ -163,12 +168,13 @@ def insert_document(conn, name, content):
         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
+        with conn_lock:
+            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}")