Update app.py

app.py

@@ -1,23 +1,33 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
+import json
+from datetime import datetime
 
-# 1. Cấu hình Model & Token (Sử dụng Llama-3.2 để Pro hơn)
+# 1. Cấu hình Model & Token
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-model_id = "meta-llama/Llama-3.2-1B-Instruct" 
+model_id = "meta-llama/Llama-3.2-1B-Instruct"
 client = InferenceClient(model=model_id, token=HF_TOKEN)
 
-# 2. CSS tinh chỉnh cho Gradio 6.x
+# 2. CSS Pro (Glassmorphism & Gradients)
 custom_css = """
-.header-pro { 
-    background: linear-gradient(135deg, #000428 0%, #004e92 100%); 
-    padding: 20px; border-radius: 15px; color: white; text-align: center; margin-bottom: 20px;
+.header-pro {
+    background: linear-gradient(135deg, #000428 0%, #004e92 100%);
+    padding: 25px; border-radius: 15px; color: white; text-align: center; margin-bottom: 20px;
+}
+.memory-box {
+    background: #f0f4ff; border-left: 5px solid #004e92; padding: 10px; border-radius: 5px;
+}
+.user-status {
+    font-weight: bold; color: #004e92;
 }
-.footer-info { font-size: 0.8em; text-align: center; color: #666; margin-top: 10px; }
 """
 
-def chat_fn(message, history):
-    # Trích xuất text từ định dạng tin nhắn Gradio 6
+def chat_fn(message, history, memory, profile: gr.OAuthProfile | None):
+    # Kiểm tra đăng nhập
+    user_name = profile.name if profile else "Guest"
+    
+    # Định dạng Input của Gradio 6.9.0
     if isinstance(message, dict):
         user_input = message.get("text", "")
     elif isinstance(message, list):
@@ -25,66 +35,106 @@ def chat_fn(message, history):
     else:
         user_input = str(message)
 
-    # Khởi tạo history nếu trống
-    if history is None: history = []
+    # Xử lý Memory: Kết hợp bộ nhớ vào Prompt hệ thống
+    system_prompt = f"Bạn là NodeAI Pro. Đang trò chuyện với {user_name}. "
+    if memory:
+        system_prompt += f"Bộ nhớ quan trọng cần nhớ: {memory}"
 
-    # Format tin nhắn người dùng
-    user_msg = {"role": "user", "content": [{"type": "text", "text": user_input}]}
-    history.append(user_msg)
+    # Tạo hội thoại gửi lên API
+    messages = [{"role": "system", "content": system_prompt}]
     
-    # Gọi API Inference
+    # Thêm lịch sử (convert Gradio format sang API format)
+    for turn in history:
+        role = turn["role"]
+        content = turn["content"][0]["text"]
+        messages.append({"role": role, "content": content})
+    
+    messages.append({"role": "user", "content": user_input})
+
+    # Gọi API
     try:
-        # Sử dụng chat_completion để có kết quả thông minh hơn
         response = client.chat_completion(
-            messages=[{"role": "user", "content": user_input}],
-            max_tokens=500,
+            messages=messages,
+            max_tokens=512,
             temperature=0.7
         )
         bot_res = response.choices[0].message.content
     except Exception as e:
-        bot_res = f"⚠️ Lỗi kết nối: Hãy đảm bảo bạn đã add HF_TOKEN vào Settings. (Chi tiết: {str(e)})"
+        bot_res = f"⚠️ Lỗi kết nối: {str(e)}"
 
-    # Format tin nhắn Bot
-    bot_msg = {"role": "assistant", "content": [{"type": "text", "text": bot_res}]}
-    history.append(bot_msg)
+    # Cập nhật History theo format Gradio 6
+    history.append({"role": "user", "content": [{"type": "text", "text": user_input}]})
+    history.append({"role": "assistant", "content": [{"type": "text", "text": bot_res}]})
     
-    return "", history
+    return "", history, memory
+
+def save_chat(history, profile: gr.OAuthProfile | None):
+    if not profile:
+        return "⚠️ Vui lòng đăng nhập để lưu cuộc trò chuyện!"
+    
+    filename = f"chat_history_{profile.name}.json"
+    with open(filename, "w", encoding="utf-8") as f:
+        json.dump(history, f, ensure_ascii=False, indent=4)
+    return f"✅ Đã lưu {len(history)//2} tin nhắn vào {filename}"
 
 with gr.Blocks(title="NodeAI Pro") as demo:
-    gr.HTML("""
+    # State để lưu Memory dài hạn trong phiên làm việc
+    memory_state = gr.State("")
+
+    gr.HTML(f"""
         <div class="header-pro">
-            <h1 style='margin:0'>🚀 NodeAI Pro</h1>
-            <p style='margin:0'>High-Performance AI Assistant</p>
+            <h1 style='margin:0'>🤖 NodeAI Pro v2.0</h1>
+            <p style='margin:0; opacity:0.8'>Bản nâng cấp: Persistent Memory & OAuth Secure</p>
         </div>
     """)
-    
+
     with gr.Row():
+        # SIDEBAR: Quản lý người dùng & Memory
         with gr.Column(scale=1):
-            gr.Markdown("### 🛠️ Core Engine")
-            gr.LoginButton()
-            gr.Markdown(f"**Engine:** `Llama-3.2-1B`")
-            gr.Markdown(f"**API Status:** {'🟢 Online' if HF_TOKEN else '🔴 Offline'}")
-            clear_btn = gr.Button("🗑️ Reset Session", variant="secondary")
-            
-        with gr.Column(scale=4):
-            # Không dùng type="messages" ở đây để tránh TypeError trên một số môi trường
-            chatbot = gr.Chatbot(label="NodeAI Chat Console", height=500)
-            
-            with gr.Row():
-                msg_input = gr.Textbox(
-                    placeholder="Nhập câu hỏi...",
-                    show_label=False,
-                    scale=8
+            with gr.Group():
+                gr.Markdown("### 👤 User Profile")
+                login_btn = gr.LoginButton()
+                
+                # Hiển thị thông tin sau khi đăng nhập
+                @gr.render(inputs=login_btn)
+                def show_user_info(p: gr.OAuthProfile | None):
+                    if p:
+                        return gr.Markdown(f"Chào mừng, <span class='user-status'>{p.name}</span>!")
+                    return gr.Markdown("Vui lòng đăng nhập để lưu lịch sử.")
+
+            with gr.Group():
+                gr.Markdown("### 🧠 AI Memory")
+                memory_input = gr.Textbox(
+                    label="Ghi chú để AI ghi nhớ (Sở thích, tên, dự án...)",
+                    placeholder="Ví dụ: Tôi thích code Python...",
+                    lines=3
                 )
-                submit_btn = gr.Button("Send", variant="primary", scale=2)
-            
-            gr.HTML("<div class='footer-info'>NodeAI Pro v1.0.9 | Powered by Hugging Face Inference API</div>")
+                gr.Markdown("<div class='memory-box'>Memory giúp AI không quên thông tin quan trọng qua các lần chat.</div>")
 
-    # Xử lý Logic
-    msg_input.submit(chat_fn, [msg_input, chatbot], [msg_input, chatbot])
-    submit_btn.click(chat_fn, [msg_input, chatbot], [msg_input, chatbot])
-    clear_btn.click(lambda: None, None, chatbot)
+            save_btn = gr.Button("💾 Lưu cuộc trò chuyện", variant="secondary")
+            status_msg = gr.Markdown("")
+            clear_btn = gr.Button("🗑️ Reset")
+
+        # MAIN CHAT
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(label="NodeAI Console", height=550)
+            with gr.Row():
+                msg_input = gr.Textbox(placeholder="Nhập tin nhắn...", show_label=False, scale=8)
+                submit_btn = gr.Button("Gửi", variant="primary", scale=2)
+
+    # LOGIC XỬ LÝ
+    # Khi nhấn Gửi: Truyền thêm memory_input và OAuth profile
+    input_params = [msg_input, chatbot, memory_input]
+    output_params = [msg_input, chatbot, memory_state]
+
+    msg_input.submit(chat_fn, input_params, output_params)
+    submit_btn.click(chat_fn, input_params, output_params)
+    
+    # Lưu chat
+    save_btn.click(save_chat, [chatbot], [status_msg])
+    
+    # Xóa chat
+    clear_btn.click(lambda: ([], ""), None, [chatbot, memory_input])
 
 if __name__ == "__main__":
-    # Đưa CSS vào launch() để fix UserWarning
     demo.launch(css=custom_css)