Update ui/ui_core.py

ui/ui_core.py

@@ -44,8 +44,8 @@ def extract_all_text_from_pdf(file_path, progress=None, index=0, total=1):
 
 def create_ui(agent: TxAgent):
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("<h1 style='text-align: center;'>💊 TxAgent: Therapeutic Reasoning</h1>")
-        chatbot = gr.Chatbot(label="TxAgent", height=600, type="messages")
+        gr.Markdown("<h1 style='text-align: center;'>📋 CPS: Clinical Patient Support System</h1>")
+        chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages")
 
         file_upload = gr.File(
             label="Upload Medical File",
@@ -58,9 +58,10 @@ def create_ui(agent: TxAgent):
 
         def handle_chat(message, history, conversation, uploaded_files, progress=gr.Progress()):
             context = (
-                "You are a clinical AI reviewing medical interview or form data. "
-                "Analyze the extracted content and reason step-by-step about what the doctor could have missed. "
-                "Don't answer yet — just reason."
+                "You are an advanced clinical reasoning AI. You have just received raw medical data extracted from patient forms, lab reports, or interview tables. "
+                "Your goal is to analyze this data like a clinical expert. Go step by step to detect patterns, spot unusual or missing info, and identify clinical red flags "
+                "or overlooked findings. Use medically grounded reasoning. Be detailed. At the end, explain what the doctor may have missed and why it matters. "
+                "Include examples, reference clinical logic, and suggest what should have been asked or done. This response will help improve real-world diagnostics."
             )
 
             if uploaded_files:
@@ -78,7 +79,7 @@ def create_ui(agent: TxAgent):
                         if progress:
                             progress((index + 1) / total_files, desc=f"Skipping unsupported file: {os.path.basename(path)}")
 
-                message = f"{context}\n\n---\n{extracted_text.strip()}\n---\n\nNow reason what the doctor might have missed."
+                message = f"{context}\n\n---\n{extracted_text.strip()}\n---\n\nBegin your reasoning."
 
             generator = agent.run_gradio_chat(
                 message=message,
@@ -99,7 +100,9 @@ def create_ui(agent: TxAgent):
         message_input.submit(fn=handle_chat, inputs=inputs, outputs=chatbot)
 
         gr.Examples([
-            ["Upload the files"],
+            ["Upload your medical form and ask what the doctor might’ve missed."],
+            ["This patient was treated with antibiotics for UTI. What else should we check?"],
+            ["Is there anything abnormal in the attached blood work report?"]
         ], inputs=message_input)
 
     return demo