Update app.py

app.py

@@ -108,63 +108,8 @@ def init_agent():
     agent.init_model()
     return agent
 
-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=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, ""
-
-    return wrapped
+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)
 
 def create_ui(agent):
     with gr.Blocks(css="""
@@ -212,13 +157,8 @@ Upload clinical Excel records below and click **Analyze** to generate a medical
         report_file = gr.File(label="Download Report", visible=False)
         full_output = gr.State(value="")
 
-        streaming_fn = stream_report_wrapper(agent)
-
-        def run_streaming(file, state):
-            return streaming_fn(file, state)
-
         analyze_btn.click(
-            fn=run_streaming,
+            fn=stream_report,
             inputs=[file_upload, full_output],
             outputs=[report_output_markdown, report_file, full_output]
         )
@@ -229,7 +169,7 @@ if __name__ == "__main__":
     try:
         agent = init_agent()
         demo = create_ui(agent)
-        demo.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=False)
+        demo.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=True)
     except Exception as e:
         print(f"Error: {str(e)}")
         sys.exit(1)