Update app.py

app.py

@@ -284,12 +284,12 @@ def load_patient_conversation(patient_key):
                 {
                     "role": "assistant",
                     "content": (
-                        "**Welcome to the Radiologist's Companion!**\n\n"
+                        "Welcome to the Radiologist's Companion!\n\n"
                         "You can ask me about the patient's medical history or available imaging data.\n"
                         "- I can summarize key details from the EHR.\n"
                         "- I can tell you which medical images are available.\n"
                         "- If you'd like an organ segmentation (e.g. spleen, liver, kidney_left, colon, femur_right) on an abdominal CT scan, just ask!\n\n"
-                        "**Example Requests:**\n"
+                        "Example Requests:\n"
                         "- \"What do we know about this patient?\"\n"
                         "- \"Which images are available for this patient?\"\n"
                         "- \"Can you segment the spleen from the CT scan?\"\n"
@@ -340,7 +340,7 @@ with gr.Blocks(css=css) as demo:
         # Chat
         with gr.Column(scale=3):
             gr.Markdown("### Chat")
-            chatbot = gr.Chatbot(label="Chat", type="messages", height=400)
+            chatbot = gr.Chatbot(label="Chat", type="messages", height=450)
             msg = gr.Textbox(label="Your message", placeholder="Enter your chat message...", show_label=False)
             with gr.Row():
                 submit_btn = gr.Button("Submit", variant="primary")
@@ -404,6 +404,17 @@ with gr.Blocks(css=css) as demo:
         chatbot,
     )
 
+    submit_btn.click(
+        add_user_message,
+        [msg, chatbot, patient_id],                
+        [msg, chatbot],
+        queue=False,
+    ).then(
+        chat_with_model,
+        [chatbot, patient_id],                       
+        chatbot,
+    )
+
 
     # Clear chat
     clear_btn.click(lambda: [], None, chatbot, queue=False)