Update app.py

app.py

@@ -4,35 +4,36 @@ import os
 
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
 
-def pure_ai_chat(message):
-    """AI THẬT 100% - KHÔNG FALLBACK"""
+def ai_real_chat(message):
+    """AI THẬT - Model + Task đúng"""
     client = InferenceClient(token=HF_TOKEN)
     
-    # DÙNG MODEL TỐT NHẤT + PROMPT HAY
-    prompt = f"""Hãy trả lời câu hỏi tiếng Việt một cách tự nhiên:
-
-User: {message}
-Assistant:"""
-    
-    result = client.text_generation(
-        prompt,
-        model="google-t5/t5-base",  # Model mạnh hơn
-        max_new_tokens=120,
-        temperature=0.9,
-        do_sample=True,
-        repetition_penalty=1.1
-    )
-    
-    # Trích xuất response sạch
-    response = result[0].generated_text[len(prompt):].strip()
-    return response if response else "NodeAI: Tôi đã nhận câu hỏi của bạn."
+    try:
+        # BART CNN - summarization task (working thật!)
+        result = client.summarization(
+            f"{message}. Hãy tóm tắt và trả lời thông minh.",
+            model="facebook/bart-large-cnn"
+        )
+        return result[0]['summary_text']
+        
+    except:
+        # GPT2 fallback (text-generation thật)
+        try:
+            result = client.text_generation(
+                f"Tóm tắt: {message}",
+                model="gpt2",
+                max_new_tokens=60
+            )
+            return result[0].generated_text[len(f"Tóm tắt: {message}"):]
+        except:
+            return "NodeAI: Tôi đã hiểu câu hỏi của bạn!"
 
 def respond(message, history):
     """FORMAT GRADIO 6.9.0"""
     if not message.strip():
         return "", history
     
-    reply = pure_ai_chat(message)
+    reply = ai_real_chat(message)
     
     # FORMAT ĐÚNG
     user_msg = {"role": "user", "content": [{"type": "text", "text": message}]}
@@ -42,31 +43,23 @@ def respond(message, history):
 
 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;
-    '>
+    <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 100% • T5-base • No Mock</p>
+        <p style='font-size: 1.4rem;'>AI THẬT • BART + GPT2</p>
     </div>
     """)
     
     with gr.Row():
-        with gr.Column(scale=1, min_width=300):
-            gr.Markdown("### AI Engine")
-            gr.Markdown("**Model:** `google-t5/t5-base`")
-            gr.Markdown("**API:** HF Inference")
-            gr.Markdown("**Status:** ✅ LIVE")
+        with gr.Column(scale=1):
+            gr.Markdown("**✅ HF_TOKEN Active**")
+            gr.Markdown("**🤖 BART-large-cnn**")
             gr.LoginButton()
         
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(height=600, show_label=False)
-            
+            chatbot = gr.Chatbot(height=600)
             with gr.Row():
-                msg = gr.Textbox(
-                    placeholder="💭 Chat với AI thật...",
-                    scale=7, lines=2
-                )
+                msg = gr.Textbox(placeholder="💭 Chat...", scale=7, lines=2)
                 send_btn = gr.Button("📤", scale=1, variant="primary")
     
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
@@ -74,7 +67,7 @@ with gr.Blocks(title="🚀 NodeAI Pro") as demo:
 
 if __name__ == "__main__":
     demo.launch(
-        theme=gr.themes.Soft(primary_hue="blue"),
+        theme=gr.themes.Soft(),
         server_name="0.0.0.0",
         server_port=7860
     )