Update app.py

app.py

@@ -177,7 +177,8 @@ def delete_chat_cb(selected_label):
     selected = labels[0] if labels else None
     return gr.update(choices=labels, value=selected), []
 
-def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens, temperature, top_p, model_choice, *args):
+# --- Fixed send_cb to show user message ---
+def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens, temperature, top_p, model_choice, dataset_choice, *args):
     sid = label_to_id(selected_label)
     if sid is None:
         sid = create_session("New chat")
@@ -190,7 +191,9 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
     api_messages = build_api_messages(sid, system_message)
     display_msgs = chatbot_msgs[:]  # preserve current chat
 
-    # Initial typing message
+    # Show user's message immediately
+    display_msgs.append({"role": "user", "content": user_text})
+    # Add assistant typing placeholder
     display_msgs.append({"role": "assistant", "content": "…"})
     yield (display_msgs, "", selected_label)
 
@@ -214,10 +217,6 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
         display_msgs[-1]["content"] = f"⚠️ Error: {str(e)}"
         yield (display_msgs, "", selected_label)
 
-
-
-
-
 def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p, model_choice, dataset_choice):
     sid = label_to_id(selected_label)
     if sid is None: