Update app.py

app.py

@@ -1,4 +1,10 @@
-
+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
 import hashlib
 import shutil
@@ -178,8 +184,9 @@ Medical Records:
                         response_chunks.extend([c.content for c in chunk if hasattr(c, 'content')])
                     partial_response = "".join(response_chunks)
                     cleaned_partial = partial_response.split("[TOOL_CALLS]")[0].strip()
-                    history[-1] = {"role": "assistant", "content": cleaned_partial or "⏳ Working..."}
-                    yield history, None
+                    if cleaned_partial:
+                        history[-1] = {"role": "assistant", "content": cleaned_partial}
+                        yield history, None
             except Exception as e:
                 history[-1] = {"role": "assistant", "content": f"❌ Error: {str(e)}"}
                 yield history, None
@@ -208,4 +215,4 @@ if __name__ == "__main__":
         show_error=True,
         allowed_paths=[report_dir],
         share=False
-    )
+    )