Update app.py

app.py

@@ -2,103 +2,103 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
+# HF Token
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
 MODEL = "gpt2"
 
-client = InferenceClient(token=HF_TOKEN, model=MODEL)
-
-def chat(message, history):
-    """Safe chat function"""
+def generate_response(message):
+    """Simple text generation"""
     try:
+        from huggingface_hub import InferenceClient
+        client = InferenceClient(token=HF_TOKEN, model=MODEL)
         prompt = f"Q: {message}\nA:"
         response = client.text_generation(
-            prompt,
-            max_new_tokens=80,
+            prompt, 
+            max_new_tokens=60,
             temperature=0.8,
-            do_sample=True,
-            pad_token_id=50256
+            do_sample=True
         )
-        reply = response[0].generated_text[len(prompt):].strip()
-        return reply or "🤖 Chào bạn! Hỏi mình nhé!"
+        return response[0].generated_text[len(prompt):].strip() or "🤖 Chào bạn!"
     except:
-        return "🤖 NodeAI sẵn sàng hỗ trợ!"
+        return "🤖 NodeAI sẵn sàng hỗ trợ bạn!"
 
 def respond(message, history):
-    """Chat handler - đúng format Gradio 6.9.0"""
-    # ✅ Chatbot format: [[{"role": "user", "content": msg}, {"role": "assistant", "content": reply}]]
-    reply = chat(message, history)
+    """Chat handler - ĐÚNG FORMAT GRADIO 6.9.0"""
+    reply = generate_response(message)
     
-    new_message = {
-        "user": [{"role": "user", "content": message}],
-        "assistant": [{"role": "assistant", "content": reply}]
-    }
+    # ✅ 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}]
+    }]
     
-    history.append(new_message)
-    return "", history
+    return "", new_history
 
-with gr.Blocks(title="NodeAI Pro") as demo:
-    # Pro Header
+with gr.Blocks(title="🚀 NodeAI Pro") as demo:
+    # Header đẹp
     gr.HTML("""
-    <div style='background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
-                color: white; text-align: center; padding: 40px; border-radius: 20px; margin: 20px 0;'>
-        <h1 style='font-size: 48px; margin: 0;'>🚀 NodeAI Pro</h1>
-        <p style='font-size: 24px; opacity: 0.95;'>AI Chatbot Tiếng Việt Siêu Thông Minh</p>
-        <div style='background: rgba(255,255,255,0.2); padding: 12px 25px; 
-                    border-radius: 25px; display: inline-block; font-weight: bold;'>
-            ✅ GPT-2 Active | Zero Config
+    <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>
     </div>
     """)
     
     with gr.Row():
-        # Sidebar
-        with gr.Column(scale=1, min_width=320):
-            gr.Markdown("## ⚙️ **Cài đặt**")
-            
-            with gr.Accordion("🤖 Model", open=True):
-                gr.Markdown("**gpt2** (Siêu ổn định)")
-                gr.Markdown("**Tokens:** 80 | **Temp:** 0.8")
+        # 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")
             
-            with gr.Accordion("👤 Kết nối"):
+            with gr.Accordion("🔐 Hugging Face"):
                 gr.HTML("""
-                <div style='text-align: center; padding: 20px; 
+                <div style="text-align: center; padding: 1.5rem; 
                            background: linear-gradient(135deg, #10b981, #059669); 
-                           color: white; border-radius: 15px; margin: 10px 0;'>
-                    <h3>✅ HF Token OK</h3>
-                    <p>Ready to chat!</p>
+                           color: white; border-radius: 12px;">
+                    <h3>✅ Connected</h3>
+                    <p>HF Token Active</p>
                 </div>
                 """)
                 gr.LoginButton()
             
-            gr.Markdown("## 📚 **Lịch sử**")
-            history_display = gr.Chatbot(height=280)
-            gr.Button("🗑️ Xóa lịch sử", variant="stop")
+            with gr.Accordion("📚 Lịch sử chat"):
+                history_view = gr.Chatbot(height=250)
+                gr.Button("🗑️ Xóa lịch sử", scale=1, variant="stop")
         
-        # Main Chat
+        # Main chat đẹp
         with gr.Column(scale=4):
             chatbot = gr.Chatbot(
-                height=550,
-                avatar_images=("🧑‍💻", "🤖"),
-                show_label=False
+                height=500,
+                show_label=False,
+                avatar_images=("🧑‍💻", "🤖")
             )
             
             with gr.Row():
-                msg = gr.Textbox(
-                    placeholder="💭 Hỏi NodeAI bất cứ điều gì... (Enter để gửi)",
+                message_input = gr.Textbox(
+                    placeholder="💭 Hỏi NodeAI bất cứ điều gì... (Enter)",
                     scale=7,
-                    lines=2
+                    lines=2,
+                    show_label=False
                 )
-                send_btn = gr.Button("📤 Gửi", scale=1, variant="primary")
+                send_button = gr.Button("📤", scale=1, variant="primary")
     
-    # Events
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    send_btn.click(respond, [msg, chatbot], [msg, chatbot])
+    # Events đơn giản
+    message_input.submit(respond, [message_input, chatbot], [message_input, chatbot])
+    send_button.click(respond, [message_input, chatbot], [message_input, 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,  # ✅ Fix asyncio error
-        ssr_mode=False  # ✅ Fix SSR warnings
+        share=False,
+        show_error=True
     )