Update app.py

app.py

@@ -451,7 +451,13 @@ def display_chat_area():
         display_chat_interface()
 
 def main():
-    st.set_page_config(layout="wide", page_title="SYNAPTYX - RFP Analysis Agent")
+    """Main application function."""
+    # Page configuration
+    st.set_page_config(
+        layout="wide",
+        page_title="SYNAPTYX - RFP Analysis Agent",
+        initial_sidebar_state="expanded"
+    )
     
     # Initialize database and session state
     if not initialize_database():
@@ -463,19 +469,24 @@ def main():
     # 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
+        if os.path.exists("img/logo.png"):
+            st.image("img/logo.png", width=200)  # Increased logo size
+        else:
+            st.info("Logo not found")
     with col2:
         st.title("SYNAPTYX - RFP Analysis Agent")
+    with col3:
+        if st.session_state.current_collection:
+            st.info(f"📁 Active Collection: {st.session_state.current_collection['name']}")
     
-    # Single sidebar implementation
+    # Sidebar
     with st.sidebar:
         # Chat Controls Section
         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
+            reset_chat_state()
             st.rerun()
         
         st.divider()
@@ -484,17 +495,23 @@ def main():
         st.title("📚 Document Manager")
         
         # Collection Creation Button
-        if st.button("➕ Create New Collection"):
+        if st.button("➕ Create New Collection", use_container_width=True):
             st.session_state.show_collection_dialog = True
 
         # Collection Selection
         collections = get_collections(st.session_state.db_conn)
         if collections:
-            upload_to = st.selectbox(
+            selected = st.selectbox(
                 "Select Collection",
-                options=["Select Collection..."] + [c['name'] for c in collections],
-                key="upload_destination"
+                options=["All Documents"] + [c['name'] for c in collections],
+                key="collection_select"
             )
+            
+            if selected != "All Documents":
+                collection = next((c for c in collections if c['name'] == selected), None)
+                if collection:
+                    st.session_state.current_collection = collection
+                    display_collection_documents(collection['id'])
         
         # Upload Section
         st.header("Upload Documents")
@@ -505,39 +522,45 @@ def main():
             help="Limit 200MB per file • PDF"
         )
 
-        if uploaded_files and upload_to != "Select Collection...":
-            collection_id = next(
-                (c['id'] for c in collections if c['name'] == upload_to),
-                None
-            )
-            if collection_id:
-                handle_document_upload(uploaded_files, collection_id=collection_id)
+        if uploaded_files:
+            handle_document_upload(uploaded_files, 
+                                 collection_id=st.session_state.current_collection.get('id') if st.session_state.current_collection else None)
     
     # Show collection creation dialog if triggered
-    if st.session_state.get('show_collection_dialog', False):
+    if st.session_state.show_collection_dialog:
         show_collection_creation_dialog()
     
     # 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:
-            display_welcome_screen()
-    
-    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 st.session_state.chat_ready:
+        display_chat_interface()
+    else:
+        display_welcome_screen()
+
+    # Debug mode (if enabled)
+    if st.session_state.debug_mode and st.sidebar.checkbox("Show Debug Info"):
+        st.sidebar.write("Current State:", get_current_state())
+
+def display_collection_documents(collection_id: int):
+    """Display documents in the selected collection."""
+    documents = get_collection_documents(st.session_state.db_conn, collection_id)
+    if documents:
+        st.markdown("### Documents in Collection")
+        for doc in documents:
+            with st.expander(f"📄 {doc['name']}", expanded=False):
+                st.caption(f"Uploaded: {doc['upload_date']}")
+                col1, col2 = st.columns(2)
+                with col1:
+                    if st.button("Preview", key=f"preview_{doc['id']}"):
+                        st.text_area(
+                            "Content Preview",
+                            value=doc['content'][:500] + "..." if len(doc['content']) > 500 else doc['content'],
+                            height=100,
+                            disabled=True
+                        )
+                with col2:
+                    if st.button("Use for Chat", key=f"chat_{doc['id']}"):
+                        initialize_chat_from_collection()
+                        st.rerun()
 
 if __name__ == "__main__":
     main()