Update src/streamlit_app.py

src/streamlit_app.py

@@ -426,39 +426,40 @@ def show_login_page():
 
 def upload_and_poll_files(api_key: str, uploaded_files: list):
     """
-    Helper function to upload and poll multiple PDF files.
-    
-    Handles uploading via upload_pdf(), polling for processing status,
-    and displaying appropriate UI messages.
+    Upload PDFs, display status, and clear messages + reset uploader after processing.
     """
     if not api_key:
         st.error("Missing API key. Please login first.")
         return
-    
-    progress_bar = st.progress(0)
+
+    # Create placeholders for overall messages
+    progress_bar = st.empty()
     status_text = st.empty()
     
+    # Track which files were successfully processed
+    processed_files = []
+
     for i, file in enumerate(uploaded_files):
         status_text.text(f"Uploading {file.name}...")
-        
         result = upload_pdf(api_key, file)
-        
+
         if not result["success"]:
             st.error(f"❌ {file.name}: {result['error']}")
             progress_bar.progress((i + 1) / len(uploaded_files))
             continue
-        
+
         data = result.get("data") or {}
         job_id = data.get("document_id") or data.get("job_id")
         status = data.get("status")
-        
+
         if status == "processing" and job_id:
-            # Polling loop for processing status
+            # Polling messages
             poll_text = st.empty()
-            poll_progress = st.progress(0)
+            poll_progress = st.empty()
+            elapsed = 0
             max_poll_seconds = 600
             poll_interval = 2
-            elapsed = 0
+
             while elapsed < max_poll_seconds:
                 try:
                     resp = requests.get(
@@ -466,16 +467,15 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
                         headers=get_headers(api_key),
                         timeout=10
                     )
-                    
                     if resp.status_code == 200:
                         job = resp.json()
                         final_status = job.get("status")
                         display_name = job.get("username") or job.get("user_id")
-                        
                         if final_status not in ["processing"]:
                             if final_status in ["success", "completed"]:
                                 res = job.get("result", {})
                                 st.success(f"✅ {file.name} processed for {display_name}: {res.get('filename', file.name)}")
+                                processed_files.append(file.name)
                             else:
                                 error_msg = job.get('error', 'unknown error')
                                 st.error(f"❌ {file.name} failed for {display_name}: {error_msg}")
@@ -484,30 +484,30 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
                             poll_text.text(f"Processing {file.name} for {display_name}... elapsed {elapsed}s")
                     else:
                         poll_text.text(f"Waiting for processing... (status {resp.status_code})")
-                        
                 except Exception as e:
                     poll_text.text(f"Waiting... ({str(e)})")
-                
                 time.sleep(poll_interval)
                 elapsed += poll_interval
                 poll_progress.progress(min(1.0, elapsed / max_poll_seconds))
-            
-            if elapsed >= max_poll_seconds:
-                st.error(f"❌ {file.name}: Processing timed out after {max_poll_seconds}s")
-                
+
+            # Clear polling messages after processing
+            poll_text.empty()
+            poll_progress.empty()
         elif status in ["success", "completed"]:
             st.success(f"✅ {file.name} uploaded and processed successfully")
-            
-        else:
-            # Unknown or missing status: assume success if data is present (backend stored it)
-            if data and (job_id or status):
-                st.success(f"✅ {file.name} uploaded successfully")
-            else:
-                st.info(f"ℹ️ {file.name}: Upload response unclear - assuming success")
-        
+            processed_files.append(file.name)
+
         progress_bar.progress((i + 1) / len(uploaded_files))
-    
-    status_text.text("✅ Upload complete!")
+
+    # Clear overall status
+    status_text.empty()
+    progress_bar.empty()
+
+    # ✅ Remove successfully processed files from uploader
+    if processed_files and "pdf_uploader" in st.session_state:
+        # Streamlit doesn't allow direct modification, so delete and rerun
+        del st.session_state["pdf_uploader"]
+        st.experimental_rerun()
 
 # ============================================
 # MAIN APPLICATION UI