Update app.py

app.py

@@ -50,36 +50,39 @@ conn_lock = Lock()
 vectorstore_lock = Lock()
 
 def initialize_database():
-    """Initialize database connection and create necessary tables."""
+    """Initialize database with persistent storage."""
     try:
-        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:
-                    st.error(f"Failed to access database: {e}")
-                    return False
-
-                conn = create_connection(db_path)
-
-                if conn is not None:
-                    # Create base tables
-                    create_tables(conn)
-                    # Create collection-specific tables
-                    create_collection_tables(conn)
-                    st.session_state.db_conn = conn
-                    return True
-                else:
-                    return False
+        if 'db_conn' not in st.session_state:
+            # Get storage path
+            if os.path.exists('/data'):
+                base_path = Path('/data')
             else:
+                base_path = Path(os.getcwd()) / 'data'
+                
+            # Ensure data directory exists
+            base_path.mkdir(parents=True, exist_ok=True)
+            
+            # Create database path
+            db_path = base_path / 'rfp_analysis.db'
+            
+            try:
+                # Ensure file can be created
+                db_path.touch(exist_ok=True)
+            except Exception as e:
+                st.error(f"Failed to create database file: {e}")
+                return False
+            
+            # Create connection
+            conn = create_connection(str(db_path))
+            if conn is not None:
+                # Create all tables (includes collection tables)
+                create_tables(conn)
+                st.session_state.db_conn = conn
                 return True
+            else:
+                return False
+        else:
+            return True
 
     except Exception as e:
         st.error(f"Database initialization error: {e}")