Update app.py

app.py

@@ -1,200 +1,53 @@
+#src/app.py
 import streamlit as st
 import backend
-from langchain.schema import HumanMessage, AIMessage
-from datetime import datetime
-
-# Initialize session state
-if 'chat_history' not in st.session_state:
-    st.session_state.chat_history = []
-if 'current_chat' not in st.session_state:
-    st.session_state.current_chat = False
-if 'selected_docs' not in st.session_state:
-    st.session_state.selected_docs = []
-if 'qa_system' not in st.session_state:
-    st.session_state.qa_system = None
-if 'documents_initialized' not in st.session_state:
-    st.session_state.documents_initialized = False
-if 'processed_files' not in st.session_state:
-    st.session_state.processed_files = set()
-
-def handle_doc_select():
-    st.session_state.current_chat = True
-    st.session_state.chat_history = []
-
-def handle_start_chat():
-    st.session_state.current_chat = True
-    st.session_state.chat_history = []
-
-def display_chat_interface():
-    st.markdown("### 💬 RFP Analysis Chat")
-    
-    # Display chat history
-    for message in st.session_state.chat_history:
-        if isinstance(message, HumanMessage):
-            st.markdown(f"🧑‍💼 **You**: {message.content}")
-        else:
-            st.markdown(f"🤖 **Assistant**: {message.content}")
-    
-    # Chat input
-    if prompt := st.chat_input("Ask about the RFPs..."):
-        # Add user message to chat history
-        st.session_state.chat_history.append(HumanMessage(content=prompt))
-        
-        try:
-            with st.spinner("Analyzing documents..."):
-                # Convert chat history to the format expected by the QA system
-                formatted_history = []
-                for msg in st.session_state.chat_history[:-1]:  # Exclude the current message
-                    if isinstance(msg, HumanMessage):
-                        formatted_history.append({"role": "user", "content": msg.content})
-                    else:
-                        formatted_history.append({"role": "assistant", "content": msg.content})
-                
-                response = st.session_state.qa_system.invoke({
-                    "input": prompt,
-                    "chat_history": formatted_history
-                })
-            
-            # Add AI response to chat history
-            st.session_state.chat_history.append(AIMessage(content=response["output"]))
-            
-        except Exception as e:
-            st.error(f"Error generating response: {e}")
-
-def display_knowledge_base(conn):
-    st.markdown("### 📚 Knowledge Base")
-    
-    if conn is not None:
-        try:
-            cursor = conn.cursor()
-            cursor.execute("SELECT id, name, upload_date FROM documents ORDER BY upload_date DESC")
-            documents_in_db = cursor.fetchall()
-            
-            if documents_in_db:
-                st.markdown("#### Available Documents")
-                
-                for doc_id, name, upload_date in documents_in_db:
-                    col1, col2 = st.columns([3, 1])
-                    
-                    with col1:
-                        selected = st.checkbox(
-                            name,
-                            value=doc_id in st.session_state.selected_docs,
-                            key=f"doc_{doc_id}",
-                            on_change=handle_doc_select
-                        )
-                        
-                        if selected and doc_id not in st.session_state.selected_docs:
-                            st.session_state.selected_docs.append(doc_id)
-                        elif not selected and doc_id in st.session_state.selected_docs:
-                            st.session_state.selected_docs.remove(doc_id)
-                    
-                    with col2:
-                        upload_date = datetime.strptime(upload_date, '%Y-%m-%d %H:%M:%S')
-                        st.text(upload_date.strftime('%Y-%m-%d'))
-                
-                if st.session_state.selected_docs and not st.session_state.documents_initialized:
-                    with st.spinner("Initializing document analysis..."):
-                        selected_documents = []
-                        selected_doc_names = []
-                        
-                        for doc_id in st.session_state.selected_docs:
-                            cursor.execute(
-                                "SELECT content, name FROM documents WHERE id = ?",
-                                (doc_id,)
-                            )
-                            result = cursor.fetchone()
-                            if result:
-                                selected_documents.append(result[0])
-                                selected_doc_names.append(result[1])
-                        
-                        embeddings = backend.get_embeddings_model()
-                        if embeddings:
-                            vector_store = backend.initialize_faiss(
-                                embeddings,
-                                selected_documents,
-                                selected_doc_names
-                            )
-                            if vector_store:
-                                st.session_state.qa_system = backend.initialize_qa_system(vector_store)
-                                st.session_state.documents_initialized = True
-                
-                if st.session_state.selected_docs:
-                    st.button("🚀 Start New Chat", 
-                             on_click=handle_start_chat,
-                             use_container_width=True)
-            else:
-                st.info("No documents in the knowledge base. Upload some documents to get started!")
-                
-        except Exception as e:
-            st.error(f"Error accessing knowledge base: {e}")
+from utils.session_state import initialize_session_state
+from utils.database import get_documents
+from components.chat import display_chat_interface
+from components.knowledge_base import display_knowledge_base
+from components.upload import handle_document_upload
+from config.settings import APP_SETTINGS
 
 def main():
+    # Page configuration
     st.set_page_config(
-        page_title="SYNAPTYX - RFP Analysis Agent",
-        page_icon="🤖",
-        layout="wide"
+        page_title=APP_SETTINGS['title'],
+        page_icon=APP_SETTINGS['icon'],
+        layout=APP_SETTINGS['layout']
     )
     
-    st.title("🤖 SYNAPTYX - RFP Analysis Agent")
+    # Initialize session state
+    initialize_session_state()
+    
+    # Main title and description
+    st.title(APP_SETTINGS['title'])
     st.markdown(
-        """
+        f"""
         <p style='color: #1E3A8A;'>
-        Upload RFP documents, analyze requirements, and get intelligent answers powered by AI.
+        {APP_SETTINGS['description']}
         </p>
         """,
         unsafe_allow_html=True
     )
     
-    database = "rfp_agent.db"
-    conn = backend.create_connection(database)
-    if conn is not None:
-        backend.create_tables(conn)
-    else:
+    # Database initialization
+    conn = backend.create_connection(APP_SETTINGS['database'])
+    if conn is None:
         st.error("Error! Cannot create the database connection.")
         return
     
+    backend.create_tables(conn)
+    
+    # Create two columns for layout
     col1, col2 = st.columns([1, 2])
     
     with col1:
-        st.markdown("### 📤 Upload Documents")
-        uploaded_files = st.file_uploader(
-            "Upload PDF documents",
-            type="pdf",
-            accept_multiple_files=True
-        )
-        
-        if uploaded_files:
-            # Filter out already processed files
-            new_files = [f for f in uploaded_files if f.name not in st.session_state.processed_files]
-            
-            if new_files:
-                with st.spinner("Processing documents..."):
-                    all_texts, document_names, document_pages = backend.upload_and_parse_documents(new_files)
-                    if all_texts:
-                        try:
-                            with conn:
-                                for doc, doc_name in zip(all_texts, document_names):
-                                    # Check if document already exists in database
-                                    cursor = conn.cursor()
-                                    cursor.execute("SELECT id FROM documents WHERE name = ?", (doc_name,))
-                                    existing_doc = cursor.fetchone()
-                                    
-                                    if not existing_doc:
-                                        conn.execute(
-                                            "INSERT INTO documents (name, content) VALUES (?, ?)",
-                                            (doc_name, doc)
-                                        )
-                                        # Add to processed files set
-                                        st.session_state.processed_files.add(doc_name)
-                                
-                            st.success(f"Successfully uploaded {len(new_files)} new documents!")
-                        except Exception as e:
-                            st.error(f"Error saving documents to database: {e}")
-                    
-        display_knowledge_base(conn)
+        # Document upload and knowledge base
+        handle_document_upload(conn, backend)
+        display_knowledge_base(conn, backend)
     
     with col2:
+        # Chat interface or welcome message
         if st.session_state.current_chat and st.session_state.qa_system:
             display_chat_interface()
         else: