Update app.py

app.py

@@ -108,6 +108,12 @@ def log_system_usage(tag=""):
     except Exception as e:
         print(f"[{tag}] GPU/CPU monitor failed: {e}")
 
+def clean_response(text: str) -> str:
+    text = sanitize_utf8(text)
+    text = re.sub(r"\[TOOL_CALLS\].*", "", text, flags=re.DOTALL)
+    text = re.sub(r"\n{3,}", "\n\n", text).strip()
+    return text
+
 def init_agent():
     print("🔁 Initializing model...")
     log_system_usage("Before Load")
@@ -141,7 +147,6 @@ def create_ui(agent):
         download_output = gr.File(label="Download Full Report")
 
         def analyze(message: str, history: List[dict], files: List):
-            # Append user message
             history.append({"role": "user", "content": message})
             history.append({"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."})
             yield history, None
@@ -166,11 +171,9 @@ Medical Records:
 """
 
             try:
-                # Remove the temporary "Analyzing..." message
                 if history and history[-1]["content"].startswith("⏳"):
                     history.pop()
 
-                # Process agent response
                 for chunk in agent.run_gradio_chat(
                     message=prompt,
                     history=[],
@@ -186,17 +189,19 @@ Medical Records:
                     if isinstance(chunk, list):
                         for m in chunk:
                             if hasattr(m, 'content') and m.content:
-                                history.append({"role": m.role, "content": sanitize_utf8(m.content)})
-                                yield history, None
+                                cleaned = clean_response(m.content)
+                                if cleaned:
+                                    history.append({"role": m.role, "content": cleaned})
+                                    yield history, None
                     elif isinstance(chunk, str) and chunk.strip():
-                        # Append new assistant message or update the last one
-                        if history and history[-1]["role"] == "assistant":
-                            history[-1]["content"] += sanitize_utf8(chunk)
-                        else:
-                            history.append({"role": "assistant", "content": sanitize_utf8(chunk)})
-                        yield history, None
-
-                # Provide report path if available
+                        cleaned = clean_response(chunk)
+                        if cleaned:
+                            if history and history[-1]["role"] == "assistant":
+                                history[-1]["content"] += cleaned
+                            else:
+                                history.append({"role": "assistant", "content": cleaned})
+                            yield history, None
+
                 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