Update app.py

app.py

@@ -2,104 +2,98 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# Chỉ dùng HF_TOKEN
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-MODEL = "gpt2"  # Model ổn định nhất
+MODEL = "gpt2"
 
 def smart_chat(message):
-    """AI thông minh + fallback"""
+    """AI + Smart fallback"""
     try:
         client = InferenceClient(token=HF_TOKEN, model=MODEL)
-        prompt = f"Human: {message}\nAI:"
+        prompt = f"Q: {message}\nA: "
         response = client.text_generation(
             prompt,
-            max_new_tokens=80,
+            max_new_tokens=60,
             temperature=0.8,
-            do_sample=True,
-            pad_token_id=50256
+            do_sample=True
         )
-        ai_reply = response[0].generated_text[len(prompt):].strip()
-        return ai_reply if ai_reply else f"🤖 {message[:30]}..."
+        reply = response[0].generated_text[len(prompt):].strip()
+        return reply if reply else "🤖 Chào bạn!"
     except:
-        # Smart fallback responses
-        msg = message.lower()
-        responses = {
+        # Smart fallback
+        msg_lower = message.lower()
+        fallbacks = {
             "xin chào": "Chào bạn! 😊 NodeAI sẵn sàng hỗ trợ!",
-            "hello": "Xin chào! Rất vui được chat với bạn! 🤗",
-            "bạn là ai": "🚀 Mình là **NodeAI Pro** - AI trợ lý siêu thông minh!",
-            "làm gì": "Mình chat 24/7, giải đáp mọi thắc mắc! 😄",
-            "cảm ơn": "Không có gì! Rất vui được giúp bạn! ✨",
+            "hello": "Xin chào! Rất vui được chat với bạn!",
+            "bạn là ai": "🚀 Mình là NodeAI Pro - AI trợ lý thông minh!",
             "default": "🤖 Hiểu rồi! Bạn cần giúp gì tiếp theo?"
         }
-        return responses.get(msg.split()[0], responses["default"])
+        for key, value in fallbacks.items():
+            if key in msg_lower:
+                return value
+        return fallbacks["default"]
 
 def respond(message, history):
-    """Chat handler Gradio 6.9.0"""
+    """✅ CORRECT GRADIO 6.9.0 FORMAT"""
     if not message.strip():
         return "", history
     
     reply = smart_chat(message)
-    history.append([message, reply])
-    return "", history
+    
+    # ✅ FORMAT ĐÚNG: list of dicts role/content
+    new_history = history + [{
+        "role": "user",
+        "content": [{"type": "text", "text": message}]
+    }, {
+        "role": "assistant", 
+        "content": [{"type": "text", "text": reply}]
+    }]
+    
+    return "", new_history
 
+# UI Pro
 with gr.Blocks(title="🚀 NodeAI Pro") as demo:
-    # Header Pro
     gr.HTML("""
-    <div style='
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
-        color: white; padding: 3rem; text-align: center; border-radius: 25px; 
-        margin: 2rem 0; box-shadow: 0 25px 50px rgba(102,126,234,0.4);
-    '>
-        <h1 style='font-size: 3rem; margin: 0;'>🚀 NodeAI Pro</h1>
-        <p style='font-size: 1.5rem; opacity: 0.95;'>AI Chatbot Tiếng Việt • HF_TOKEN Only</p>
-        <div style='
-            background: rgba(255,255,255,0.2); padding: 1rem 2.5rem; 
-            border-radius: 50px; display: inline-block; font-weight: bold;
-        '>
-            ✅ HF_TOKEN Active | GPT-2 + Smart AI
+    <div style='background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
+                color: white; padding: 2.5rem; text-align: center; border-radius: 20px;'>
+        <h1 style='font-size: 2.8rem;'>🚀 NodeAI Pro</h1>
+        <p style='font-size: 1.3rem;'>AI Chatbot Tiếng Việt • HF_TOKEN Only</p>
+        <div style='background: rgba(255,255,255,0.2); padding: 1rem 2rem; 
+                    border-radius: 30px; display: inline-block; font-weight: bold;'>
+            ✅ GPT-2 Active
         </div>
     </div>
     """)
     
     with gr.Row():
-        # Sidebar Pro
-        with gr.Column(scale=1, min_width=350):
-            with gr.Accordion("🤖 AI Engine", open=True):
-                gr.Markdown("**Model:** `gpt2` (stable)")
-                gr.Markdown("**Tokens:** 80 | **Temp:** 0.8")
-                gr.Markdown("**Status:** ✅ **LIVE**")
+        # Sidebar
+        with gr.Column(scale=1, min_width=320):
+            with gr.Accordion("⚙️ Settings", open=True):
+                gr.Markdown("**Model:** `gpt2`")
+                gr.Markdown("**Status:** ✅ LIVE")
             
             with gr.Accordion("🔐 HF Token"):
-                gr.HTML("""
-                <div style='
-                    text-align: center; padding: 2rem; 
-                    background: linear-gradient(135deg, #10b981, #059669); 
-                    color: white; border-radius: 15px;
-                '>
+                status = gr.HTML("""
+                <div style='padding: 1.5rem; text-align: center; 
+                           background: linear-gradient(135deg, #10b981, #059669); 
+                           color: white; border-radius: 12px;'>
                     <h3>✅ Connected</h3>
-                    <p>HF_TOKEN detected</p>
                 </div>
                 """)
                 gr.LoginButton()
-            
-            with gr.Accordion("📚 Recent Chats"):
-                history_view = gr.Chatbot(height=300)
-                gr.Button("🗑️ Clear", variant="stop")
         
-        # Main Chat Pro
+        # Main Chat
         with gr.Column(scale=4):
             chatbot = gr.Chatbot(
-                height=600,
+                height=550,
                 avatar_images=("🧑", "🤖"),
                 show_label=False
             )
             
             with gr.Row():
                 msg = gr.Textbox(
-                    placeholder="💭 Hỏi NodeAI: 'Viết code', 'Giải thích AI'... (Enter)",
+                    placeholder="💭 Hỏi NodeAI bất cứ điều gì...",
                     scale=7,
-                    lines=2,
-                    show_label=False
+                    lines=2
                 )
                 send_btn = gr.Button("📤", scale=1, variant="primary")