Update app.py

app.py

@@ -2,103 +2,115 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# HF Token
+# Chỉ dùng HF_TOKEN
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-MODEL = "gpt2"
+MODEL = "gpt2"  # Model ổn định nhất
 
-def generate_response(message):
-    """Simple text generation"""
+def smart_chat(message):
+    """AI thông minh + fallback"""
     try:
-        from huggingface_hub import InferenceClient
         client = InferenceClient(token=HF_TOKEN, model=MODEL)
-        prompt = f"Q: {message}\nA:"
+        prompt = f"Human: {message}\nAI:"
         response = client.text_generation(
-            prompt, 
-            max_new_tokens=60,
+            prompt,
+            max_new_tokens=80,
             temperature=0.8,
-            do_sample=True
+            do_sample=True,
+            pad_token_id=50256
         )
-        return response[0].generated_text[len(prompt):].strip() or "🤖 Chào bạn!"
+        ai_reply = response[0].generated_text[len(prompt):].strip()
+        return ai_reply if ai_reply else f"🤖 {message[:30]}..."
     except:
-        return "🤖 NodeAI sẵn sàng hỗ trợ bạn!"
+        # Smart fallback responses
+        msg = message.lower()
+        responses = {
+            "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! ✨",
+            "default": "🤖 Hiểu rồi! Bạn cần giúp gì tiếp theo?"
+        }
+        return responses.get(msg.split()[0], responses["default"])
 
 def respond(message, history):
-    """Chat handler - ĐÚNG FORMAT GRADIO 6.9.0"""
-    reply = generate_response(message)
+    """Chat handler Gradio 6.9.0"""
+    if not message.strip():
+        return "", history
     
-    # ✅ FORMAT CHUẨN: list of dicts với role/content
-    new_history = history + [{
-        "role": "user", 
-        "content": [{"type": "text", "text": message}]
-    }, {
-        "role": "assistant", 
-        "content": [{"type": "text", "text": reply}]
-    }]
-    
-    return "", new_history
+    reply = smart_chat(message)
+    history.append([message, reply])
+    return "", history
 
 with gr.Blocks(title="🚀 NodeAI Pro") as demo:
-    # Header đẹp
+    # Header Pro
     gr.HTML("""
-    <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
-                color: white; padding: 2rem; text-align: center; border-radius: 20px; margin-bottom: 2rem;">
-        <h1 style="font-size: 2.5rem; margin: 0;">🚀 NodeAI Pro</h1>
-        <p style="font-size: 1.2rem; opacity: 0.9;">AI Chatbot Tiếng Việt Siêu Thông Minh</p>
-        <div style="background: rgba(255,255,255,0.2); padding: 0.8rem 1.5rem; 
-                    border-radius: 25px; display: inline-block; font-weight: bold;">
-            ✅ GPT-2 | Zero Config | Production Ready
+    <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>
     </div>
     """)
     
     with gr.Row():
-        # Sidebar gọn nhẹ
-        with gr.Column(scale=1, min_width=300):
-            with gr.Accordion("⚙️ Cài đặt", open=True):
-                gr.Markdown("**Model:** `gpt2` (100% stable)")
-                gr.Markdown("**Tokens:** 60 | **Temp:** 0.8")
+        # 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**")
             
-            with gr.Accordion("🔐 Hugging Face"):
+            with gr.Accordion("🔐 HF Token"):
                 gr.HTML("""
-                <div style="text-align: center; padding: 1.5rem; 
-                           background: linear-gradient(135deg, #10b981, #059669); 
-                           color: white; border-radius: 12px;">
+                <div style='
+                    text-align: center; padding: 2rem; 
+                    background: linear-gradient(135deg, #10b981, #059669); 
+                    color: white; border-radius: 15px;
+                '>
                     <h3>✅ Connected</h3>
-                    <p>HF Token Active</p>
+                    <p>HF_TOKEN detected</p>
                 </div>
                 """)
                 gr.LoginButton()
             
-            with gr.Accordion("📚 Lịch sử chat"):
-                history_view = gr.Chatbot(height=250)
-                gr.Button("🗑️ Xóa lịch sử", scale=1, variant="stop")
+            with gr.Accordion("📚 Recent Chats"):
+                history_view = gr.Chatbot(height=300)
+                gr.Button("🗑️ Clear", variant="stop")
         
-        # Main chat đẹp
+        # Main Chat Pro
         with gr.Column(scale=4):
             chatbot = gr.Chatbot(
-                height=500,
-                show_label=False,
-                avatar_images=("🧑‍💻", "🤖")
+                height=600,
+                avatar_images=("🧑", "🤖"),
+                show_label=False
             )
             
             with gr.Row():
-                message_input = gr.Textbox(
-                    placeholder="💭 Hỏi NodeAI bất cứ điều gì... (Enter)",
+                msg = gr.Textbox(
+                    placeholder="💭 Hỏi NodeAI: 'Viết code', 'Giải thích AI'... (Enter)",
                     scale=7,
                     lines=2,
                     show_label=False
                 )
-                send_button = gr.Button("📤", scale=1, variant="primary")
+                send_btn = gr.Button("📤", scale=1, variant="primary")
     
-    # Events đơn giản
-    message_input.submit(respond, [message_input, chatbot], [message_input, chatbot])
-    send_button.click(respond, [message_input, chatbot], [message_input, chatbot])
+    # Events
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    send_btn.click(respond, [msg, chatbot], [msg, chatbot])
 
 if __name__ == "__main__":
     demo.launch(
         theme=gr.themes.Soft(primary_hue="blue"),
-        server_name="0.0.0.0", 
+        server_name="0.0.0.0",
         server_port=7860,
-        share=False,
-        show_error=True
+        share=False
     )