Update app.py

app.py

@@ -290,7 +290,8 @@ def create_ui(agent):
                         "assets/user.png", 
                         "assets/assistant.png"
                     ),
-                    render=False  # Disable auto-render for better streaming control
+                    render=False,
+                    bubble_full_width=False
                 )
             with gr.Column(scale=1):
                 final_summary = gr.Markdown(
@@ -332,13 +333,11 @@ def create_ui(agent):
 
         def update_progress(current, total, stage=""):
             progress = f"{stage} - {current}/{total}" if stage else f"{current}/{total}"
-            return {
-                progress_text: gr.Textbox(
-                    value=progress,
-                    visible=True,
-                    label=f"Progress: {progress}"
-                )
-            }
+            return gr.Textbox.update(
+                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.
@@ -375,7 +374,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                 "chatbot": history,
                 "download_output": None,
                 "final_summary": "",
-                "progress_text": gr.Textbox(visible=True, value="Starting analysis...")
+                "progress_text": update_progress(0, 1, "Starting")
             }
 
             extracted = []
@@ -392,7 +391,9 @@ Patient Record Excerpt (Chunk {0} of {1}):
                     for i, future in enumerate(as_completed(futures), 1):
                         try:
                             extracted.extend(future.result())
-                            yield update_progress(i, len(files), "Processing files")
+                            yield {
+                                "progress_text": update_progress(i, len(files), "Processing files")
+                            }
                         except Exception as e:
                             logger.error(f"File processing error: {e}")
                             extracted.append({"error": f"Error processing file: {str(e)}"})
@@ -403,7 +404,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                     "chatbot": history,
                     "download_output": None,
                     "final_summary": "",
-                    "progress_text": gr.Textbox(visible=True, value="Files processed, analyzing content...")
+                    "progress_text": update_progress(len(files), len(files), "Files processed")
                 }
 
             # Convert extracted data to JSON text
@@ -423,10 +424,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                         "chatbot": history,
                         "download_output": None,
                         "final_summary": "",
-                        "progress_text": gr.Textbox(
-                            visible=True,
-                            value=f"Analyzing chunk {chunk_idx}/{len(chunks)}"
-                        )
+                        "progress_text": update_progress(chunk_idx, len(chunks), "Analyzing")
                     }
                     
                     # Process and stream the response
@@ -439,10 +437,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                             "chatbot": history,
                             "download_output": None,
                             "final_summary": "",
-                            "progress_text": gr.Textbox(
-                                visible=True,
-                                value=f"Analyzing chunk {chunk_idx}/{len(chunks)}"
-                            )
+                            "progress_text": update_progress(chunk_idx, len(chunks), "Analyzing")
                         }
                     
                     combined_response += f"--- Analysis for Chunk {chunk_idx} ---\n{chunk_response}\n"
@@ -462,7 +457,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                     "chatbot": history,
                     "download_output": gr.File(report_path) if report_path and os.path.exists(report_path) else None,
                     "final_summary": summary,
-                    "progress_text": gr.Textbox(visible=False)
+                    "progress_text": gr.Textbox.update(visible=False)
                 }
 
             except Exception as e:
@@ -472,18 +467,18 @@ 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(visible=False)
+                    "progress_text": gr.Textbox.update(visible=False)
                 }
 
         def clear_and_start():
-            return {
-                "chatbot": [],
-                "download_output": None,
-                "final_summary": "",
-                "msg_input": "",
-                "file_upload": None,
-                "progress_text": gr.Textbox(visible=False)
-            }
+            return [
+                [],  # chatbot
+                None,  # download_output
+                "",  # final_summary
+                "",  # msg_input
+                None,  # file_upload
+                gr.Textbox.update(visible=False)  # progress_text
+            ]
 
         # Event handlers
         send_btn.click(