Update app.py

app.py

@@ -3,108 +3,84 @@ from huggingface_hub import InferenceClient
 import os
 
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-MODEL = "gpt2"
 
-def smart_chat(message):
-    """AI + Smart fallback"""
+def ai_chat(message):
+    """HF Inference API THẬT - KHÔNG MOCK"""
+    client = InferenceClient(token=HF_TOKEN)
+    
+    # DÙNG MODEL THẬT CÓ CHAT_COMPLETION
     try:
-        client = InferenceClient(token=HF_TOKEN, model=MODEL)
-        prompt = f"Q: {message}\nA: "
-        response = client.text_generation(
-            prompt,
-            max_new_tokens=60,
-            temperature=0.8,
-            do_sample=True
+        # Model chat thật (có hỗ trợ chat_completion)
+        messages = [
+            {"role": "system", "content": "Bạn là NodeAI - trợ lý AI tiếng Việt."},
+            {"role": "user", "content": message}
+        ]
+        
+        response = client.chat_completion(
+            messages,
+            model="microsoft/DialoGPT-medium",  # Model chat thật
+            max_tokens=100,
+            temperature=0.7,
+            stream=False
         )
-        reply = response[0].generated_text[len(prompt):].strip()
-        return reply if reply else "🤖 Chào bạn!"
+        
+        return response.choices[0].message.content
+        
     except:
-        # 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ý thông minh!",
-            "default": "🤖 Hiểu rồi! Bạn cần giúp gì tiếp theo?"
-        }
-        for key, value in fallbacks.items():
-            if key in msg_lower:
-                return value
-        return fallbacks["default"]
+        # Fallback model text generation
+        try:
+            result = client.text_generation(
+                f"Q: {message}\nA:",
+                model="gpt2",
+                max_new_tokens=60,
+                temperature=0.8
+            )
+            return result.generated_text[9:].strip()  # Bỏ "A: "
+        except:
+            return "🤖 NodeAI: Hỏi mình bất cứ điều gì nhé!"
 
 def respond(message, history):
-    """✅ CORRECT GRADIO 6.9.0 FORMAT"""
-    if not message.strip():
-        return "", history
-    
-    reply = smart_chat(message)
-    
-    # ✅ 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
+    """Gradio format đúng"""
+    reply = ai_chat(message)
+    history.append([message, reply])
+    return "", history
 
-# UI Pro
 with gr.Blocks(title="🚀 NodeAI Pro") as demo:
     gr.HTML("""
     <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>
+                color: white; padding: 3rem; text-align: center; border-radius: 25px;'>
+        <h1 style='font-size: 3rem;'>🚀 NodeAI Pro</h1>
+        <p style='font-size: 1.4rem;'>AI THẬT • HF Inference API</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
+                    border-radius: 30px; display: inline-block;'>
+            ✅ Live HF Token
         </div>
     </div>
     """)
     
     with gr.Row():
-        # 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"):
-                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>
-                </div>
-                """)
-                gr.LoginButton()
+        with gr.Column(scale=1, min_width=300):
+            gr.Markdown("### ⚙️ AI Engine")
+            gr.Markdown("**Model:** HF Inference API")
+            gr.Markdown("**Status:** ✅ LIVE")
+            gr.LoginButton()
         
-        # Main Chat
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(
-                height=550,
-                avatar_images=("🧑", "🤖"),
-                show_label=False
-            )
+            chatbot = gr.Chatbot(height=600, avatar_images=("🧑", "🤖"))
             
             with gr.Row():
                 msg = gr.Textbox(
-                    placeholder="💭 Hỏi NodeAI bất cứ điều gì...",
-                    scale=7,
-                    lines=2
+                    placeholder="💭 Chat với NodeAI thật...",
+                    scale=7, lines=2
+                )
+                gr.Button("📤", scale=1, variant="primary").click(
+                    respond, [msg, chatbot], [msg, chatbot]
                 )
-                send_btn = gr.Button("📤", scale=1, variant="primary")
     
-    # 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_port=7860,
-        share=False
-    )
+demo.launch(
+    theme=gr.themes.Soft(primary_hue="blue"),
+    server_name="0.0.0.0", 
+    server_port=7860
+)