Update app.py

app.py

@@ -108,60 +108,63 @@ def init_agent():
     agent.init_model()
     return agent
 
-def stream_report(agent, file: Union[str, 'file'], full_output: str) -> Generator:
-    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 = ""
+def stream_report_wrapper(agent):
+    def wrapped(file: Union[str, 'file'], full_output: str) -> Generator:
+        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=prompt, history=[], temperature=0.2,
+                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):
-                    partial += res
+                    final_report += 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
+                    final_report += res.content
 
-    except Exception as e:
-        yield f"❌ Error: {str(e)}", None, ""
+            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, ""
+
+    return wrapped
 
 def create_ui(agent):
     with gr.Blocks(css="""
@@ -210,7 +213,7 @@ Upload clinical Excel records below and click **Analyze** to generate a medical
         full_output = gr.State(value="")
 
         analyze_btn.click(
-            fn=lambda file, state: stream_report(agent, file, state),
+            fn=stream_report_wrapper(agent),
             inputs=[file_upload, full_output],
             outputs=[report_output_markdown, report_file, full_output]
         )
@@ -224,4 +227,4 @@ if __name__ == "__main__":
         demo.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=False)
     except Exception as e:
         print(f"Error: {str(e)}")
-        sys.exit(1)
+        sys.exit(1)