Update app.py

app.py

@@ -1,4 +1,7 @@
 import streamlit as st
+import os
+import sqlite3
+from pathlib import Path
 from components.chat import display_chat_interface
 from utils.database import (
     display_vector_store_info, 
@@ -7,29 +10,55 @@ from utils.database import (
     create_tables
 )
 import time
-import os
 
 def initialize_database():
-    """Initialize database connection and tables."""
-    if 'db_conn' not in st.session_state:
-        # Create the data directory if it doesn't exist
-        data_dir = "data"
-        if not os.path.exists(data_dir):
-            os.makedirs(data_dir)
-        
-        # Initialize database connection
-        db_path = os.path.join(data_dir, 'rfp_analysis.db')
-        conn = create_connection(db_path)
-        
-        if conn is not None:
-            # Create tables
-            create_tables(conn)
-            # Store connection in session state
-            st.session_state.db_conn = conn
-            return True
+    """Initialize database connection and tables with debug info."""
+    try:
+        if 'db_conn' not in st.session_state:
+            # Create the data directory if it doesn't exist
+            data_dir = "data"
+            st.write(f"Checking data directory: {os.path.abspath(data_dir)}")
+            
+            if not os.path.exists(data_dir):
+                st.write("Data directory not found, creating...")
+                os.makedirs(data_dir)
+                st.write("Data directory created successfully")
+            else:
+                st.write("Data directory already exists")
+            
+            # Initialize database connection
+            db_path = os.path.join(data_dir, 'rfp_analysis.db')
+            st.write(f"Full database path: {os.path.abspath(db_path)}")
+            
+            # Check write permissions
+            try:
+                with open(db_path, 'a') as f:
+                    pass
+                st.write("Write permission test successful")
+            except IOError as e:
+                st.error(f"Write permission error: {str(e)}")
+                return False
+            
+            conn = create_connection(db_path)
+            
+            if conn is not None:
+                st.write("Creating database tables...")
+                create_tables(conn)
+                st.session_state.db_conn = conn
+                st.write("Database initialization complete!")
+                return True
+            else:
+                st.error("Failed to create database connection")
+                return False
         else:
-            st.error("Error! Cannot create the database connection.")
-            return False
+            st.write("Database connection already exists in session state")
+            return True
+            
+    except Exception as e:
+        st.error(f"Database initialization error: {str(e)}")
+        import traceback
+        st.error(f"Detailed error: {traceback.format_exc()}")
+        return False
 
 def display_example_questions():
     examples = [