Update app.py

app.py

@@ -2,72 +2,36 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
+# 1. Khởi tạo Client với GPT-2
+# Token được lấy tự động từ Secret HF_TOKEN của Space
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
+client = InferenceClient("gpt2", token=HF_TOKEN)
 
-def ai_real_chat(message):
-    """AI THẬT - Model + Task đúng"""
-    client = InferenceClient(token=HF_TOKEN)
-    
+def respond(message, history):
+    # 2. Xử lý định dạng tin nhắn của Gradio 6.9.0
+    # Trích xuất text từ cấu trúc: [{"role": "user", "content": [{"type": "text", "text": msg}]}]
     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!"
+        user_prompt = message[0]["content"][0]["text"]
+    except (IndexKey, KeyError, TypeError):
+        user_prompt = str(message)
 
-def respond(message, history):
-    """FORMAT GRADIO 6.9.0"""
-    if not message.strip():
-        return "", history
-    
-    reply = ai_real_chat(message)
-    
-    # FORMAT ĐÚNG
-    user_msg = {"role": "user", "content": [{"type": "text", "text": message}]}
-    ai_msg = {"role": "assistant", "content": [{"type": "text", "text": reply}]}
-    
-    return "", history + [user_msg, ai_msg]
+    # 3. Gọi API Inference (không dùng fallback)
+    response = ""
+    # Stream kết quả từ GPT-2
+    for token in client.text_generation(user_prompt, max_new_tokens=100, stream=True):
+        response += token
+        yield response
 
-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 • BART + GPT2</p>
-    </div>
-    """)
-    
-    with gr.Row():
-        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)
-            with gr.Row():
-                msg = gr.Textbox(placeholder="💭 Chat...", scale=7, lines=2)
-                send_btn = gr.Button("📤", scale=1, variant="primary")
-    
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    send_btn.click(respond, [msg, chatbot], [msg, chatbot])
+# 4. Giao diện UI Pro với ChatInterface
+demo = gr.ChatInterface(
+    fn=respond,
+    title="🤖 NodeAI Pro",
+    description="Hệ thống trợ lý trí tuệ nhân tạo thế hệ mới",
+    type="messages", # Bắt buộc cho định dạng tin nhắn mới
+    fill_height=True,
+    # Thêm nút đăng nhập HF để quản lý token status
+    head=gr.LoginButton() if not HF_TOKEN else None 
+)
 
 if __name__ == "__main__":
-    demo.launch(
-        theme=gr.themes.Soft(),
-        server_name="0.0.0.0",
-        server_port=7860
-    )
+    demo.launch()