Update app.py

app.py

@@ -177,7 +177,7 @@ 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, dataset_choice):
+def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens, temperature, top_p, model_choice):
     sid = label_to_id(selected_label)
     if sid is None:
         sid = create_session("New chat")
@@ -187,20 +187,12 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
     add_message(sid, "user", user_text)
     update_session_title_if_needed(sid, user_text)
 
-    # Load selected dataset (used internally for reference / semantic checks)
-    dataset = load_dataset_by_name(dataset_choice)
-
     api_messages = build_api_messages(sid, system_message)
     display_msgs = get_messages(sid)
-    display_msgs.append({"role": "assistant", "content": ""})
 
-    # Block excessive frontend requests
-    if "frontend" in user_text.lower() and "too much" in user_text.lower():
-        warning = "⚠️ I'm a backend-focused assistant and cannot provide excessive frontend code."
-        display_msgs[-1]["content"] = warning
-        yield (display_msgs, "", selected_label)
-        add_message(sid, "assistant", warning)
-        return
+    # Add initial empty assistant message for typing indicator
+    display_msgs.append({"role": "assistant", "content": "…"})
+    yield (display_msgs, "", selected_label)
 
     client = get_client(model_choice)
     partial = ""
@@ -223,6 +215,7 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
         display_msgs[-1]["content"] = err
         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: