Update app.py

app.py

@@ -1,13 +1,64 @@
 import gradio as gr
 import os
 import requests
+import json
+from datetime import datetime
+import hashlib
 
 groq_key = os.getenv("GROQ_API_KEY")
-MODEL = "llama-3.3-70b-versatile"  # ✅ Model mới nhất Groq 2026
+MODEL = "llama3-groq-70b-8192-tool-use-preview"  # Model 2026 preview
+
+# User management
+users = {}  # {username: {"token": hash, "history": []}}
+current_user = None
 
 status_msg = "❌ Chưa set GROQ_API_KEY"
 if groq_key:
-    status_msg = f"✅ NodeAI Ready - {MODEL}"
+    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)
+    
+    return "❌ Lỗi đăng nhập", gr.update(visible=False), gr.update(visible=True)
+
+def logout():
+    global current_user
+    current_user = None
+    return "👋 Đã đăng xuất", gr.update(visible=False), gr.update(visible=True), []
+
+def save_chat(username, message, reply):
+    """Lưu tin nhắn vào history user"""
+    if username not in users:
+        users[username] = {"history": []}
+    
+    users[username]["history"].append({
+        "timestamp": datetime.now().strftime("%H:%M"),
+        "user": message,
+        "ai": reply
+    })
+    
+    # Giữ 50 tin nhắn gần nhất
+    if len(users[username]["history"]) > 50:
+        users[username]["history"] = users[username]["history"][-50:]
+
+def load_history(username):
+    """Load lịch sử chat"""
+    if username in users and users[username]["history"]:
+        return [
+            [{"role": "user", "content": h["user"]}, 
+             {"role": "assistant", "content": h["ai"]}] 
+            for h in users[username]["history"][-10:]
+        ]
+    return []
 
 def call_groq(messages):
     url = "https://api.groq.com/openai/v1/chat/completions"
@@ -25,42 +76,105 @@ def call_groq(messages):
     
     try:
         response = requests.post(url, headers=headers, json=data, timeout=20)
-        if response.status_code != 200:
-            return f"❌ Model lỗi. Thử: llama-3.3-70b-versatile"
-        return response.json()["choices"][0]["message"]["content"]
-    except:
-        # Fallback model nếu lỗi
-        data["model"] = "mixtral-8x7b-32768"
-        try:
-            response = requests.post(url, headers=headers, json=data, timeout=20)
+        if response.status_code == 200:
             return response.json()["choices"][0]["message"]["content"]
-        except:
-            return "❌ Không kết nối được Groq"
+        return f"❌ Lỗi {response.status_code}"
+    except:
+        return "❌ Không kết nối Groq"
 
-def respond(message, history):
-    if not groq_key:
-        return "", history + [{"role": "assistant", "content": status_msg}]
+def respond(message, history, username):
+    global current_user
     
-    messages = [{"role": "system", "content": "Bạn là NodeAI - trợ lý AI tiếng Việt."}]
-    messages.append({"role": "user", "content": message})
+    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)
-    return "", history + [{"role": "assistant", "content": reply}]
+    
+    # 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}"
 
-with gr.Blocks(title="NodeAI") as demo:
-    gr.Markdown("# 🚀 NodeAI - Groq AI")
+def load_user_history(username):
+    return load_history(username)
+
+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}")
     
-    chatbot = gr.Chatbot(height=500, avatar_images=("👤", "🤖"))
-    
     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)
+        # Sidebar - Login & History
+        with gr.Column(scale=1, min_width=300):
+            gr.Markdown("## 👤 Tài khoản")
+            
+            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.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")
+            
+            history_display = gr.Chatbot(
+                height=300, 
+                label="Lịch sử chat", 
+                visible=False
+            )
+        
+        # Main Chat
+        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]
+    )
+    
+    load_history_btn.click(
+        load_user_history, 
+        inputs=gr.State(), 
+        outputs=history_display
+    ).then(
+        lambda x: gr.update(visible=True), 
+        outputs=[history_display]
+    )
+    
+    send_btn.click(
+        respond, 
+        inputs=[msg, chatbot, gr.State()], 
+        outputs=[msg, chatbot, status_display]
+    )
+    
+    msg.submit(
+        respond, 
+        inputs=[msg, chatbot, gr.State()], 
+        outputs=[msg, chatbot, status_display]
+    )
     
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    send_btn.click(respond, [msg, chatbot], [msg, chatbot])
-    clear_btn.click(lambda: [], None, chatbot)
+    clear_btn.click(lambda: [], outputs=chatbot)
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft(primary_hue="blue"))
+    demo.launch()