Debug

app.py

@@ -91,13 +91,13 @@ class RichTextStreamer(TextIteratorStreamer):
 
 
 @spaces.GPU
-def chat_with_model(messages):
+def chat_with_model(messages, pid):            
     global current_model, current_tokenizer
     if current_model is None or current_tokenizer is None:
         yield messages + [{"role": "assistant", "content": "⚠️ No model loaded."}]
         return
 
-    current_id = patient_id.value
+    current_id = pid
     if not current_id:
         yield messages
         return
@@ -254,15 +254,12 @@ def format_prompt(messages):
     prompt += "Assistant:"
     return prompt
 
-def add_user_message(user_input, history):
-    current_id = patient_id.value
-    if not current_id:
-        # No patient selected yet — just return empty chat
+def add_user_message(user_input, history, pid):
+    if not pid:                                  # <-- use the arg, not .value
         return "", []
-
-    conversation = patient_conversations.get(current_id, [])
-    conversation.append({"role": "user", "content": user_input})
-    patient_conversations[current_id] = conversation
+    conv = patient_conversations.get(pid, [])
+    conv.append({"role": "user", "content": user_input})
+    patient_conversations[pid] = conv
     return "", [msg for msg in ([{
         "role": "assistant",
         "content": (
@@ -276,9 +273,7 @@ def add_user_message(user_input, history):
             "- \"Which images are available for this patient?\"\n"
             "- \"Can you segment the spleen from the CT scan?\"\n"
         )
-    }] + conversation)]
-
-
+    }] + conv)]
 
 def autofill_patient(patient_key):
     if patient_key in patient_db:
@@ -405,16 +400,15 @@ with gr.Blocks(css=".gradio-container {height: 100vh; overflow: hidden;}") as de
         load_model_on_selection, inputs=default_model, outputs=model_status
     )
 
-    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
+    msg.submit(
+        add_user_message,
+        [msg, chatbot, patient_id],                
+        [msg, chatbot],
+        queue=False,
     ).then(
-        chat_with_model, chatbot, chatbot
+        chat_with_model,
+        [chatbot, patient_id],                       
+        chatbot,
     )