Update app.py

app.py

@@ -5,82 +5,79 @@ import os
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
 
 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
+    """AI THẬT - HF Inference API"""
     try:
-        # 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,
+        client = InferenceClient(token=HF_TOKEN)
+        # Text generation model ổn định
+        result = client.text_generation(
+            f"Q: {message}\nA:",
+            model="gpt2",
+            max_new_tokens=80,
             temperature=0.7,
-            stream=False
+            do_sample=True
         )
-        
-        return response.choices[0].message.content
-        
+        return result[0].generated_text[9:].strip()
     except:
-        # 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é!"
+        return f"🤖 {message[:50]}..."
 
 def respond(message, history):
-    """Gradio format đúng"""
+    """✅ FORMAT GRADIO 6.9.0 CHUẨN XÁC"""
+    if not message:
+        return "", history
+    
     reply = ai_chat(message)
-    history.append([message, reply])
-    return "", history
+    
+    # GRADIO 6.9.0: list of dicts {"role": "user", "content": "..."}
+    new_msg = {
+        "role": "user", 
+        "content": [{"type": "text", "text": message}]
+    }
+    new_reply = {
+        "role": "assistant",
+        "content": [{"type": "text", "text": reply}]
+    }
+    
+    return "", history + [new_msg, new_reply]
 
 with gr.Blocks(title="🚀 NodeAI Pro") as demo:
     gr.HTML("""
     <div style='background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
-                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;'>
-            ✅ Live HF Token
-        </div>
+                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 THẬT • HF Inference API</p>
     </div>
     """)
     
     with gr.Row():
         with gr.Column(scale=1, min_width=300):
-            gr.Markdown("### ⚙️ AI Engine")
-            gr.Markdown("**Model:** HF Inference API")
-            gr.Markdown("**Status:** ✅ LIVE")
+            gr.Markdown("# ⚙️ Status")
+            gr.Markdown("**✅ HF_TOKEN Active**")
+            gr.Markdown("**🤖 GPT-2 Live**")
             gr.LoginButton()
         
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(height=600, avatar_images=("🧑", "🤖"))
+            chatbot = gr.Chatbot(
+                height=600,
+                show_label=False,
+                avatar_images=("🧑", "🤖")
+            )
             
             with gr.Row():
                 msg = gr.Textbox(
-                    placeholder="💭 Chat với NodeAI thật...",
-                    scale=7, lines=2
-                )
-                gr.Button("📤", scale=1, variant="primary").click(
-                    respond, [msg, chatbot], [msg, chatbot]
+                    placeholder="💭 Chat với AI thật...",
+                    scale=7,
+                    lines=2
                 )
+                send_btn = gr.Button("📤", scale=1, variant="primary")
     
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    # Events
+    msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
+    send_btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
 
-demo.launch(
-    theme=gr.themes.Soft(primary_hue="blue"),
-    server_name="0.0.0.0", 
-    server_port=7860
-)
+if __name__ == "__main__":
+    demo.launch(
+        theme=gr.themes.Soft(primary_hue="blue"),
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )