Update app.py

app.py

@@ -6,10 +6,8 @@ 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):
+    """Chat function - 7 args đúng chuẩn ChatInterface"""
     if not hf_token or not hf_token.token:
         yield "❌ Vui lòng đăng nhập Hugging Face (nút Login sidebar)!"
         return
@@ -17,59 +15,71 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, hf
     try:
         client = InferenceClient(token=hf_token.token, model=MODEL)
         messages = [{"role": "system", "content": system_message}]
-        messages.extend(history)
+        
+        # Convert history to messages format
+        for user_msg, assistant_msg in history:
+            messages.append({"role": "user", "content": str(user_msg)})
+            if assistant_msg:
+                messages.append({"role": "assistant", "content": str(assistant_msg)})
+        
         messages.append({"role": "user", "content": message})
         
         response = ""
-        for message in client.chat_completion(
-            messages, max_tokens=max_tokens, stream=True,
-            temperature=temperature, top_p=top_p
+        for chunk in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p
         ):
-            if message.choices and message.choices[0].delta.content:
-                response += message.choices[0].delta.content
+            if chunk.choices[0].delta.content:
+                response += chunk.choices[0].delta.content
                 yield response
                 
     except Exception as e:
-        yield f"❌ Lỗi: {str(e)}"
+        yield f"❌ Lỗi model: {str(e)}"
 
 def load_history(hf_token):
-    user_id = get_user_id(hf_token)
-    if user_id and user_id in chat_history:
-        return [[h["user"], h["ai"]] for h in chat_history[user_id][-10:]]
+    """Load history - Gradio tự inject hf_token"""
+    if not hf_token or not hf_token.token:
+        return []
+    
+    user_id = hash(hf_token.token) % 1000000
+    if user_id in chat_history:
+        return chat_history[user_id][-10:]
     return []
 
-def save_history(hf_token, message, response):
-    user_id = get_user_id(hf_token)
-    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:]
+def clear_history(hf_token):
+    """Clear history"""
+    if hf_token and hf_token.token:
+        user_id = hash(hf_token.token) % 1000000
+        chat_history[user_id] = []
+    return []
 
-with gr.Blocks(title="NodeAI") as demo:  # ❌ NO theme ở đây
+with gr.Blocks(title="NodeAI") as demo:
     gr.Markdown("# 🚀 NodeAI - HF Login + History")
     
     with gr.Row():
         # Sidebar
         with gr.Column(scale=1, min_width=350):
             gr.Markdown("## 🔐 Hugging Face")
-            login_btn = gr.LoginButton()  # ✅ CHỈ LoginButton
+            gr.LoginButton()  # ✅ Native HF Login
             
             gr.Markdown("## 📚 Lịch sử Chat")
             history_display = gr.Chatbot(
-                height=300, label="Lịch sử", 
+                height=300, 
+                label="Lịch sử tin nhắn",
                 avatar_images=("👤", "🤖")
             )
             
             with gr.Row():
-                load_btn = gr.Button("📥 Load", variant="secondary")
-                clear_btn = gr.Button("🗑️ Clear", variant="stop")
+                load_btn = gr.Button("📥 Load History", variant="secondary")
+                clear_btn = gr.Button("🗑️ Clear History", variant="stop")
         
         # Main Chat
         with gr.Column(scale=4):
             chatbot = gr.ChatInterface(
-                respond,
+                respond,  # ✅ Auto nhận 7 args từ ChatInterface
                 additional_inputs=[
                     gr.Textbox(
                         value="Bạn là NodeAI - trợ lý AI thông minh, trả lời tiếng Việt.",
@@ -80,17 +90,16 @@ with gr.Blocks(title="NodeAI") as demo:  # ❌ NO theme ở đây
                     gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p"),
                 ],
                 title="💬 NodeAI Chat",
-                description="**Đăng nhập HF sidebar → Chat → Auto save history**"
+                description="**Đăng nhập HF → Chat → Auto save history theo user**"
             )
     
-    # Events
-    load_btn.click(load_history, gr.OAuthToken(), history_display)
-    clear_btn.click(lambda: [], outputs=history_display)
+    # ✅ History events - Gradio tự inject OAuthToken
+    load_btn.click(load_history, outputs=history_display)
+    clear_btn.click(clear_history, outputs=history_display)
 
 if __name__ == "__main__":
-    # ✅ Theme ở launch() - Gradio 6.0 syntax
     demo.launch(
-        theme=gr.themes.Soft(primary_hue="blue"),
+        theme=gr.themes.Soft(primary_hue="blue"),  # ✅ Theme ở launch()
         server_name="0.0.0.0",
         server_port=7860
     )