Update app.py

app.py

@@ -1,114 +1,94 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
+import json
 
-# 1. Khởi tạo Client với Qwen 2.5 (Ổn định và mạnh mẽ hơn cho Inference API)
+# 1. Khởi tạo Client
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-# Qwen/Qwen2.5-7B-Instruct là lựa chọn an toàn và Pro nhất hiện tại
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
 custom_css = """
 .header-node { 
     background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%); 
-    padding: 25px; border-radius: 12px; color: white; text-align: center;
-    border: 1px solid #334155; margin-bottom: 20px;
+    padding: 20px; border-radius: 12px; color: white; text-align: center; margin-bottom: 20px;
+}
+.log-container { 
+    background: #1e293b; color: #34d399; font-family: 'Courier New', monospace; 
+    padding: 10px; border-radius: 8px; font-size: 0.85em; height: 150px; overflow-y: auto;
 }
-.sidebar-box { background: #f8fafc; padding: 15px; border-radius: 10px; border: 1px solid #e2e8f0; }
 """
 
 def process_chat(message, history, session_memory, profile: gr.OAuthProfile | None):
-    # Trích xuất nội dung tin nhắn
-    if isinstance(message, dict):
-        user_input = message.get("text", "")
-    elif isinstance(message, list):
-        user_input = message[-1]["content"][0]["text"]
-    else:
-        user_input = str(message)
+    # Trích xuất nội dung
+    if isinstance(message, dict): user_input = message.get("text", "")
+    elif isinstance(message, list): user_input = message[-1]["content"][0]["text"]
+    else: user_input = str(message)
 
     user_name = profile.name if profile else "Guest"
     
-    # 2. AUTO-MEMORY LOGIC (Lưu trữ nội bộ)
-    system_msg = f"Bạn là NodeAI Pro, một trợ lý AI cao cấp. Đang hỗ trợ {user_name}."
-    if session_memory:
-        system_msg += f" [Hệ thống nhớ]: {session_memory}"
+    # Logic Memory
+    system_msg = f"Bạn là NodeAI Pro. Đang hỗ trợ {user_name}."
+    if session_memory: system_msg += f" [Memory]: {session_memory}"
 
     messages = [{"role": "system", "content": system_msg}]
-    
-    # Chuyển history sang định dạng API
-    if history:
-        for turn in history:
-            role = turn["role"]
-            # Đảm bảo lấy đúng text từ format phức tạp của Gradio 6
-            content = turn["content"][0]["text"] if isinstance(turn["content"], list) else turn["content"]
-            messages.append({"role": role, "content": content})
-    
+    for turn in history:
+        messages.append({"role": turn["role"], "content": turn["content"][0]["text"]})
     messages.append({"role": "user", "content": user_input})
 
     try:
         full_response = ""
-        # Gọi API với Qwen
-        for token in client.chat_completion(messages=messages, max_tokens=1500, stream=True, temperature=0.7):
+        for token in client.chat_completion(messages=messages, max_tokens=1500, stream=True):
             delta = token.choices[0].delta.content
             if delta:
                 full_response += delta
-                # Cập nhật history theo chuẩn Gradio 6
-                new_history = history + [
+                yield "", history + [
                     {"role": "user", "content": [{"type": "text", "text": user_input}]},
                     {"role": "assistant", "content": [{"type": "text", "text": full_response}]}
-                ]
-                yield "", new_history, session_memory
+                ], session_memory
         
-        # Tự động cập nhật bộ nhớ ẩn sau lượt chat
+        # Cập nhật bộ nhớ sau khi chat
         if len(user_input) > 5:
-            new_info = f"Người dùng quan tâm: {user_input[:40]}."
-            session_memory = (session_memory + " " + new_info)[-500:] # Giữ 500 ký tự cuối
-
+            session_memory += f" | {user_name} hỏi về {user_input[:30]}"
     except Exception as e:
-        error_msg = f"❌ API Error: {str(e)}"
-        yield "", history + [{"role": "assistant", "content": [{"type": "text", "text": error_msg}]}], session_memory
+        yield "", history + [{"role": "assistant", "content": [{"type": "text", "text": f"Error: {str(e)}"}]}], session_memory
 
 with gr.Blocks(title="NodeAI Pro") as demo:
-    # State lưu trữ Memory trong script
     internal_memory = gr.State("")
 
-    gr.HTML("""
-        <div class="header-node">
-            <h1 style='margin:0'>NODE AI PRO</h1>
-            <p style='margin:0; opacity: 0.7;'>Qwen-2.5 Neural Interface • Memory Enabled</p>
-        </div>
-    """)
+    gr.HTML("<div class='header-node'><h1>NODE AI PRO v3.3</h1><p>Persistent Memory & History Interface</p></div>")
 
     with gr.Row():
+        # SIDEBAR: Chỉ hiện thị đầy đủ khi đã Login
         with gr.Column(scale=1):
             with gr.Group():
-                gr.Markdown("### 👤 Profile")
+                gr.Markdown("### 👤 Tài khoản")
                 login_btn = gr.LoginButton()
+                
                 @gr.render(inputs=login_btn)
-                def render_user(profile: gr.OAuthProfile | None, *args):
+                def render_history_panel(profile: gr.OAuthProfile | None, *args):
                     if profile:
-                        return gr.Markdown(f"✅ User: **{profile.name}**")
-                    return gr.Markdown("🌐 Mode: **Anonymous**")
+                        gr.Markdown(f"👋 Chào, **{profile.name}**")
+                        with gr.Accordion("📂 Nhật ký phiên chat", open=True):
+                            # Hiển thị log bộ nhớ hiện tại
+                            gr.Markdown("**Dữ liệu bộ nhớ AI đang giữ:**")
+                            gr.HTML("<div class='log-container' id='mem-log'>Đang phân tích ngữ cảnh...</div>")
+                            save_btn = gr.Button("💾 Xuất file Lịch sử (.json)", size="sm")
+                    else:
+                        gr.Markdown("⚠️ Đăng nhập để kích hoạt bộ nhớ và lưu lịch sử.")
 
-            with gr.Group():
-                gr.Markdown("### ⚙️ System")
-                gr.Markdown(f"**Engine:** `Qwen-2.5-7B`")
-                gr.Markdown("**Memory:** `Auto-Active` ✅")
-
-            clear_btn = gr.Button("🗑️ Reset Session", variant="secondary")
-            
-        with gr.Column(scale=4):
-            chatbot = gr.Chatbot(label="NodeAI Console", height=550)
+            clear_btn = gr.Button("🗑️ Xóa phiên chat", variant="secondary")
             
+        # CHAT MAIN
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(label="Neural Feed", height=500)
             with gr.Row():
-                msg_input = gr.Textbox(placeholder="Nhập yêu cầu...", show_label=False, scale=9)
+                msg_input = gr.Textbox(placeholder="Gõ lệnh tại đây...", show_label=False, scale=9)
                 submit_btn = gr.Button("🚀", variant="primary", scale=1)
 
-    # Xử lý sự kiện
-    input_data = [msg_input, chatbot, internal_memory]
-    output_data = [msg_input, chatbot, internal_memory]
-    
-    msg_input.submit(process_chat, input_data, output_data)
-    submit_btn.click(process_chat, input_data, output_data)
+    # Xử lý Logic
+    chat_args = {"fn": process_chat, "inputs": [msg_input, chatbot, internal_memory], "outputs": [msg_input, chatbot, internal_memory]}
+    msg_input.submit(**chat_args)
+    submit_btn.click(**chat_args)
     clear_btn.click(lambda: (None, [], ""), None, [msg_input, chatbot, internal_memory])
 
 if __name__ == "__main__":