Update app.py

app.py

@@ -2,97 +2,89 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# 1. Khởi tạo Client (Sử dụng Secret HF_TOKEN)
+# 1. Cấu hình Model & Token (Sử dụng Llama-3.2 để Pro hơn)
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-# Sử dụng gpt2 để sinh văn bản (text-generation)
-client = InferenceClient("gpt2", token=HF_TOKEN)
+model_id = "meta-llama/Llama-3.2-1B-Instruct" 
+client = InferenceClient(model=model_id, token=HF_TOKEN)
 
-# 2. CSS chuyên nghiệp
+# 2. CSS tinh chỉnh cho Gradio 6.x
 custom_css = """
-.header-pro {
-    background: linear-gradient(135deg, #1e3a8a 0%, #7e22ce 100%);
-    padding: 25px;
-    border-radius: 12px;
-    color: white;
-    text-align: center;
-    margin-bottom: 20px;
-}
-#chatbot-style {
-    border-radius: 10px !important;
-    border: 1px solid #e5e7eb !important;
+.header-pro { 
+    background: linear-gradient(135deg, #000428 0%, #004e92 100%); 
+    padding: 20px; border-radius: 15px; color: white; text-align: center; margin-bottom: 20px;
 }
+.footer-info { font-size: 0.8em; text-align: center; color: #666; margin-top: 10px; }
 """
 
-def respond(message, history):
-    # Định dạng msg đúng yêu cầu Gradio 6.9.0 của bạn:
-    # [{"role": "user", "content": [{"type": "text", "text": msg}]}]
-    
-    # Lấy text từ message (Gradio 6 truyền list các dict tin nhắn)
-    if isinstance(message, list):
-        user_text = message[-1]["content"][0]["text"]
+def chat_fn(message, history):
+    # Trích xuất text từ định dạng tin nhắn Gradio 6
+    if isinstance(message, dict):
+        user_input = message.get("text", "")
+    elif isinstance(message, list):
+        user_input = message[-1]["content"][0]["text"]
     else:
-        user_text = message
+        user_input = str(message)
+
+    # Khởi tạo history nếu trống
+    if history is None: history = []
 
-    # Cấu trúc tin nhắn mới để lưu vào history
-    user_msg = {"role": "user", "content": [{"type": "text", "text": user_text}]}
+    # Format tin nhắn người dùng
+    user_msg = {"role": "user", "content": [{"type": "text", "text": user_input}]}
     history.append(user_msg)
     
-    # Gọi API Inference thực tế
+    # Gọi API Inference
     try:
-        # GPT-2 trả về text nối tiếp từ input
-        api_output = client.text_generation(user_text, max_new_tokens=100)
-        bot_text = api_output
+        # Sử dụng chat_completion để có kết quả thông minh hơn
+        response = client.chat_completion(
+            messages=[{"role": "user", "content": user_input}],
+            max_tokens=500,
+            temperature=0.7
+        )
+        bot_res = response.choices[0].message.content
     except Exception as e:
-        bot_text = f"❌ API Error: {str(e)}"
-    
-    bot_msg = {"role": "assistant", "content": [{"type": "text", "text": bot_text}]}
+        bot_res = f"⚠️ Lỗi kết nối: Hãy đảm bảo bạn đã add HF_TOKEN vào Settings. (Chi tiết: {str(e)})"
+
+    # Format tin nhắn Bot
+    bot_msg = {"role": "assistant", "content": [{"type": "text", "text": bot_res}]}
     history.append(bot_msg)
     
     return "", history
 
 with gr.Blocks(title="NodeAI Pro") as demo:
-    # Header UI Pro
     gr.HTML("""
         <div class="header-pro">
             <h1 style='margin:0'>🚀 NodeAI Pro</h1>
-            <p style='margin:0; opacity: 0.8;'>Hệ thống Trợ lý AI Thông minh (V6.9.0)</p>
+            <p style='margin:0'>High-Performance AI Assistant</p>
         </div>
     """)
     
     with gr.Row():
-        # Sidebar
         with gr.Column(scale=1):
-            with gr.Group():
-                gr.Markdown("### 🛠️ Hệ thống")
-                gr.LoginButton()
-                gr.Markdown("---")
-                gr.Markdown(f"**Token:** {'✅ OK' if HF_TOKEN else '⚠️ Thiếu'}")
-                gr.Markdown("**Model:** `gpt2` (Real API)")
-                gr.Markdown("**Trạng thái:** Sẵn sàng")
-        
-        # Chat area
+            gr.Markdown("### 🛠️ Core Engine")
+            gr.LoginButton()
+            gr.Markdown(f"**Engine:** `Llama-3.2-1B`")
+            gr.Markdown(f"**API Status:** {'🟢 Online' if HF_TOKEN else '🔴 Offline'}")
+            clear_btn = gr.Button("🗑️ Reset Session", variant="secondary")
+            
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(
-                label="Hội thoại",
-                elem_id="chatbot-style",
-                height=550
-            )
+            # Không dùng type="messages" ở đây để tránh TypeError trên một số môi trường
+            chatbot = gr.Chatbot(label="NodeAI Chat Console", height=500)
             
             with gr.Row():
                 msg_input = gr.Textbox(
-                    placeholder="Nhập yêu cầu cho NodeAI Pro...",
+                    placeholder="Nhập câu hỏi...",
                     show_label=False,
-                    scale=9
+                    scale=8
                 )
-                submit_btn = gr.Button("Gửi", variant="primary", scale=1)
+                submit_btn = gr.Button("Send", variant="primary", scale=2)
             
-            clear = gr.Button("🗑️ Xóa hội thoại", size="sm")
+            gr.HTML("<div class='footer-info'>NodeAI Pro v1.0.9 | Powered by Hugging Face Inference API</div>")
 
-    # Sự kiện điều khiển
-    msg_input.submit(respond, [msg_input, chatbot], [msg_input, chatbot])
-    submit_btn.click(respond, [msg_input, chatbot], [msg_input, chatbot])
-    clear.click(lambda: [], None, chatbot)
+    # Xử lý Logic
+    msg_input.submit(chat_fn, [msg_input, chatbot], [msg_input, chatbot])
+    submit_btn.click(chat_fn, [msg_input, chatbot], [msg_input, chatbot])
+    clear_btn.click(lambda: None, None, chatbot)
 
 if __name__ == "__main__":
-    # CSS được đẩy vào launch() theo chuẩn Gradio 6
+    # Đưa CSS vào launch() để fix UserWarning
     demo.launch(css=custom_css)