Update app.py

app.py

@@ -4,13 +4,11 @@ from supabase import create_client, Client
 import os
 import uuid
 
-# 1. CẤU HÌNH CORE & SUPABASE
+# 1. CẤU HÌNH CORE
 URL = os.environ.get("SUPABASE_URL")
 KEY = os.environ.get("SUPABASE_KEY")
 HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN")
 
-MASTER_ID = "lamgamepro"
-
 try:
     supabase: Client = create_client(URL, KEY) if URL and KEY else None
 except:
@@ -18,118 +16,123 @@ except:
 
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
-# 2. HỆ THỐNG MEMORY
-def get_db_history():
-    if not supabase: return []
+# 2. HỆ THỐNG MEMORY (Chỉ chạy khi có Profile)
+def get_db_history(profile: gr.OAuthProfile | None):
+    if not supabase or not profile: 
+        return []
     try:
-        res = supabase.table("chats").select("*").eq("username", MASTER_ID).order("created_at", desc=True).execute()
+        res = supabase.table("chats").select("*").eq("username", profile.username).order("created_at", desc=True).execute()
         return res.data
-    except Exception as e:
-        print(f"Lỗi tải DB: {e}")
+    except:
         return []
 
-def save_to_db(session_id, title, history):
-    if not supabase: 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:
+        return 
     try:
-        data = {"username": MASTER_ID, "session_id": session_id, "title": title, "history": history}
+        data = {
+            "username": profile.username, 
+            "session_id": session_id, 
+            "title": title, 
+            "history": history
+        }
         supabase.table("chats").upsert(data, on_conflict="session_id").execute()
-    except Exception as e:
-        print(f"Lỗi lưu DB: {e}")
+    except:
+        pass
 
-# 3. GIAO DIỆN GLASSMORPHISM
+# 3. GIAO DIỆN
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@700&family=Plus+Jakarta+Sans:wght@400;600&display=swap');
-:root { --primary: #38bdf8; --accent: #f97316; }
-.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.7) !important; backdrop-filter: blur(20px) !important; border-right: 1px solid rgba(56, 189, 248, 0.2) !important; height: 100vh !important; padding: 20px; overflow-y: auto; }
-.logo-area { font-family: 'Orbitron', sans-serif; background: linear-gradient(90deg, var(--primary), #818cf8); -webkit-background-clip: text; -webkit-text-fill-color: transparent; text-align: center; padding: 10px 0 30px 0; font-size: 24px; }
-.input-area { background: rgba(30, 41, 59, 0.9) !important; border: 1px solid var(--primary) !important; border-radius: 20px !important; box-shadow: 0 0 20px rgba(56, 189, 248, 0.2) !important; margin-top: 15px; }
-.orange-btn { background: linear-gradient(135deg, #f97316 0%, #fb923c 100%) !important; color: white !important; font-weight: bold !important; border: none !important; border-radius: 12px !important; margin-bottom: 20px; }
-footer { display: none !important; }
+.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; }
 """
 
-# 4. LOGIC XỬ LÝ
-def process_content(message):
+def chat_engine(message, history, session_id, profile: gr.OAuthProfile | None):
     text = message.get("text", "")
-    files = message.get("files", [])
-    combined_prompt = text
-    for f in files:
-        name = os.path.basename(f)
-        if f.lower().endswith(('.lua', '.py', '.txt', '.js', '.html', '.css')):
-            try:
-                with open(f, "r", encoding="utf-8") as file:
-                    combined_prompt += f"\n\n[FILE: {name}]\n{file.read()}"
-            except: pass
-    return combined_prompt
-
-def chat_engine(message, history, session_id):
-    full_prompt = process_content(message)
-    if not full_prompt.strip(): 
-        yield gr.update(value={"text": "", "files": []}), history, session_id
-        return
-
-    messages = [{"role": "system", "content": "Bạn là NodeAI Genesis. Chuyên gia Roblox Studio và Web Development."}]
+    if not text.strip(): return gr.update(value={"text": "", "files": []}), history, session_id
+
+    # Xác định Master hay Khách
+    user_name = profile.username if profile else "Guest"
+    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Đang hỗ trợ: {user_name}."}]
+    
     for msg in history:
         messages.append({"role": msg["role"], "content": msg["content"]})
-    
-    messages.append({"role": "user", "content": full_prompt})
-    
-    # Gradio 6 dùng list of dicts mặc định
-    new_history = history + [{"role": "user", "content": message.get("text") or "Đã gửi tệp tin"}]
+    messages.append({"role": "user", "content": text})
+
+    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)*"
+    
     yield gr.update(value={"text": "", "files": []}), new_history + [{"role": "assistant", "content": "..."}], session_id
     
     try:
         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}], session_id
+                yield gr.update(), new_history + [{"role": "assistant", "content": ans + (save_status if not token.choices[0].delta.content else "")}], session_id
         
-        save_to_db(session_id, (message.get("text") or "Code Analysis")[:25], new_history + [{"role": "assistant", "content": ans}])
+        # CHỈ LƯU VÀO DB 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
 
-# 5. UI GIAO DIỆN (Đã sửa lỗi gr.Chatbot và gr.Blocks)
-with gr.Blocks(title="NodeAI Genesis Pro") as demo:
+# 4. UI
+with gr.Blocks() as demo:
     session_id = gr.State(str(uuid.uuid4()))
     db_state = gr.State([])
 
     with gr.Row():
         with gr.Column(scale=1, elem_classes="sidebar-node"):
             gr.HTML("<div class='logo-area'>NODEAI GENESIS</div>")
-            new_chat_btn = gr.Button("＋ WORKSPACE MỚI", elem_classes="orange-btn")
+            
+            gr.LoginButton(value="Sign in with Hugging Face", size="sm")
+            
+            # Hiển thị trạng thái đăng nhập trực quan
+            with gr.Group():
+                status_html = gr.HTML()
+            
+            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()
             @gr.render(inputs=[db_state])
-            def render_history(items):
+            def show_history(items):
                 with gr.Column():
-                    for item in items or []:
-                        b = gr.Button(f"◈ {item['title']}", variant="ghost")
-                        b.click(lambda h=item['history'], s=item['session_id']: (h, s), None, [chatbot, session_id])
+                    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])
 
         with gr.Column(scale=4):
-            # FIX: Bỏ tham số 'type="messages"' vì bản 6.0 đã mặc định cấu trúc này
-            chatbot = gr.Chatbot(show_label=False, height="80vh")
-            
-            with gr.Row(elem_classes="input-area"):
-                msg_input = gr.MultimodalTextbox(
-                    show_label=False, 
-                    placeholder="Hỏi đáp hoặc gửi code Lua tại đây...",
-                    file_types=["image", ".lua", ".py", ".js", ".html"],
-                    container=False
-                )
-
-    demo.load(lambda: get_db_history(), None, db_state)
-    
-    msg_input.submit(
-        chat_engine, 
-        [msg_input, chatbot, session_id], 
-        [msg_input, chatbot, session_id]
-    ).then(lambda: get_db_history(), None, db_state)
+            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):
+        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
+    )
     
+    # Nút chat mới
     new_chat_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
 
 if __name__ == "__main__":
-    # FIX: Chuyển css xuống launch()
     demo.launch(css=custom_css)