Update app.py

app.py

@@ -105,10 +105,10 @@ def init_agent():
     agent.init_model()
     return agent
 
-def process_final_report(agent, file, chatbot_state: List[Tuple[str, str]]) -> Tuple[List[Tuple[str, str]], Union[str, None]]:
+def process_final_report(agent, file, chatbot_state: List[Tuple[str, str]]) -> Tuple[List[Tuple[str, str]], Union[str, None, str]]:
     messages = chatbot_state if chatbot_state else []
     if file is None or not hasattr(file, "name"):
-        return messages + [("assistant", "❌ Please upload a valid Excel file.")], None
+        return messages + [("assistant", "❌ Please upload a valid Excel file.")], None, ""
 
     messages.append(("user", f"Processing Excel file: {os.path.basename(file.name)}"))
     text = extract_text_from_excel(file.name)
@@ -141,7 +141,7 @@ def process_final_report(agent, file, chatbot_state: List[Tuple[str, str]]) -> T
 
     valid = [r for r in chunk_responses if r and not r.startswith("❌")]
     if not valid:
-        return messages + [("assistant", "❌ No valid chunk results.")], None
+        return messages + [("assistant", "❌ No valid chunk results.")], None, ""
 
     summary_prompt = f"Summarize this analysis in a final structured report:\n\n" + "\n\n".join(valid)
     messages.append(("assistant", "📊 Generating final report..."))
@@ -162,9 +162,8 @@ def process_final_report(agent, file, chatbot_state: List[Tuple[str, str]]) -> T
     with open(report_path, 'w') as f:
         f.write(f"# 🧠 Final Patient Report\n\n{cleaned}")
 
-    messages.append(("assistant", cleaned))
     messages.append(("assistant", f"✅ Report generated and saved: {os.path.basename(report_path)}"))
-    return messages, report_path
+    return messages, report_path, cleaned
 
 def create_ui(agent):
     with gr.Blocks(css="""
@@ -207,19 +206,20 @@ def create_ui(agent):
 Upload clinical Excel records below and click **Analyze** to generate a medical summary.
 """)
         chatbot = gr.Chatbot(label="Chatbot", elem_classes="chatbot", type="tuples")
+        report_output_markdown = gr.Markdown(visible=False)
         file_upload = gr.File(label="Upload Excel File", file_types=[".xlsx"])
         analyze_btn = gr.Button("Analyze")
         report_output = gr.File(label="Download Report", visible=False)
         chatbot_state = gr.State(value=[])
 
         def update_ui(file, current_state):
-            messages, report_path = process_final_report(agent, file, current_state)
-            return messages, gr.update(visible=report_path is not None, value=report_path), messages
+            messages, report_path, final_text = process_final_report(agent, file, current_state)
+            return messages, gr.update(visible=report_path is not None, value=report_path), messages, gr.update(visible=True, value=final_text)
 
         analyze_btn.click(
             fn=update_ui,
             inputs=[file_upload, chatbot_state],
-            outputs=[chatbot, report_output, chatbot_state]
+            outputs=[chatbot, report_output, chatbot_state, report_output_markdown]
         )
 
     return demo