Update ui/ui_core.py

ui/ui_core.py

@@ -90,6 +90,7 @@ def create_ui(agent: TxAgent):
         message_input = gr.Textbox(placeholder="Ask a biomedical question or just upload the files...", show_label=False)
         send_button = gr.Button("Send", variant="primary")
         conversation_state = gr.State([])
+        status_box = gr.Markdown(visible=False)
 
         def handle_chat(message: str, history: list, conversation: list, uploaded_files: list, progress=gr.Progress()):
             context = (
@@ -102,13 +103,14 @@ def create_ui(agent: TxAgent):
             )
 
             try:
+                yield gr.update(visible=True, value="⏳ Processing in progress... Please wait..."), history
+
                 extracted_text = ""
                 if uploaded_files and isinstance(uploaded_files, list):
                     total_files = len(uploaded_files)
                     for index, file in enumerate(uploaded_files):
                         if not hasattr(file, 'name'):
                             continue
-
                         path = file.name
                         try:
                             if path.endswith((".csv", ".xls", ".xlsx")):
@@ -123,6 +125,7 @@ def create_ui(agent: TxAgent):
 
                 sanitized = sanitize_utf8(extracted_text.strip())
                 chunks = chunk_text(sanitized, max_tokens=8192)
+                all_responses = ""
 
                 for i, chunk in enumerate(chunks):
                     full_message = (
@@ -139,20 +142,20 @@ def create_ui(agent: TxAgent):
                         uploaded_files=uploaded_files,
                         max_round=30
                     )
-
                     for update in generator:
-                        if isinstance(update, list):
-                            yield update
-                        elif isinstance(update, str):
-                            yield [("assistant", sanitize_utf8(update))]
+                        if isinstance(update, str):
+                            all_responses += update
+
+                all_responses = sanitize_utf8(all_responses.strip())
+                yield gr.update(visible=False), history + [[message, all_responses]]
 
             except Exception as chat_error:
                 print(f"Chat handling error: {chat_error}")
-                yield [("assistant", "An error occurred while processing your request. Please try again.")]
+                yield gr.update(visible=False), history + [[message, "An error occurred while processing your request."]]
 
         inputs = [message_input, chatbot, conversation_state, file_upload]
-        send_button.click(fn=handle_chat, inputs=inputs, outputs=chatbot)
-        message_input.submit(fn=handle_chat, inputs=inputs, outputs=chatbot)
+        send_button.click(fn=handle_chat, inputs=inputs, outputs=[status_box, chatbot])
+        message_input.submit(fn=handle_chat, inputs=inputs, outputs=[status_box, chatbot])
 
         gr.Examples([
             ["Upload your medical form and ask what the doctor might've missed."],