Update app.py

app.py

@@ -1,13 +1,17 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-import os
 
-MODEL = "microsoft/DialoGPT-medium"  # Model HF Inference API
+MODEL = "mistralai/Mistral-7B-Instruct-v0.1"
+
+# In-memory chat history
+chat_history = {}
+
+def get_user_id(hf_token):
+    return hash(hf_token.token) % 1000000 if hf_token and hf_token.token else None
 
 def respond(message, history, system_message, max_tokens, temperature, top_p, hf_token):
-    """HF InferenceClient với OAuth login"""
     if not hf_token or not hf_token.token:
-        yield "❌ Vui lòng đăng nhập Hugging Face (Sidebar)!"
+        yield "❌ Vui lòng đăng nhập Hugging Face (nút Login sidebar)!"
         return
     
     try:
@@ -18,31 +22,15 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, hf
         
         response = ""
         for message in client.chat_completion(
-            messages,
-            max_tokens=max_tokens,
-            stream=True,
-            temperature=temperature,
-            top_p=top_p,
+            messages, max_tokens=max_tokens, stream=True,
+            temperature=temperature, top_p=top_p
         ):
-            if message.choices[0].delta.content:
+            if message.choices and message.choices[0].delta.content:
                 response += message.choices[0].delta.content
                 yield response
+                
     except Exception as e:
-        yield f"❌ Lỗi model: {str(e)}\nThử model khác: mistralai/Mistral-7B-Instruct-v0.1"
-
-# User chat history (in-memory)
-chat_history = {}
-
-def get_user_id(hf_token):
-    """Lấy user_id từ token"""
-    return hash(hf_token.token) % 1000000 if hf_token and hf_token.token else None
-
-def save_chat(user_id, message, response):
-    if user_id not in chat_history:
-        chat_history[user_id] = []
-    chat_history[user_id].append({"user": message, "ai": response})
-    if len(chat_history[user_id]) > 20:
-        chat_history[user_id] = chat_history[user_id][-20:]
+        yield f"❌ Lỗi: {str(e)}"
 
 def load_history(hf_token):
     user_id = get_user_id(hf_token)
@@ -50,25 +38,29 @@ def load_history(hf_token):
         return [[h["user"], h["ai"]] for h in chat_history[user_id][-10:]]
     return []
 
-def clear_history(hf_token):
+def save_history(hf_token, message, response):
     user_id = get_user_id(hf_token)
-    if user_id:
-        chat_history[user_id] = []
-    return []
+    if user_id and hf_token.token:
+        if user_id not in chat_history:
+            chat_history[user_id] = []
+        chat_history[user_id].append({"user": message, "ai": response})
+        if len(chat_history[user_id]) > 20:
+            chat_history[user_id] = chat_history[user_id][-20:]
 
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
-    gr.Markdown("# 🚀 NodeAI - HF Login + Chat History")
+with gr.Blocks(title="NodeAI") as demo:  # ❌ NO theme ở đây
+    gr.Markdown("# 🚀 NodeAI - HF Login + History")
     
     with gr.Row():
         # Sidebar
         with gr.Column(scale=1, min_width=350):
             gr.Markdown("## 🔐 Hugging Face")
-            with gr.Row():
-                gr.LoginButton()
-                gr.LogoutButton()
+            login_btn = gr.LoginButton()  # ✅ CHỈ LoginButton
             
             gr.Markdown("## 📚 Lịch sử Chat")
-            history_display = gr.Chatbot(height=300, label="Lịch sử", avatar_images=("👤", "🤖"))
+            history_display = gr.Chatbot(
+                height=300, label="Lịch sử", 
+                avatar_images=("👤", "🤖")
+            )
             
             with gr.Row():
                 load_btn = gr.Button("📥 Load", variant="secondary")
@@ -88,12 +80,17 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
                     gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p"),
                 ],
                 title="💬 NodeAI Chat",
-                description="**Đăng nhập HF → Chat → Lưu history tự động!**"
+                description="**Đăng nhập HF sidebar → Chat → Auto save history**"
             )
     
-    # History controls
+    # Events
     load_btn.click(load_history, gr.OAuthToken(), history_display)
-    clear_btn.click(clear_history, gr.OAuthToken(), history_display)
+    clear_btn.click(lambda: [], outputs=history_display)
 
 if __name__ == "__main__":
-    demo.launch()
+    # ✅ Theme ở launch() - Gradio 6.0 syntax
+    demo.launch(
+        theme=gr.themes.Soft(primary_hue="blue"),
+        server_name="0.0.0.0",
+        server_port=7860
+    )