Update app.py

app.py

@@ -162,7 +162,10 @@ 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}")
 
+    # Add the report content to the chat messages
     messages.append(("assistant", f"✅ Report generated and saved: {os.path.basename(report_path)}"))
+    messages.append(("assistant", f"## Final Report\n\n{cleaned}"))
+    
     return messages, report_path, cleaned
 
 def create_ui(agent):
@@ -201,12 +204,18 @@ def create_ui(agent):
         .gr-button:hover {
             background: linear-gradient(135deg, #37b6e9, #4b4ced);
         }
+        .report-content {
+            background-color: #1a1f2e;
+            padding: 15px;
+            border-radius: 8px;
+            margin-top: 10px;
+            border: 1px solid #2c3344;
+        }
     """) as demo:
         gr.Markdown("""# 🧠 Clinical Reasoning Assistant
 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)
@@ -214,12 +223,12 @@ Upload clinical Excel records below and click **Analyze** to generate a medical
 
         def update_ui(file, current_state):
             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)
+            return messages, gr.update(visible=report_path is not None, value=report_path), messages
 
         analyze_btn.click(
             fn=update_ui,
             inputs=[file_upload, chatbot_state],
-            outputs=[chatbot, report_output, chatbot_state, report_output_markdown]
+            outputs=[chatbot, report_output, chatbot_state]
         )
 
     return demo
@@ -231,4 +240,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)