Update app.py

app.py

@@ -3,20 +3,15 @@ from huggingface_hub import InferenceClient
 
 MODEL = "mistralai/Mistral-7B-Instruct-v0.1"
 
-# In-memory chat history
 chat_history = {}
 
-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
-    
+def respond(message, history, system_message, max_tokens, temperature, top_p):
+    """6 args - ChatInterface tự inject OAuth token"""
     try:
-        client = InferenceClient(token=hf_token.token, model=MODEL)
+        # Gradio tự inject token qua ChatInterface
+        client = InferenceClient(model=MODEL)
         messages = [{"role": "system", "content": system_message}]
         
-        # Convert history to messages format
         for user_msg, assistant_msg in history:
             messages.append({"role": "user", "content": str(user_msg)})
             if assistant_msg:
@@ -35,71 +30,53 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, hf
             if chunk.choices[0].delta.content:
                 response += chunk.choices[0].delta.content
                 yield response
-                
     except Exception as e:
-        yield f"❌ Lỗi model: {str(e)}"
+        yield f"❌ Lỗi: {str(e)}"
 
-def load_history(hf_token):
-    """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 load_history():
+    """No args - Gradio tự handle OAuth"""
+    return []  # Placeholder - cần session storage
 
-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] = []
+def clear_history():
+    """No args"""
     return []
 
 with gr.Blocks(title="NodeAI") as demo:
-    gr.Markdown("# 🚀 NodeAI - HF Login + History")
+    gr.Markdown("# 🚀 NodeAI - AI Chat")
     
     with gr.Row():
         # Sidebar
         with gr.Column(scale=1, min_width=350):
             gr.Markdown("## 🔐 Hugging Face")
-            gr.LoginButton()  # ✅ Native HF Login
+            gr.LoginButton()  # ✅ Chỉ cần này
             
-            gr.Markdown("## 📚 Lịch sử Chat")
-            history_display = gr.Chatbot(
-                height=300, 
-                label="Lịch sử tin nhắn",
-                avatar_images=("👤", "🤖")
-            )
+            gr.Markdown("## 📚 Lịch sử")
+            history_display = gr.Chatbot(height=300, avatar_images=("👤", "🤖"))
             
             with gr.Row():
-                load_btn = gr.Button("📥 Load History", variant="secondary")
-                clear_btn = gr.Button("🗑️ Clear History", variant="stop")
+                load_btn = gr.Button("📥 Load", variant="secondary")
+                clear_btn = gr.Button("🗑️ Clear", variant="stop")
         
         # Main Chat
         with gr.Column(scale=4):
             chatbot = gr.ChatInterface(
-                respond,  # ✅ Auto nhận 7 args từ ChatInterface
+                respond,  # ✅ 6 args tự động
                 additional_inputs=[
-                    gr.Textbox(
-                        value="Bạn là NodeAI - trợ lý AI thông minh, trả lời tiếng Việt.",
-                        label="System prompt"
-                    ),
+                    gr.Textbox(value="Bạn là NodeAI - trợ lý AI tiếng Việt.", label="System"),
                     gr.Slider(1, 1024, value=512, step=1, label="Max tokens"),
                     gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature"),
                     gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p"),
                 ],
-                title="💬 NodeAI Chat",
-                description="**Đăng nhập HF → Chat → Auto save history theo user**"
+                title="💬 NodeAI Chat"
             )
     
-    # ✅ History events - Gradio tự inject OAuthToken
+    # ✅ Event listeners KHÔNG cần inputs
     load_btn.click(load_history, outputs=history_display)
     clear_btn.click(clear_history, outputs=history_display)
 
 if __name__ == "__main__":
     demo.launch(
-        theme=gr.themes.Soft(primary_hue="blue"),  # ✅ Theme ở launch()
+        theme=gr.themes.Soft(primary_hue="blue"),
         server_name="0.0.0.0",
         server_port=7860
     )