Update app.py

app.py

@@ -1,19 +1,38 @@
 #src/app.py
+
 import streamlit as st
 from utils.session_state import initialize_session_state
-from utils.database import get_documents, insert_document
+from utils.database import (
+    get_documents, 
+    insert_document, 
+    display_vector_store_info, 
+    handle_document_upload,
+    create_connection,
+    create_tables
+)
 from components.chat import display_chat_interface
-from utils.database import display_vector_store_info, handle_document_upload
 from components.knowledge_base import display_knowledge_base
-from components.upload import handle_document_upload
 from config.settings import APP_SETTINGS
 
-import streamlit as st
-from components.chat import display_chat_interface
-# Make sure to import explicitly from database.py
-from utils.database import display_vector_store_info, handle_document_upload, verify_vector_store
+
+def initialize_database():
+    """Initialize database connection and tables."""
+    if 'db_conn' not in st.session_state:
+        # Initialize database connection
+        conn = create_connection('rfp_analysis.db')
+        if conn is not None:
+            # Create tables
+            create_tables(conn)
+            # Store connection in session state
+            st.session_state.db_conn = conn
+        else:
+            st.error("Error! Cannot create the database connection.")
+
 
 def main():
+    # Initialize database connection
+    initialize_database()
+    
     st.title("🤖 SYNAPTYX - RFP Analysis Agent")
     st.markdown("Upload RFP documents, analyze requirements, and get intelligent answers powered by AI.")
 
@@ -30,8 +49,8 @@ def main():
     if uploaded_files:
         # Only process if files haven't been processed yet
         if 'processed_files' not in st.session_state or uploaded_files != st.session_state.processed_files:
-            # Make sure we're using the correct function from database.py
             handle_document_upload(uploaded_files)
+            # Store the processed files to avoid reprocessing
             st.session_state.processed_files = uploaded_files
 
     # Knowledge Base Section