Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import streamlit as st
 import os
 import sqlite3
@@ -10,34 +12,39 @@ from utils.database import (
     create_tables
 )
 import time
+from threading import Lock
+
+# Create a lock for database operations
+conn_lock = Lock()
 
 def initialize_database():
     """Initialize database connection and tables."""
     try:
-        if 'db_conn' not in st.session_state:
-            data_dir = "data"
-            if not os.path.exists(data_dir):
-                os.makedirs(data_dir)
-            
-            db_path = os.path.join(data_dir, 'rfp_analysis.db')
-            
-            try:
-                with open(db_path, 'a') as f:
-                    pass
-            except IOError as e:
-                return False
-            
-            conn = create_connection(db_path)
-            
-            if conn is not None:
-                create_tables(conn)
-                st.session_state.db_conn = conn
-                return True
+        with conn_lock:
+            if 'db_conn' not in st.session_state:
+                data_dir = "data"
+                if not os.path.exists(data_dir):
+                    os.makedirs(data_dir)
+
+                db_path = os.path.join(data_dir, 'rfp_analysis.db')
+
+                try:
+                    with open(db_path, 'a') as f:
+                        pass
+                except IOError as e:
+                    return False
+
+                conn = create_connection(db_path)
+
+                if conn is not None:
+                    create_tables(conn)
+                    st.session_state.db_conn = conn
+                    return True
+                else:
+                    return False
             else:
-                return False
-        else:
-            return True
-            
+                return True
+
     except Exception:
         return False
 
@@ -111,7 +118,7 @@ def main():
                     st.session_state.processed_files = uploaded_files
                     st.session_state.chat_ready = True
                     time.sleep(1)
-                    st.rerun()
+                    st.experimental_rerun()
 
         # Knowledge Base Status
         if st.session_state.get('vector_store'):
@@ -148,4 +155,4 @@ def main():
         display_chat_interface()
 
 if __name__ == "__main__":
-    main()
+    main()