Update app.py

app.py

@@ -253,7 +253,7 @@ def summarize_findings(combined_response: str) -> str:
     
     return summary.strip()
 
-def update_progress(current, total, stage=""):
+def update_progress(current: int, total: int, stage: str = "") -> Dict[str, Any]:
     progress = f"{stage} - {current}/{total}" if stage else f"{current}/{total}"
     return {"value": progress, "visible": True, "label": f"Progress: {progress}"}
 
@@ -364,7 +364,7 @@ Patient Record Excerpt (Chunk {chunk_idx} of {len(chunks)}):
             history.append({"role": "assistant", "content": ""})
             outputs.update({
                 "chatbot": history,
-                "progress_text": update_progress(chunk_idx, len(chunks)), "Analyzing")
+                "progress_text": update_progress(chunk_idx, len(chunks), "Analyzing")
             })
             yield outputs
             
@@ -375,7 +375,7 @@ Patient Record Excerpt (Chunk {chunk_idx} of {len(chunks)}):
                 chunk_response = update["content"]
                 outputs.update({
                     "chatbot": history,
-                    "progress_text": update_progress(chunk_idx, len(chunks)), "Analyzing")
+                    "progress_text": update_progress(chunk_idx, len(chunks), "Analyzing")
                 })
                 yield outputs