Update app.py

app.py

@@ -141,8 +141,9 @@ def create_ui(agent):
         download_output = gr.File(label="Download Full Report")
 
         def analyze(message: str, history: list, files: list):
-            history = history + [{"role": "user", "content": message},
-                                 {"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."}]
+            history = history + [{"role": "user", "content": message}]
+            assistant_msg = {"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."}
+            history.append(assistant_msg)
             yield history, None
 
             extracted = ""
@@ -163,7 +164,8 @@ Medical Records:
 {extracted[:12000]}
 ### Potential Oversights:
 """
-            response = ""
+
+            streamed_response = ""
             try:
                 for chunk in agent.run_gradio_chat(
                     message=prompt,
@@ -177,15 +179,19 @@ Medical Records:
                     if chunk is None:
                         continue
                     if isinstance(chunk, str):
-                        response += chunk
+                        streamed_response += chunk
                     elif isinstance(chunk, list):
-                        response += "".join([c.content for c in chunk if hasattr(c, 'content') and c.content])
+                        streamed_response += "".join([c.content for c in chunk if hasattr(c, 'content')])
+
+                    current_cleaned = streamed_response.split("[TOOL_CALLS]")[0].strip()
+                    history[-1]["content"] = current_cleaned if current_cleaned else "⏳ Still processing..."
+                    yield history, None
 
-                cleaned = response.split("[TOOL_CALLS]")[0].strip()
-                if not cleaned:
-                    cleaned = "No clear oversights identified. Recommend comprehensive review."
+                final_cleaned = streamed_response.split("[TOOL_CALLS]")[0].strip()
+                if not final_cleaned:
+                    final_cleaned = "No clear oversights identified. Recommend comprehensive review."
 
-                history[-1] = {"role": "assistant", "content": cleaned}
+                history[-1] = {"role": "assistant", "content": final_cleaned}
                 report_path = os.path.join(report_dir, f"{file_hash_value}_report.txt") if file_hash_value else None
                 yield history, report_path if report_path and os.path.exists(report_path) else None
 
@@ -208,4 +214,4 @@ if __name__ == "__main__":
         show_error=True,
         allowed_paths=[report_dir],
         share=False
-    )
+    )