Update ui/ui_core.py

ui/ui_core.py

@@ -33,7 +33,7 @@ def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, t
             line = " | ".join(str(cell) for cell in row if pd.notna(cell))
             if line:
                 lines.append(line)
-        return f"\U0001F4C4 {os.path.basename(file_path)}\n\n" + "\n".join(lines)
+        return f"📄 {os.path.basename(file_path)}\n\n" + "\n".join(lines)
 
     except Exception as e:
         return f"[Error reading {os.path.basename(file_path)}]: {str(e)}"
@@ -54,7 +54,7 @@ def extract_all_text_from_pdf(file_path: str, progress=None, index=0, total=1) -
                         progress((index + (i / num_pages)) / total, desc=f"Reading PDF: {os.path.basename(file_path)} ({i+1}/{num_pages})")
                 except Exception as e:
                     extracted.append(f"[Error reading page {i+1}]: {str(e)}")
-        return f"\U0001F4C4 {os.path.basename(file_path)}\n\n" + "\n\n".join(extracted)
+        return f"📄 {os.path.basename(file_path)}\n\n" + "\n\n".join(extracted)
 
     except Exception as e:
         return f"[Error reading PDF {os.path.basename(file_path)}]: {str(e)}"
@@ -78,7 +78,7 @@ 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;'>\U0001F4CB CPS: Clinical Patient Support System</h1>")
+        gr.Markdown("<h1 style='text-align: center;'>📋 CPS: Clinical Patient Support System</h1>")
 
         chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages", show_copy_button=True)
         file_upload = gr.File(
@@ -95,15 +95,18 @@ def create_ui(agent: TxAgent):
                 "You are an expert clinical AI assistant reviewing medical form or interview data. "
                 "Your job is to analyze this data and reason about any information or red flags that a human doctor might have overlooked. "
                 "Provide a **detailed and structured response**, including examples, supporting evidence from the form, and clinical rationale for why these items matter. "
-                "Ensure the output is informative and helpful for improving patient care. "
                 "Do not hallucinate. Base the response only on the provided form content. "
                 "End with a section labeled '🧠 Final Analysis' where you summarize key findings the doctor may have missed."
             )
 
-            try:
-                # Show centered loading message
-                yield history + [{"role": "assistant", "content": "<div style='text-align:center'>⏳ Processing... Please wait while I analyze the files.</div>"}]
+            # Show centered loading message in chatbot
+            updated_history = history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": "<div style='text-align:center'>⏳ Processing... Please wait while I analyze the files.</div>"}
+            ]
+            yield updated_history
 
+            try:
                 extracted_text = ""
                 if uploaded_files and isinstance(uploaded_files, list):
                     total_files = len(uploaded_files)
@@ -145,19 +148,18 @@ def create_ui(agent: TxAgent):
                             all_responses += update
 
                 all_responses = sanitize_utf8(all_responses.strip())
-                final_history = history + [
-                    {"role": "user", "content": message},
-                    {"role": "assistant", "content": all_responses}
-                ]
-                yield final_history
+
+                # Replace the temporary loading message with the final answer
+                updated_history[-1] = {"role": "assistant", "content": all_responses}
+                yield updated_history
 
             except Exception as chat_error:
                 print(f"Chat error: {chat_error}")
-                final_history = history + [
-                    {"role": "user", "content": message},
-                    {"role": "assistant", "content": "❌ An error occurred while processing your request."}
-                ]
-                yield final_history
+                updated_history[-1] = {
+                    "role": "assistant",
+                    "content": "❌ An error occurred while processing your request. Please try again."
+                }
+                yield updated_history
 
         inputs = [message_input, chatbot, conversation_state, file_upload]
         send_button.click(fn=handle_chat, inputs=inputs, outputs=chatbot)
@@ -169,4 +171,4 @@ def create_ui(agent: TxAgent):
             ["Is there anything abnormal in the attached blood work report?"]
         ], inputs=message_input)
 
-    return demo
+    return demo