Update src/streamlit_app.py

src/streamlit_app.py

@@ -432,7 +432,6 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
     if "processed_files" not in st.session_state:
         st.session_state.processed_files = []
 
-    # Only keep files not yet processed
     files_to_process = [f for f in uploaded_files if f.name not in st.session_state.processed_files]
 
     if not files_to_process:
@@ -452,7 +451,8 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
 
         data = result.get("data") or {}
         job_id = data.get("document_id") or data.get("job_id")
-        status = data.get("status", "queued")  # default to queued if not provided
+        status = data.get("status", "queued")
+        display_name = st.session_state.username or "You"
 
         # Poll until final status
         if job_id:
@@ -463,8 +463,6 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
             elapsed = 0
 
             final_status = status
-            display_name = st.session_state.username or "You"
-
             while elapsed < max_poll_seconds:
                 try:
                     resp = requests.get(
@@ -497,6 +495,11 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
             if elapsed >= max_poll_seconds:
                 st.error(f"❌ {file.name}: Processing timed out after {max_poll_seconds}s")
 
+            # Clear polling message after a short delay
+            time.sleep(2)
+            poll_text.empty()
+            poll_progress.empty()
+
         elif status == "success":
             st.success(f"✅ {file.name} uploaded and processed successfully")
         else:
@@ -506,7 +509,11 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
         st.session_state.processed_files.append(file.name)
         progress_bar.progress((i + 1) / len(files_to_process))
 
-    status_text.text("✅ Upload complete!")
+    # Final upload complete message, disappears after 2s
+    status_text.success("✅ Upload complete!")
+    time.sleep(2)
+    status_text.empty()
+    progress_bar.empty()
 
 # ============================================
 # MAIN APPLICATION UI
@@ -562,15 +569,23 @@ def show_main_app():
 
         # Integrated uploader (upload before asking)
         uploaded_files = st.file_uploader(
-            "Upload PDF files (optional)",
-            type=["pdf"],
-            accept_multiple_files=True,
-            key="pdf_uploader"
-        )
+        "Upload PDF files (optional)",
+        type=["pdf"],
+        accept_multiple_files=True,
+        key="pdf_uploader")
+
+        # Filter out already processed files so they disappear from uploader
+        display_files = [f for f in uploaded_files if f.name not in st.session_state.get("processed_files", [])]
 
-        if uploaded_files:
+
+        if display_files:
             if st.button("📤 Upload All", use_container_width=True, type="secondary"):
-                upload_and_poll_files(st.session_state.api_key, uploaded_files)
+                upload_and_poll_files(st.session_state.api_key, display_files)
+        else:
+            if uploaded_files:
+                st.info("✅ All uploaded files have been processed and removed from uploader.")
+
+
 
         # Chat interface
         # Provide a scope selector: General (all docs) or a specific uploaded document