Update app.py

app.py

@@ -141,8 +141,8 @@ 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.append({"role": "user", "content": message})
+            history.append({"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."})
             yield history, None
 
             extracted = ""
@@ -165,8 +165,9 @@ Medical Records:
 
 ### Potential Oversights:
 """
-            response = ""
+
             try:
+                response = ""
                 for chunk in agent.run_gradio_chat(
                     message=prompt,
                     history=[],
@@ -174,26 +175,24 @@ Medical Records:
                     max_new_tokens=2048,
                     max_token=4096,
                     call_agent=False,
-                    conversation=[],
+                    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])
+                        response += "".join([c.content for c in chunk if hasattr(c, "content") and c.content])
 
-                cleaned = response.split("[TOOL_CALLS]")[0].strip()
-                if not cleaned:
-                    cleaned = "No clear oversights identified. Recommend comprehensive review."
+                clean_response = response.split("[TOOL_CALLS]")[0].strip()
+                if not clean_response:
+                    clean_response = "⚠️ No clear oversights identified or an error occurred."
 
-                history[-1] = {"role": "assistant", "content": cleaned}
+                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:", e)
-                history[-1] = {"role": "assistant", "content": f"❌ Error occurred: {str(e)}"}
+                print("ERROR:", str(e))
+                history[-1] = {"role": "assistant", "content": f"❌ An error occurred: {str(e)}"}
                 yield history, None
 
         send_btn.click(analyze, inputs=[msg_input, gr.State([]), file_upload], outputs=[chatbot, download_output])
@@ -210,4 +209,4 @@ if __name__ == "__main__":
         show_error=True,
         allowed_paths=[report_dir],
         share=False
-    )
+    )