Update src/streamlit_app.py

src/streamlit_app.py

@@ -428,6 +428,7 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
     """
     Upload PDFs to backend and poll until processed.
     After processing, files disappear from uploader.
+    Compatible with HF backend response.
     """
     if "processed_files" not in st.session_state:
         st.session_state.processed_files = []
@@ -452,17 +453,18 @@ 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")
+        initial_status = data.get("status")
 
-        # Poll for processing status
-        if status == "processing" and job_id:
+        # If backend returns "processing", poll until finished
+        if initial_status in ["processing", "success"] and job_id:
             poll_text = st.empty()
             poll_progress = st.progress(0)
             max_poll_seconds = 600
             poll_interval = 2
             elapsed = 0
+            final_status = initial_status
 
-            while elapsed < max_poll_seconds:
+            while elapsed < max_poll_seconds and final_status == "processing":
                 try:
                     resp = requests.get(
                         f"{API_BASE_URL}/upload-status/{job_id}",
@@ -473,16 +475,17 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
                     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 != "processing":
-                            if final_status == "success":
-                                res = job.get("result", {})
-                                st.success(f"✅ {file.name} processed for {display_name}: {res.get('filename', file.name)}")
-                            else:
-                                st.error(f"❌ {file.name} failed for {display_name}: {job.get('error', 'unknown error')}")
-                            break
-                        else:
+                        display_name = job.get("username") or job.get("user_id") or "You"
+
+                        if final_status == "processing":
                             poll_text.text(f"Processing {file.name} for {display_name}... elapsed {elapsed}s")
+                        elif final_status == "success":
+                            res = job.get("result", {})
+                            st.success(f"✅ {file.name} processed for {display_name}: {res.get('filename', file.name)}")
+                        elif final_status == "failed":
+                            st.error(f"❌ {file.name} failed for {display_name}: {job.get('error', 'unknown error')}")
+                        else:
+                            st.info(f"ℹ️ {file.name} has unknown status: {final_status}")
                     else:
                         poll_text.text(f"Waiting for processing... (status {resp.status_code})")
 
@@ -493,10 +496,10 @@ def upload_and_poll_files(api_key: str, uploaded_files: list):
                 elapsed += poll_interval
                 poll_progress.progress(min(1.0, elapsed / max_poll_seconds))
 
-            if elapsed >= max_poll_seconds:
+            if final_status == "processing":
                 st.error(f"❌ {file.name}: Processing timed out after {max_poll_seconds}s")
 
-        elif status == "success":
+        elif initial_status == "success":
             st.success(f"✅ {file.name} uploaded and processed successfully")
         else:
             st.info(f"ℹ️ {file.name}: {data}")