Update app.py

app.py

@@ -1,4 +1,8 @@
-
+import sys
+import os
+import pandas as pd
+import pdfplumber
+import json
 import gradio as gr
 from typing import List
 from concurrent.futures import ThreadPoolExecutor, as_completed
@@ -164,6 +168,8 @@ Medical Records:
 
             try:
                 response = ""
+                finish_detected = False
+
                 for chunk in agent.run_gradio_chat(
                     message=prompt,
                     history=[],
@@ -173,21 +179,26 @@ Medical Records:
                     call_agent=False,
                     conversation=[]
                 ):
+                    if chunk is None:
+                        continue
                     if isinstance(chunk, str):
                         response += chunk
                     elif isinstance(chunk, list):
-                        response += "".join([c.content for c in chunk if hasattr(c, "content") and c.content])
+                        chunk_str = "".join([c.content for c in chunk if hasattr(c, "content") and c.content])
+                        response += chunk_str
+                        if '"name": "Finish"' in chunk_str:
+                            finish_detected = True
 
-                clean_response = response.split("[TOOL_CALLS]")[0].strip()
+                clean_response = response.rsplit("[TOOL_CALLS]", 1)[0].strip()
                 if not clean_response:
-                    clean_response = "⚠️ No clear oversights identified or an error occurred."
+                    clean_response = "⚠️ No clear oversights identified or model output was invalid."
 
                 history[-1] = {"role": "assistant", "content": clean_response}
                 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
 
             except Exception as e:
-                print("ERROR:", str(e))
+                print("❌ ERROR:", str(e))
                 history[-1] = {"role": "assistant", "content": f"❌ An error occurred: {str(e)}"}
                 yield history, None