Update app.py

app.py

@@ -183,36 +183,21 @@ def display_collection_sidebar():
     with st.sidebar:
         st.title("📚 Document Manager")
 
-        # Collection Creation Section
-        st.header("Collections")
-        if st.button("➕ Create New Collection"):
-            show_collection_creation_dialog()
+        # Collection Creation Button
+        if st.button("➕ Create New Collection", use_container_width=True):
+            st.session_state.show_collection_dialog = True
 
-        # Collection Selection
+        # Upload Section
+        st.header("Upload Documents")
+        
+        # Get collections for dropdown
         collections = get_collections(st.session_state.db_conn)
         if collections:
-            selected_collection = st.selectbox(
-                "Select Collection",
-                options=["All Documents"] + [c['name'] for c in collections],
-                key="collection_selector"
+            upload_to = st.selectbox(
+                "Upload to",
+                options=["Select Collection..."] + [c['name'] for c in collections],
+                key="upload_destination"
             )
-            
-            if selected_collection != "All Documents":
-                current_collection = next(
-                    (c for c in collections if c['name'] == selected_collection),
-                    None
-                )
-                if current_collection:
-                    st.session_state.current_collection = current_collection
-                    show_collection_details(current_collection)
-
-        # Upload Section
-        st.header("Upload Documents")
-        upload_to = st.selectbox(
-            "Upload to",
-            options=["New Collection"] + [c['name'] for c in collections],
-            key="upload_destination"
-        )
         
         uploaded_files = st.file_uploader(
             "Upload PDF documents",
@@ -221,8 +206,17 @@ def display_collection_sidebar():
             help="Limit 200MB per file • PDF"
         )
 
-        if uploaded_files:
-            handle_uploads(uploaded_files, upload_to)
+        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)
+
+    # Show collection creation dialog if triggered
+    if st.session_state.get('show_collection_dialog', False):
+        show_collection_creation_dialog()
 
 def display_collection_dialog():
     """Display the create collection dialog."""
@@ -289,30 +283,27 @@ def display_chat_area():
         display_chat_interface()
 
 def main():
-    st.set_page_config(
-        layout="wide",
-        page_title="SYNAPTYX - RFP Analysis Agent",
-        initial_sidebar_state="expanded"
-    )
+    st.set_page_config(layout="wide")
     
-    # Initialize database silently
-    if not initialize_database():
-        st.error("Failed to initialize database. Please contact support.")
-        return
-
+    # Initialize database and session state
+    initialize_database()
     initialize_session_state()
-        # Display sidebar first
+    
+    # Display sidebar
     display_collection_sidebar()
-    display_top_bar()
-
-    if st.session_state.chat_ready:
-        display_chat_interface()
-    else:
-        display_welcome_screen()
-
-    # Show collection creation dialog if needed
-    if st.session_state.get('show_collection_dialog'):
-        display_collection_dialog()
+    
+    # Main content area tabs
+    tab1, tab2 = st.tabs(["Chat", "Collections"])
+    
+    with tab1:
+        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()
 
 if __name__ == "__main__":
     main()