Update app.py

app.py

@@ -119,13 +119,13 @@ def init_agent():
     agent.init_model()
     return agent
 
-def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], Dict[str, Any]]:
+def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], Union[str, None]]:
     messages = chatbot_state if chatbot_state else []
-    report_output = {"visible": False, "value": None}
+    report_path = None
 
     if file is None or not hasattr(file, "name"):
         messages.append({"role": "assistant", "content": "❌ Please upload a valid Excel file before analyzing."})
-        return messages, report_output
+        return messages, report_path
 
     try:
         messages.append({"role": "user", "content": f"Processing Excel file: {os.path.basename(file.name)}"})
@@ -193,12 +193,11 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
             f.write(final_report)
 
         messages.append({"role": "assistant", "content": f"✅ Report generated and saved: report_{timestamp}.md"})
-        report_output = {"visible": True, "value": report_path}
 
     except Exception as e:
         messages.append({"role": "assistant", "content": f"❌ Error processing file: {str(e)}"})
 
-    return messages, report_output
+    return messages, report_path
 
 def create_ui(agent):
     with gr.Blocks(title="Patient History Chat", css=".gradio-container {max-width: 900px !important}") as demo:
@@ -236,8 +235,9 @@ def create_ui(agent):
         chatbot_state = gr.State(value=[])
 
         def update_ui(file, current_state):
-            messages, report_output = process_final_report(agent, file, current_state)
-            return messages, report_output, messages
+            messages, report_path = process_final_report(agent, file, current_state)
+            report_update = gr.update(visible=report_path is not None, value=report_path)
+            return messages, report_update, messages
 
         analyze_btn.click(
             fn=update_ui,