Update app.py

app.py

@@ -285,12 +285,12 @@ def create_ui(agent):
                 chatbot = gr.Chatbot(
                     label="Analysis Conversation",
                     height=600,
-                    bubble_full_width=False,
                     show_copy_button=True,
                     avatar_images=(
                         "assets/user.png", 
                         "assets/assistant.png"
-                    )
+                    ),
+                    render=False  # Disable auto-render for better streaming control
                 )
             with gr.Column(scale=1):
                 final_summary = gr.Markdown(
@@ -324,11 +324,22 @@ def create_ui(agent):
                 min_width=100
             )
         
-        progress_bar = gr.Progress(
-            label="Processing Progress",
-            visible=False
+        progress_text = gr.Textbox(
+            label="Progress",
+            visible=False,
+            interactive=False
         )
 
+        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}"
+                )
+            }
+
         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.
 Patient Record Excerpt (Chunk {0} of {1}):
@@ -364,7 +375,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                 "chatbot": history,
                 "download_output": None,
                 "final_summary": "",
-                "progress_bar": gr.Progress(visible=True)
+                "progress_text": gr.Textbox(visible=True, value="Starting analysis...")
             }
 
             extracted = []
@@ -378,9 +389,10 @@ Patient Record Excerpt (Chunk {0} of {1}):
                         file_type = f.name.split(".")[-1].lower()
                         futures.append(executor.submit(process_file, f.name, file_type))
                     
-                    for future in as_completed(futures):
+                    for i, future in enumerate(as_completed(futures), 1):
                         try:
                             extracted.extend(future.result())
+                            yield 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)}"})
@@ -391,7 +403,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                     "chatbot": history,
                     "download_output": None,
                     "final_summary": "",
-                    "progress_bar": gr.Progress(0.2, visible=True, label="Processing files")
+                    "progress_text": gr.Textbox(visible=True, value="Files processed, analyzing content...")
                 }
 
             # Convert extracted data to JSON text
@@ -411,10 +423,9 @@ Patient Record Excerpt (Chunk {0} of {1}):
                         "chatbot": history,
                         "download_output": None,
                         "final_summary": "",
-                        "progress_bar": gr.Progress(
-                            0.2 + (chunk_idx/len(chunks))*0.7, 
-                            visible=True, 
-                            label=f"Analyzing chunk {chunk_idx}/{len(chunks)}"
+                        "progress_text": gr.Textbox(
+                            visible=True,
+                            value=f"Analyzing chunk {chunk_idx}/{len(chunks)}"
                         )
                     }
                     
@@ -428,9 +439,9 @@ Patient Record Excerpt (Chunk {0} of {1}):
                             "chatbot": history,
                             "download_output": None,
                             "final_summary": "",
-                            "progress_bar": gr.Progress(
-                                0.2 + (chunk_idx/len(chunks))*0.7, 
-                                visible=True
+                            "progress_text": gr.Textbox(
+                                visible=True,
+                                value=f"Analyzing chunk {chunk_idx}/{len(chunks)}"
                             )
                         }
                     
@@ -451,7 +462,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_bar": gr.Progress(1.0, visible=False)
+                    "progress_text": gr.Textbox(visible=False)
                 }
 
             except Exception as e:
@@ -461,7 +472,7 @@ Patient Record Excerpt (Chunk {0} of {1}):
                     "chatbot": history,
                     "download_output": None,
                     "final_summary": f"Error occurred during analysis: {str(e)}",
-                    "progress_bar": gr.Progress(visible=False)
+                    "progress_text": gr.Textbox(visible=False)
                 }
 
         def clear_and_start():
@@ -470,27 +481,28 @@ Patient Record Excerpt (Chunk {0} of {1}):
                 "download_output": None,
                 "final_summary": "",
                 "msg_input": "",
-                "file_upload": None
+                "file_upload": None,
+                "progress_text": gr.Textbox(visible=False)
             }
 
         # Event handlers
         send_btn.click(
             analyze,
             inputs=[msg_input, chatbot, file_upload],
-            outputs=[chatbot, download_output, final_summary, progress_bar],
+            outputs=[chatbot, download_output, final_summary, progress_text],
             show_progress="hidden"
         )
         
         msg_input.submit(
             analyze,
             inputs=[msg_input, chatbot, file_upload],
-            outputs=[chatbot, download_output, final_summary, progress_bar],
+            outputs=[chatbot, download_output, final_summary, progress_text],
             show_progress="hidden"
         )
         
         demo.load(
             clear_and_start,
-            outputs=[chatbot, download_output, final_summary, msg_input, file_upload],
+            outputs=[chatbot, download_output, final_summary, msg_input, file_upload, progress_text],
             queue=False
         )
     
@@ -509,8 +521,7 @@ if __name__ == "__main__":
             server_port=7860,
             show_error=True,
             allowed_paths=[report_dir],
-            share=False,
-            favicon_path="assets/favicon.ico"
+            share=False
         )
     except Exception as e:
         logger.error(f"Failed to launch app: {e}")