Update app.py

app.py

@@ -183,8 +183,37 @@ 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 Selection
+        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"
+            )
+            
+            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", anchor=False)
+        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",
             type=['pdf'],
@@ -192,63 +221,8 @@ def display_collection_sidebar():
             help="Limit 200MB per file • PDF"
         )
 
-        # Collection Selection
-        st.header("Collections")
-        collections = get_collections(st.session_state.db_conn)
-        
-        # Create New Collection button
-        if st.button("➕ Create New Collection"):
-            st.session_state.show_collection_dialog = True
-
-        # Collection selector
-        collection_options = ["All Documents"] + [c['name'] for c in collections]
-        selected_collection = st.selectbox(
-            "Select Collection",
-            collection_options,
-            index=0,
-            key="collection_selector"
-        )
-
-        # Update current collection in session state
-        if selected_collection != "All Documents":
-            st.session_state.current_collection = next(
-                (c for c in collections if c['name'] == selected_collection),
-                None
-            )
-        else:
-            st.session_state.current_collection = None
-
-        # Process uploads automatically when files are selected
         if uploaded_files:
-            if 'processed_files' not in st.session_state or uploaded_files != st.session_state.processed_files:
-                with st.spinner("Processing documents..."):
-                    collection_id = st.session_state.current_collection['id'] if st.session_state.current_collection else None
-                    # Create a dictionary with the collection_id
-                    upload_params = {'collection_id': collection_id} if collection_id else {}
-                    
-                    # Pass the parameters as a dictionary to handle_document_upload
-                    handle_document_upload(uploaded_files, **upload_params)
-                    st.session_state.processed_files = uploaded_files
-                    st.session_state.chat_ready = True
-                    # Switch to chat interface
-                    st.rerun()
-
-        # Display current collection info
-        if st.session_state.current_collection:
-            st.header("Collection Info")
-            collection = st.session_state.current_collection
-            documents = get_collection_documents(st.session_state.db_conn, collection['id'])
-            
-            st.markdown(f"""
-            **Collection:** {collection['name']}  
-            **Documents:** {len(documents)}  
-            **Created:** {collection['created_at']}
-            """)
-
-            if documents:
-                st.subheader("Documents in Collection")
-                for doc in documents:
-                    st.write(f"📄 {doc['name']}")
+            handle_uploads(uploaded_files, upload_to)
 
 def display_collection_dialog():
     """Display the create collection dialog."""