Update app.py

app.py

@@ -16,19 +16,19 @@ except:
 
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
-# 2. HỆ THỐNG MEMORY (Chỉ chạy khi có Profile)
+# 2. HỆ THỐNG MEMORY (Xử lý theo từng User)
 def get_db_history(profile: gr.OAuthProfile | None):
-    if not supabase or not profile: 
+    if not supabase or not profile or not profile.username: 
         return []
     try:
         res = supabase.table("chats").select("*").eq("username", profile.username).order("created_at", desc=True).execute()
         return res.data
-    except:
+    except Exception as e:
+        print(f"Lỗi DB: {e}")
         return []
 
 def save_to_db(session_id, title, history, profile: gr.OAuthProfile | None):
-    # CHẶN LƯU: Nếu không có profile (chưa đăng nhập) thì thoát luôn
-    if not supabase or not profile or profile.username is None:
+    if not supabase or not profile or not profile.username:
         return 
     try:
         data = {
@@ -38,27 +38,29 @@ def save_to_db(session_id, title, history, profile: gr.OAuthProfile | None):
             "history": history
         }
         supabase.table("chats").upsert(data, on_conflict="session_id").execute()
-    except:
-        pass
+    except Exception as e:
+        print(f"Lỗi lưu: {e}")
 
-# 3. GIAO DIỆN
+# 3. GIAO DIỆN GLASSMORPHISM
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@700&family=Plus+Jakarta+Sans:wght@400;600&display=swap');
-.gradio-container { background: radial-gradient(circle at 50% 0%, #0f172a 0%, #020617 100%) !important; color: #e2e8f0 !important; }
-.sidebar-node { background: rgba(15, 23, 42, 0.8) !important; backdrop-filter: blur(15px); border-right: 1px solid #38bdf833; padding: 20px; overflow-y: auto; height: 100vh; }
-.logo-area { font-family: 'Orbitron'; background: linear-gradient(90deg, #38bdf8, #818cf8); -webkit-background-clip: text; -webkit-text-fill-color: transparent; font-size: 22px; margin-bottom: 20px; text-align: center; }
-.status-tag { font-size: 0.8em; padding: 5px 10px; border-radius: 20px; text-align: center; margin-bottom: 15px; }
-.status-online { background: rgba(34, 197, 94, 0.2); color: #4ade80; border: 1px solid #22c55e44; }
-.status-offline { background: rgba(239, 68, 68, 0.2); color: #f87171; border: 1px solid #ef444444; }
+.gradio-container { background: radial-gradient(circle at 50% 0%, #0f172a 0%, #020617 100%) !important; color: #e2e8f0 !important; font-family: 'Plus Jakarta Sans', sans-serif !important; }
+.sidebar-node { background: rgba(15, 23, 42, 0.8) !important; backdrop-filter: blur(15px); border-right: 1px solid #38bdf833; padding: 20px; height: 100vh; overflow-y: auto; }
+.logo-area { font-family: 'Orbitron'; background: linear-gradient(90deg, #38bdf8, #818cf8); -webkit-background-clip: text; -webkit-text-fill-color: transparent; font-size: 22px; margin-bottom: 25px; text-align: center; font-weight: bold; }
+.status-tag { font-size: 0.85em; padding: 8px; border-radius: 12px; text-align: center; margin-bottom: 15px; border: 1px solid rgba(255,255,255,0.1); }
+.history-btn { margin-bottom: 5px !important; text-align: left !important; justify-content: flex-start !important; }
 """
 
+# 4. LOGIC XỬ LÝ CHAT
 def chat_engine(message, history, session_id, profile: gr.OAuthProfile | None):
-    text = message.get("text", "")
-    if not text.strip(): return gr.update(value={"text": "", "files": []}), history, session_id
+    text = message.get("text", "").strip()
+    if not text:
+        yield gr.update(value={"text": "", "files": []}), history, session_id
+        return
 
-    # Xác định Master hay Khách
+    # Xác định danh tính
     user_name = profile.username if profile else "Guest"
-    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Đang hỗ trợ: {user_name}."}]
+    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {user_name}. Phản hồi bằng tiếng Việt chuyên nghiệp."}]
     
     for msg in history:
         messages.append({"role": msg["role"], "content": msg["content"]})
@@ -67,8 +69,8 @@ def chat_engine(message, history, session_id, profile: gr.OAuthProfile | None):
     new_history = history + [{"role": "user", "content": text}]
     ans = ""
     
-    # Hiển thị trạng thái lưu
-    save_status = "" if profile else "\n\n*(Lưu ý: Bạn chưa đăng nhập, lịch sử này sẽ mất khi tải lại trang)*"
+    # Thông báo nếu không lưu được
+    save_info = "" if profile else "\n\n*(Lưu ý: Bạn đang ở chế độ Guest, lịch sử sẽ không được lưu)*"
     
     yield gr.update(value={"text": "", "files": []}), new_history + [{"role": "assistant", "content": "..."}], session_id
     
@@ -76,63 +78,84 @@ def chat_engine(message, history, session_id, profile: gr.OAuthProfile | None):
         for token in client.chat_completion(messages=messages, max_tokens=2048, stream=True):
             if token.choices[0].delta.content:
                 ans += token.choices[0].delta.content
-                yield gr.update(), new_history + [{"role": "assistant", "content": ans + (save_status if not token.choices[0].delta.content else "")}], session_id
+                yield gr.update(), new_history + [{"role": "assistant", "content": ans}], session_id
         
-        # CHỈ LƯU VÀO DB KHI CÓ PROFILE
+        # Chỉ lưu khi đã có profile
         if profile:
             save_to_db(session_id, text[:25], new_history + [{"role": "assistant", "content": ans}], profile)
             
     except Exception as e:
-        yield gr.update(), new_history + [{"role": "assistant", "content": f"Lỗi: {str(e)}"}], session_id
+        yield gr.update(), new_history + [{"role": "assistant", "content": f"Lỗi kết nối: {str(e)}"}], session_id
 
-# 4. UI
-with gr.Blocks() as demo:
+# 5. UI GIAO DIỆN
+with gr.Blocks(title="NodeAI Genesis Pro") as demo:
     session_id = gr.State(str(uuid.uuid4()))
     db_state = gr.State([])
 
     with gr.Row():
+        # SIDEBAR
         with gr.Column(scale=1, elem_classes="sidebar-node"):
             gr.HTML("<div class='logo-area'>NODEAI GENESIS</div>")
             
-            gr.LoginButton(value="Sign in with Hugging Face", size="sm")
+            gr.LoginButton(value="Đăng nhập Hugging Face", size="sm")
             
-            # Hiển thị trạng thái đăng nhập trực quan
-            with gr.Group():
-                status_html = gr.HTML()
+            status_display = gr.HTML("<div class='status-tag'>🔴 Chưa đăng nhập</div>")
             
             new_chat_btn = gr.Button("＋ CHAT MỚI", variant="primary")
+            
             gr.Markdown("### 🗄️ Database Memory")
             
-            # Render danh sách lịch sử
-            history_box = gr.Column()
+            # Sử dụng gr.render để cập nhật danh sách lịch sử mà không gây lỗi Key
             @gr.render(inputs=[db_state])
-            def show_history(items):
+            def render_history_list(items):
                 with gr.Column():
+                    if not items:
+                        gr.Markdown("*Không có dữ liệu*")
                     for item in items:
-                        btn = gr.Button(f"◈ {item['title']}", variant="ghost")
-                        btn.click(lambda h=item['history'], s=item['session_id']: (h, s), None, [chatbot, session_id])
-
+                        # Dùng hàm lambda bọc kín để tránh lỗi truyền tham số trong vòng lặp
+                        btn = gr.Button(f"◈ {item['title']}", variant="ghost", elem_classes="history-btn")
+                        btn.click(
+                            fn=lambda h=item['history'], s=item['session_id']: (h, s),
+                            outputs=[chatbot, session_id]
+                        )
+
+        # MAIN CHAT
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(show_label=False, height="85vh")
-            msg_input = gr.MultimodalTextbox(show_label=False, placeholder="Hỏi NodeAI...", container=False)
-
-    # Logic khi tải trang / Đăng nhập
-    def check_user(profile: gr.OAuthProfile | None):
+            chatbot = gr.Chatbot(show_label=False, height="82vh")
+            
+            with gr.Row():
+                msg_input = gr.MultimodalTextbox(
+                    show_label=False, 
+                    placeholder="Hỏi NodeAI hoặc dán code Roblox tại đây...",
+                    container=False,
+                    scale=7
+                )
+
+    # XỬ LÝ SỰ KIỆN
+    def on_load(profile: gr.OAuthProfile | None):
         if profile:
-            msg = f"<div class='status-tag status-online'>● Online: {profile.username}</div>"
-            return msg, get_db_history(profile)
-        msg = "<div class='status-tag status-offline'>● Guest Mode (No Save)</div>"
-        return msg, []
-
-    demo.load(check_user, None, [status_html, db_state])
-
-    # Sự kiện chat
-    msg_input.submit(chat_engine, [msg_input, chatbot, session_id], [msg_input, chatbot, session_id]).then(
-        lambda profile: get_db_history(profile) if profile else [], None, db_state
+            status = f"<div class='status-tag' style='color:#4ade80'>🟢 Master: {profile.username}</div>"
+            history = get_db_history(profile)
+            return status, history
+        return "<div class='status-tag'>🔴 Chế độ Guest (Không lưu)</div>", []
+
+    # Tải dữ liệu khi mở trang
+    demo.load(on_load, None, [status_display, db_state])
+
+    # Gửi tin nhắn
+    msg_input.submit(
+        chat_engine, 
+        [msg_input, chatbot, session_id], 
+        [msg_input, chatbot, session_id]
+    ).then(
+        fn=on_load, outputs=[status_display, db_state]
     )
     
-    # Nút chat mới
-    new_chat_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
+    # Tạo chat mới
+    new_chat_btn.click(
+        fn=lambda: ([], str(uuid.uuid4())), 
+        outputs=[chatbot, session_id]
+    )
 
 if __name__ == "__main__":
     demo.launch(css=custom_css)