Load patient conversations each time

app.py

@@ -135,9 +135,6 @@ def chat_with_model(messages):
         }
     ]
 
-    # Remove welcome message (only once shown)
-    # filtered_messages = [msg for msg in messages if not (msg["role"] == "assistant" and "Welcome to the Radiologist's Companion" in msg["content"])]
-
     # FULL conversation
     full_messages = system_messages + messages
 
@@ -323,6 +320,14 @@ def load_patient_conversation(patient_key):
         return [welcome_message] + history
     return []
 
+
+def get_patient_conversation():
+    current_id = patient_id.value
+    if not current_id:
+        return []
+    return patient_conversations.get(current_id, [])
+
+
 # --- Gradio App ---
 
 with gr.Blocks(css=".gradio-container {height: 100vh; overflow: hidden;}") as demo:
@@ -400,14 +405,19 @@ with gr.Blocks(css=".gradio-container {height: 100vh; overflow: hidden;}") as de
         load_model_on_selection, inputs=default_model, outputs=model_status
     )
 
-    # Submit message
     msg.submit(add_user_message, [msg, chatbot], [msg, chatbot], queue=False).then(
+        get_patient_conversation, None, chatbot
+    ).then(
         chat_with_model, chatbot, chatbot
     )
+
     submit_btn.click(add_user_message, [msg, chatbot], [msg, chatbot], queue=False).then(
+        get_patient_conversation, None, chatbot
+    ).then(
         chat_with_model, chatbot, chatbot
     )
 
+
     # Clear chat
     clear_btn.click(lambda: [], None, chatbot, queue=False)