Update app.py

app.py

@@ -68,8 +68,6 @@ def initialize_session_state():
         st.session_state.current_collection = None
     if 'processed_files' not in st.session_state:
         st.session_state.processed_files = []
-    if 'show_upload_dialog' not in st.session_state:
-        st.session_state.show_upload_dialog = False
     if 'vector_stores' not in st.session_state:
         st.session_state.vector_stores = {}
 
@@ -130,15 +128,20 @@ def display_collection_sidebar():
             help="Limit 200MB per file • PDF"
         )
 
+        # Process uploads automatically when files are selected
         if uploaded_files:
-            if st.button("Process Documents"):
+            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
-                    success = handle_document_upload(uploaded_files, collection_id)
-                    if success:
-                        st.success("Documents processed successfully!")
-                        st.session_state.chat_ready = True
-                        st.rerun()
+                    # 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: