Update ui/ui_core.py

ui/ui_core.py

@@ -80,8 +80,8 @@ def chunk_text(text: str, max_tokens: int = 8192) -> List[str]:
 def create_ui(agent: TxAgent):
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("<h1 style='text-align: center;'>📋 CPS: Clinical Patient Support System</h1>")
-        chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages")
 
+        chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages", show_copy_button=True)
         file_upload = gr.File(
             label="Upload Medical File",
             file_types=[".pdf", ".txt", ".docx", ".jpg", ".png", ".csv", ".xls", ".xlsx"],
@@ -90,7 +90,12 @@ 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)
+
+        # Centered status box styled
+        status_box = gr.Markdown(
+            "<div style='text-align:center; font-size: 18px;'>⏳ Processing in progress... Please wait...</div>",
+            visible=False
+        )
 
         def handle_chat(message: str, history: list, conversation: list, uploaded_files: list, progress=gr.Progress()):
             context = (
@@ -103,8 +108,10 @@ def create_ui(agent: TxAgent):
             )
 
             try:
-                yield gr.update(visible=True, value="⏳ Processing in progress... Please wait..."), history
+                # Show loading status
+                yield gr.update(visible=True), history
 
+                # Extract file text
                 extracted_text = ""
                 if uploaded_files and isinstance(uploaded_files, list):
                     total_files = len(uploaded_files)
@@ -123,8 +130,10 @@ def create_ui(agent: TxAgent):
                             extracted_text += f"[Error processing file: {os.path.basename(path)}] — {str(file_error)}\n"
                             continue
 
+                # Sanitize and chunk
                 sanitized = sanitize_utf8(extracted_text.strip())
                 chunks = chunk_text(sanitized, max_tokens=8192)
+
                 all_responses = ""
 
                 for i, chunk in enumerate(chunks):
@@ -144,14 +153,20 @@ def create_ui(agent: TxAgent):
                     )
                     for update in generator:
                         if isinstance(update, str):
-                            all_responses += update
+                            all_responses += update  # collect only final string output
 
                 all_responses = sanitize_utf8(all_responses.strip())
-                yield gr.update(visible=False), history + [[message, all_responses]]
+
+                # Convert to gr.Chatbot message format
+                history.append({"role": "user", "content": message})
+                history.append({"role": "assistant", "content": all_responses})
+                yield gr.update(visible=False), history
 
             except Exception as chat_error:
                 print(f"Chat handling error: {chat_error}")
-                yield gr.update(visible=False), history + [[message, "An error occurred while processing your request."]]
+                history.append({"role": "user", "content": message})
+                history.append({"role": "assistant", "content": "❌ An error occurred while processing your request. Please try again."})
+                yield gr.update(visible=False), history
 
         inputs = [message_input, chatbot, conversation_state, file_upload]
         send_button.click(fn=handle_chat, inputs=inputs, outputs=[status_box, chatbot])
@@ -164,3 +179,4 @@ def create_ui(agent: TxAgent):
         ], inputs=message_input)
 
     return demo
+