Update app.py

app.py

@@ -290,8 +290,7 @@ def create_ui(agent):
                         "assets/user.png", 
                         "assets/assistant.png"
                     ),
-                    render=False,
-                    bubble_full_width=False
+                    render=False
                 )
             with gr.Column(scale=1):
                 final_summary = gr.Markdown(
@@ -333,11 +332,7 @@ def create_ui(agent):
 
         def update_progress(current, total, stage=""):
             progress = f"{stage} - {current}/{total}" if stage else f"{current}/{total}"
-            return gr.Textbox.update(
-                value=progress,
-                visible=True,
-                label=f"Progress: {progress}"
-            )
+            return {"value": progress, "visible": True, "label": f"Progress: {progress}"}
 
         prompt_template = """
 Analyze the patient record excerpt for missed diagnoses only. Provide a concise, evidence-based summary as a single paragraph without headings or bullet points. Include specific clinical findings (e.g., 'elevated blood pressure (160/95) on page 10'), their potential implications (e.g., 'may indicate untreated hypertension'), and a recommendation for urgent review. Do not include other oversight categories like medication conflicts. If no missed diagnoses are found, state 'No missed diagnoses identified' in a single sentence.
@@ -374,7 +369,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                 "chatbot": history,
                 "download_output": None,
                 "final_summary": "",
-                "progress_text": update_progress(0, 1, "Starting")
+                "progress_text": {"value": "Starting analysis...", "visible": True}
             }
 
             extracted = []
@@ -455,9 +450,9 @@ Patient Record Excerpt (Chunk {0} of {1}):
                 
                 yield {
                     "chatbot": history,
-                    "download_output": gr.File(report_path) if report_path and os.path.exists(report_path) else None,
+                    "download_output": report_path if report_path and os.path.exists(report_path) else None,
                     "final_summary": summary,
-                    "progress_text": gr.Textbox.update(visible=False)
+                    "progress_text": {"visible": False}
                 }
 
             except Exception as e:
@@ -467,7 +462,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                     "chatbot": history,
                     "download_output": None,
                     "final_summary": f"Error occurred during analysis: {str(e)}",
-                    "progress_text": gr.Textbox.update(visible=False)
+                    "progress_text": {"visible": False}
                 }
 
         def clear_and_start():
@@ -477,7 +472,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                 "",  # final_summary
                 "",  # msg_input
                 None,  # file_upload
-                gr.Textbox.update(visible=False)  # progress_text
+                {"visible": False}  # progress_text
             ]
 
         # Event handlers