Update app.py

app.py

@@ -1,180 +1,140 @@
 import gradio as gr
+from huggingface_hub import InferenceClient
 import os
-import requests
-import json
-from datetime import datetime
-import hashlib
 
-groq_key = os.getenv("GROQ_API_KEY")
-MODEL = "llama3-groq-70b-8192-tool-use-preview"  # Model 2026 preview
+# Config
+MODEL = "microsoft/DialoGPT-medium"  # Hoặc model HF Inference API khác
+GROQ_KEY = os.getenv("GROQ_API_KEY")  # Fallback nếu cần
 
-# User management
-users = {}  # {username: {"token": hash, "history": []}}
-current_user = None
-
-status_msg = "❌ Chưa set GROQ_API_KEY"
-if groq_key:
-    status_msg = f"✅ Groq {MODEL} Ready"
-
-def hash_token(token):
-    return hashlib.sha256(token.encode()).hexdigest()
-
-def login(username, password):
-    global current_user
-    token = hash_token(f"{username}:{password}")
-    if token in users:
-        current_user = username
-        return f"✅ Đăng nhập thành công: {username}", gr.update(visible=True), gr.update(visible=False)
-    else:
-        return "❌ Sai tên đăng nhập/mật khẩu", gr.update(visible=False), gr.update(visible=True)
+def respond(message, history, system_message, max_tokens, temperature, top_p, hf_token):
+    """Chat với HF Inference API + OAuth token"""
+    if not hf_token or not hf_token.token:
+        yield "❌ Vui lòng đăng nhập Hugging Face!"
+        return
+    
+    client = InferenceClient(token=hf_token.token, model=MODEL)
     
-    return "❌ Lỗi đăng nhập", gr.update(visible=False), gr.update(visible=True)
+    messages = [{"role": "system", "content": system_message}]
+    messages.extend(history)
+    messages.append({"role": "user", "content": message})
+    
+    response = ""
+    try:
+        for message in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            choices = message.choices
+            token = ""
+            if len(choices) and choices[0].delta.content:
+                token = choices[0].delta.content
+                response += token
+                yield response
+    except Exception as e:
+        yield f"❌ Lỗi: {str(e)}"
 
-def logout():
-    global current_user
-    current_user = None
-    return "👋 Đã đăng xuất", gr.update(visible=False), gr.update(visible=True), []
+# User history storage
+chat_history = {}
 
-def save_chat(username, message, reply):
-    """Lưu tin nhắn vào history user"""
-    if username not in users:
-        users[username] = {"history": []}
+def save_chat(user_id, message, full_response):
+    """Lưu lịch sử chat theo user"""
+    if user_id not in chat_history:
+        chat_history[user_id] = []
     
-    users[username]["history"].append({
-        "timestamp": datetime.now().strftime("%H:%M"),
+    chat_history[user_id].append({
+        "timestamp": "14:35",  # Có thể dùng datetime.now()
         "user": message,
-        "ai": reply
+        "ai": full_response[-1] if full_response else "Lỗi"
     })
     
-    # Giữ 50 tin nhắn gần nhất
-    if len(users[username]["history"]) > 50:
-        users[username]["history"] = users[username]["history"][-50:]
+    # Giữ 20 tin nhắn gần nhất
+    if len(chat_history[user_id]) > 20:
+        chat_history[user_id] = chat_history[user_id][-20:]
 
-def load_history(username):
-    """Load lịch sử chat"""
-    if username in users and users[username]["history"]:
+def load_user_history(hf_token):
+    """Load lịch sử từ HF token (user_id)"""
+    if not hf_token or not hf_token.token:
+        return []
+    
+    # Dùng hash token làm user_id
+    user_id = hash(hf_token.token) % 1000000
+    if user_id in chat_history:
         return [
-            [{"role": "user", "content": h["user"]}, 
-             {"role": "assistant", "content": h["ai"]}] 
-            for h in users[username]["history"][-10:]
+            [
+                {"role": "user", "content": chat["user"]},
+                {"role": "assistant", "content": chat["ai"]}
+            ]
+            for chat in chat_history[user_id][-10:]
         ]
     return []
 
-def call_groq(messages):
-    url = "https://api.groq.com/openai/v1/chat/completions"
-    headers = {
-        "Authorization": f"Bearer {groq_key}",
-        "Content-Type": "application/json"
-    }
-    
-    data = {
-        "model": MODEL,
-        "messages": messages,
-        "temperature": 0.7,
-        "max_tokens": 500
-    }
-    
-    try:
-        response = requests.post(url, headers=headers, json=data, timeout=20)
-        if response.status_code == 200:
-            return response.json()["choices"][0]["message"]["content"]
-        return f"❌ Lỗi {response.status_code}"
-    except:
-        return "❌ Không kết nối Groq"
-
-def respond(message, history, username):
-    global current_user
-    
-    if not current_user:
-        return "", history, "❌ Vui lòng đăng nhập!"
-    
-    # Simple messages (no complex history)
-    messages = [
-        {"role": "system", "content": "Bạn là NodeAI - trợ lý AI tiếng Việt thông minh, hữu ích."},
-        {"role": "user", "content": message}
-    ]
-    
-    reply = call_groq(messages)
-    
-    # Lưu history
-    save_chat(current_user, message, reply)
-    
-    new_history = history + [{"role": "user", "content": message}, {"role": "assistant", "content": reply}]
-    return "", new_history, f"👤 {current_user}"
-
-def load_user_history(username):
-    return load_history(username)
+def clear_user_history(hf_token):
+    """Xóa lịch sử user"""
+    if hf_token and hf_token.token:
+        user_id = hash(hf_token.token) % 1000000
+        chat_history[user_id] = []
+    return []
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
-    gr.Markdown("# 🚀 NodeAI - AI Chat với Login & History")
-    gr.Markdown(f"**Status:** {status_msg}")
+    gr.Markdown("# 🚀 NodeAI - HF Login + Chat History")
     
     with gr.Row():
         # Sidebar - Login & History
-        with gr.Column(scale=1, min_width=300):
-            gr.Markdown("## 👤 Tài khoản")
+        with gr.Column(scale=1, min_width=350):
+            gr.Markdown("## 🔐 Hugging Face Login")
             
-            with gr.Group(visible=True) as login_group:
-                username = gr.Textbox(label="Tên đăng nhập", placeholder="yourname")
-                password = gr.Textbox(label="Mật khẩu", type="password")
-                login_btn = gr.Button("🔐 Đăng nhập", variant="primary")
+            with gr.Row():
+                gr.LoginButton(containers=[demo])
+                gr.LogoutButton(containers=[demo])
             
-            with gr.Group(visible=False) as user_group:
-                gr.Markdown("### ✅ Đã đăng nhập")
-                user_info = gr.Markdown("**Chưa có user**")
-                logout_btn = gr.Button("🚪 Đăng xuất", variant="stop")
-                load_history_btn = gr.Button("📚 Load History")
+            gr.Markdown("---")
+            gr.Markdown("## 📚 Lịch sử Chat")
             
-            history_display = gr.Chatbot(
-                height=300, 
-                label="Lịch sử chat", 
-                visible=False
+            history_chatbot = gr.Chatbot(
+                height=350,
+                label="Lịch sử tin nhắn",
+                avatar_images=("👤", "🤖")
             )
+            
+            history_btn = gr.Button("📥 Load History", variant="secondary")
+            clear_history_btn = gr.Button("🗑️ Clear History", variant="stop")
         
-        # Main Chat
+        # Main Chat Area
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(height=500, avatar_images=("👤", "🤖"))
-            status_display = gr.Markdown("**Chưa đăng nhập**")
-            
-            with gr.Row():
-                msg = gr.Textbox(placeholder="Hỏi NodeAI...", scale=7)
-                send_btn = gr.Button("Gửi", variant="primary", scale=1)
-                clear_btn = gr.Button("🗑️", scale=1)
-    
-    # Events
-    login_btn.click(
-        login, 
-        inputs=[username, password], 
-        outputs=[status_display, user_group, login_group]
-    )
-    
-    logout_btn.click(
-        logout, 
-        outputs=[status_display, login_group, user_group, history_display]
-    )
+            chatbot = gr.ChatInterface(
+                respond,
+                additional_inputs=[
+                    gr.Textbox(
+                        value="Bạn là NodeAI - trợ lý AI thông minh, trả lời tiếng Việt.",
+                        label="System Message"
+                    ),
+                    gr.Slider(1, 2048, value=512, step=1, label="Max Tokens"),
+                    gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature"),
+                    gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p"),
+                ],
+                title="NodeAI Chat",
+                description="Đăng nhập Hugging Face để chat và lưu lịch sử!"
+            )
     
-    load_history_btn.click(
-        load_user_history, 
-        inputs=gr.State(), 
-        outputs=history_display
-    ).then(
-        lambda x: gr.update(visible=True), 
-        outputs=[history_display]
-    )
+    # History Events
+    def update_history(hf_token):
+        history = load_user_history(hf_token)
+        return history
     
-    send_btn.click(
-        respond, 
-        inputs=[msg, chatbot, gr.State()], 
-        outputs=[msg, chatbot, status_display]
+    history_btn.click(
+        update_history,
+        inputs=gr.OAuthToken(),
+        outputs=history_chatbot
     )
     
-    msg.submit(
-        respond, 
-        inputs=[msg, chatbot, gr.State()], 
-        outputs=[msg, chatbot, status_display]
+    clear_history_btn.click(
+        clear_user_history,
+        inputs=gr.OAuthToken(),
+        outputs=history_chatbot
     )
-    
-    clear_btn.click(lambda: [], outputs=chatbot)
 
 if __name__ == "__main__":
     demo.launch()