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
@@ -161,41 +165,36 @@ Medical Records:
 
 ### Potential Oversights:
 """
-            response_chunks = []
+            response = ""
             try:
                 for chunk in agent.run_gradio_chat(
                     message=prompt,
                     history=[],
                     temperature=0.2,
-                    max_new_tokens=1024,
+                    max_new_tokens=2048,
                     max_token=4096,
                     call_agent=False,
-                    conversation=[]
+                    conversation=[],
                 ):
-                    if not chunk:
+                    if chunk is None:
                         continue
                     if isinstance(chunk, str):
-                        response_chunks.append(chunk)
+                        response += chunk
                     elif isinstance(chunk, list):
-                        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()
-                    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
-                return
+                        response += "".join([c.content for c in chunk if hasattr(c, 'content') and c.content])
 
-            full_response = "".join(response_chunks)
-            final_output = full_response.split("[TOOL_CALLS]")[0].strip()
-            if not final_output:
-                final_output = "No clear oversights identified. Recommend comprehensive review."
-            history[-1] = {"role": "assistant", "content": final_output}
+                cleaned = response.split("[TOOL_CALLS]")[0].strip()
+                if not cleaned:
+                    cleaned = "No clear oversights identified. Recommend comprehensive review."
 
-            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
+                history[-1] = {"role": "assistant", "content": cleaned}
+                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)}"}
+                yield history, None
 
         send_btn.click(analyze, inputs=[msg_input, gr.State([]), file_upload], outputs=[chatbot, download_output])
         msg_input.submit(analyze, inputs=[msg_input, gr.State([]), file_upload], outputs=[chatbot, download_output])
@@ -211,4 +210,4 @@ if __name__ == "__main__":
         show_error=True,
         allowed_paths=[report_dir],
         share=False
-    )
+    )