Update app.py

app.py

@@ -109,7 +109,59 @@ def init_agent():
     return agent
 
 def stream_report(agent, file: Union[str, 'file'], full_output: str) -> Generator[Tuple[str, Union[str, None], str], None, None]:
-    yield from stream_report_wrapper(agent)(file, full_output)
+    accumulated_text = ""
+    try:
+        if file is None:
+            yield "❌ Please upload a valid Excel file.", None, ""
+            return
+
+        if hasattr(file, "read"):
+            text = extract_text_from_excel(file)
+        elif isinstance(file, str) and os.path.exists(file):
+            text = extract_text_from_excel(file)
+        else:
+            raise ValueError("❌ Invalid or missing file.")
+
+        chunks = split_text_into_chunks(text)
+
+        for i, chunk in enumerate(chunks):
+            prompt = build_prompt_from_text(chunk)
+            partial = ""
+            for res in agent.run_gradio_chat(
+                message=prompt, history=[], temperature=0.2,
+                max_new_tokens=MAX_NEW_TOKENS, max_token=MAX_MODEL_TOKENS,
+                call_agent=False, conversation=[]
+            ):
+                if isinstance(res, str):
+                    partial += res
+                elif hasattr(res, "content"):
+                    partial += res.content
+            cleaned = clean_response(partial)
+            accumulated_text += f"\n\n📄 **Chunk {i+1}**:\n{cleaned}"
+            yield accumulated_text, None, ""
+
+        summary_prompt = f"Summarize this analysis in a final structured report:\n\n" + accumulated_text
+        final_report = ""
+        for res in agent.run_gradio_chat(
+            message=summary_prompt, history=[], temperature=0.2,
+            max_new_tokens=MAX_NEW_TOKENS, max_token=MAX_MODEL_TOKENS,
+            call_agent=False, conversation=[]
+        ):
+            if isinstance(res, str):
+                final_report += res
+            elif hasattr(res, "content"):
+                final_report += res.content
+
+        cleaned = clean_response(final_report)
+        accumulated_text += f"\n\n📊 **Final Summary**:\n{cleaned}"
+        report_path = os.path.join(report_dir, f"report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md")
+        with open(report_path, 'w') as f:
+            f.write(f"# 🧠 Final Patient Report\n\n{cleaned}")
+
+        yield accumulated_text, report_path, cleaned
+
+    except Exception as e:
+        yield f"❌ Error: {str(e)}", None, ""
 
 def create_ui(agent):
     with gr.Blocks(css="""