Update app.py

app.py

@@ -177,6 +177,13 @@ def delete_chat_cb(selected_label):
     selected = labels[0] if labels else None
     return gr.update(choices=labels, value=selected), []
 
+FRONTEND_KEYWORDS = [
+    "react", "vue", "angular", "html", "css", "javascript", "tailwind", "recharts", "typescript"
+]
+
+def is_frontend_request(user_text: str) -> bool:
+    text_lower = user_text.lower()
+    return any(kw in text_lower for kw in FRONTEND_KEYWORDS)
 # --- 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)
@@ -185,19 +192,27 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
         labels, _ = list_sessions()
         selected_label = next((lbl for lbl in labels if lbl.startswith(f"{sid} ")), None)
 
+    # Save user message
     add_message(sid, "user", user_text)
     update_session_title_if_needed(sid, user_text)
 
-    api_messages = build_api_messages(sid, system_message)
-    display_msgs = chatbot_msgs[:]  # preserve current chat
-
-    # Show user's message immediately
+    display_msgs = chatbot_msgs[:]
     display_msgs.append({"role": "user", "content": user_text})
-    # Add assistant typing placeholder
+
+    # Check for frontend-heavy request
+    if is_frontend_request(user_text):
+        apology = "⚠️ I'm a backend-focused assistant and cannot provide frontend code."
+        display_msgs.append({"role": "assistant", "content": apology})
+        add_message(sid, "assistant", apology)
+        yield (display_msgs, "", selected_label)
+        return
+
+    # Normal backend response
     display_msgs.append({"role": "assistant", "content": "…"})
     yield (display_msgs, "", selected_label)
 
     client = get_client(model_choice)
+    api_messages = build_api_messages(sid, system_message)
     partial = ""
     try:
         for chunk in client.chat_completion(