Update app.py

app.py

@@ -52,7 +52,15 @@ if not os.path.exists('/data'):
     except Exception as e:
         st.error(f"Setup error: {e}")
         st.stop()
-
+        
+def handle_chat_state_change():
+    """Handle changes in chat state when switching tabs or collections."""
+    if st.session_state.get('chat_ready'):
+        # Check if we need to reinitialize with new documents
+        if st.session_state.get('reinitialize_chat'):
+            initialize_chat_from_collection()
+            st.session_state.reinitialize_chat = False
+            
 def verify_database_tables(conn):
     """Verify that all required tables exist."""
     try:
@@ -128,7 +136,7 @@ def initialize_database():
         return False
 
 def initialize_session_state():
-    """Initialize all session state variables."""
+    """Initialize all session state variables.
     if 'chat_ready' not in st.session_state:
         st.session_state.chat_ready = False
     if 'current_collection' not in st.session_state:
@@ -145,6 +153,23 @@ def initialize_session_state():
         st.session_state.show_collection_dialog = False
     if 'selected_collection' not in st.session_state:
         st.session_state.selected_collection = None
+       """
+    
+    default_states = {
+        'show_collection_dialog': False,
+        'selected_collection': None,
+        'chat_ready': False,
+        'current_collection': None,
+        'vector_store': None,
+        'qa_system': None,
+        'messages': [],
+        'current_chat_id': None,
+        'reinitialize_chat': False
+    }
+    
+    for key, default_value in default_states.items():
+        if key not in st.session_state:
+            st.session_state[key] = default_value
 
 def display_top_bar():
     """Display the application's top navigation bar."""
@@ -194,6 +219,52 @@ def initialize_chat_from_existing():
             st.error(f"Error initializing chat: {e}")
     return False
 
+def initialize_chat_from_collection():
+    """Initialize chat system from existing collection or documents."""
+    try:
+        if not st.session_state.chat_ready:
+            documents = None
+            if st.session_state.get('current_collection'):
+                documents = get_collection_documents(st.session_state.db_conn, 
+                                                  st.session_state.current_collection['id'])
+            else:
+                documents = get_all_documents(st.session_state.db_conn)
+
+            if documents:
+                embeddings = get_embeddings_model()
+                text_splitter = RecursiveCharacterTextSplitter(
+                    chunk_size=1000,
+                    chunk_overlap=200,
+                    length_function=len,
+                    separators=["\n\n", "\n", " ", ""]
+                )
+                
+                # Process documents into chunks
+                chunks = []
+                for doc in documents:
+                    doc_chunks = text_splitter.split_text(doc['content'])
+                    for chunk in doc_chunks:
+                        chunks.append({
+                            'content': chunk,
+                            'metadata': {'source': doc['name'], 'document_id': doc['id']}
+                        })
+                
+                # Initialize vector store
+                vector_store = FAISS.from_texts(
+                    [chunk['content'] for chunk in chunks],
+                    embeddings,
+                    [chunk['metadata'] for chunk in chunks]
+                )
+                
+                st.session_state.vector_store = vector_store
+                st.session_state.qa_system = initialize_qa_system(vector_store)
+                st.session_state.chat_ready = True
+                return True
+        return st.session_state.chat_ready
+    except Exception as e:
+        st.error(f"Error initializing chat: {e}")
+        return False
+
 def display_collection_sidebar():
     """Display the document management sidebar."""
     with st.sidebar:
@@ -299,19 +370,46 @@ def display_chat_area():
         display_chat_interface()
 
 def main():
-    st.set_page_config(layout="wide")
+    st.set_page_config(layout="wide", page_title="SYNAPTYX - RFP Analysis Agent")
     
     # Initialize database and session state
-    initialize_database()
+    if not initialize_database():
+        st.error("Failed to initialize database. Please contact support.")
+        return
+
     initialize_session_state()
     
-    # Display sidebar
-    display_collection_sidebar()
+    # Top bar with larger logo
+    col1, col2, col3 = st.columns([1.5, 4, 1])
+    with col1:
+        st.image("img/logo.png", width=200)  # Increased logo size
+    with col2:
+        st.title("SYNAPTYX - RFP Analysis Agent")
     
-    # Main content area tabs
+    # Sidebar with chat controls
+    with st.sidebar:
+        st.title("💬 Chat Controls")
+        
+        # New Chat button
+        if st.button("🔄 Start New Chat", use_container_width=True):
+            st.session_state.messages = []
+            st.session_state.current_chat_id = None
+            st.rerun()
+        
+        st.divider()
+        
+        # Collection management below
+        st.title("📚 Document Manager")
+        display_collection_sidebar()
+    
+    # Main content area
     tab1, tab2 = st.tabs(["Chat", "Collections"])
     
     with tab1:
+        # Initialize chat if documents exist
+        if not st.session_state.chat_ready:
+            initialize_chat_from_collection()
+            
         if st.session_state.chat_ready:
             display_chat_interface()
         else:
@@ -320,6 +418,10 @@ def main():
     with tab2:
         from components.collection_manager import display_enhanced_collections
         display_enhanced_collections()
+        # After collection operations, check if we need to reinitialize chat
+        if st.session_state.get('reinitialize_chat'):
+            initialize_chat_from_collection()
+            st.session_state.reinitialize_chat = False
 
 if __name__ == "__main__":
     main()