Update app.py

app.py

@@ -178,12 +178,13 @@ Medical Records:
                         continue
 
                     if isinstance(chunk, list) and all(hasattr(m, 'role') and hasattr(m, 'content') for m in chunk):
-                        messages = [{"role": m.role, "content": re.sub(r"\[TOOL_CALLS\].*?(?=(\[|\Z))", "", m.content, flags=re.DOTALL).strip()} for m in chunk]
-                        history = messages
-                        yield history, None
-
+                        for m in chunk:
+                            cleaned_content = re.sub(r"\[TOOL_CALLS\].*?(?=(\[|\Z))", "", m.content, flags=re.DOTALL).strip()
+                            history.append({"role": m.role, "content": cleaned_content or "⚠️ No answer returned."})
+                            yield history, None
                     elif isinstance(chunk, str):
-                        history[-1]["content"] += chunk
+                        cleaned_chunk = re.sub(r"\[TOOL_CALLS\].*?(?=(\[|\Z))", "", chunk, flags=re.DOTALL).strip()
+                        history[-1]["content"] += cleaned_chunk or "⚠️ No content generated."
                         yield history, None
 
                 report_path = os.path.join(report_dir, f"{file_hash_value}_report.txt") if file_hash_value else None
@@ -191,7 +192,7 @@ Medical Records:
 
             except Exception as e:
                 print("🚨 ERROR:", e)
-                history[-1] = {"role": "assistant", "content": f"❌ Error occurred: {str(e)}"}
+                history.append({"role": "assistant", "content": f"❌ Error occurred: {str(e)}"})
                 yield history, None
 
         send_btn.click(analyze, inputs=[msg_input, gr.State([]), file_upload], outputs=[chatbot, download_output])