Update app.py

app.py

@@ -4,7 +4,7 @@ from supabase import create_client, Client
 import os
 import uuid
 
-# 1. CẤU HÌNH HỆ THỐNG
+# 1. SETUP
 URL = os.environ.get("SUPABASE_URL")
 KEY = os.environ.get("SUPABASE_KEY")
 HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN")
@@ -16,11 +16,11 @@ except:
 
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
-# 2. JAVASCRIPT "DIỆT" TẬN GỐC (Dạng chuỗi HTML)
+# 2. JS & CSS (Xóa sạch Header/Footer)
 js_cleaner = """
 <img src="x" onerror="
     const observer = new MutationObserver(() => {
-        ['footer', '.built-with', 'header', '#header', '.share-button', '.gr-pro'].forEach(s => {
+        ['footer', '.built-with', 'header', '#header', '.share-button'].forEach(s => {
             const el = document.querySelector(s);
             if (el) el.remove();
         });
@@ -29,59 +29,72 @@ js_cleaner = """
 " style="display:none;">
 """
 
-# 3. GIAO DIỆN CSS
 custom_css = """
-footer, .built-with, header, #header { display: none !important; visibility: hidden !important; }
+footer, .built-with, header, #header { display: none !important; }
 .gradio-container { max-width: 100% !important; padding: 0 !important; }
-.sidebar-node { background: rgba(15, 23, 42, 0.8) !important; border-right: 1px solid #38bdf833; padding: 20px; }
+.sidebar-node { background: rgba(15, 23, 42, 0.9) !important; border-right: 1px solid #38bdf833; padding: 20px; }
 .logo-area { 
     font-family: 'Orbitron', sans-serif; 
     background: linear-gradient(90deg, #38bdf8, #818cf8); 
-    -webkit-background-clip: text; 
-    -webkit-text-fill-color: transparent; 
+    -webkit-background-clip: text; -webkit-text-fill-color: transparent; 
     font-size: 22px; text-align: center; font-weight: bold; margin-bottom: 20px; 
 }
 .custom-warning { text-align: center; color: #94a3b8; font-size: 12px; margin: 8px 0; font-style: italic; }
 """
 
-# 4. LOGIC XỬ LÝ (Giữ nguyên)
+# 3. DATABASE LOGIC
 def get_db_history(profile: gr.OAuthProfile | None):
     if not supabase or not profile: return []
     try:
-        res = supabase.table("chats").select("*").eq("username", profile.username).order("created_at", desc=True).execute()
+        res = supabase.table("chats").select("*").eq("username", profile.username).order("last_updated", desc=True).execute()
         return res.data
     except: return []
 
 def load_selected_chat(evt: gr.SelectData, db_data):
     selected_item = db_data[evt.index]
+    # Trả về lịch sử từ SQL và session_id tương ứng
     return selected_item['history'], selected_item['session_id']
 
+# 4. CHAT ENGINE (Đọc/Ghi Memory trực tiếp vào SQL)
 def chat_engine(message, history, session_id, profile: gr.OAuthProfile | None):
     text = message.get("text", "").strip()
     if not text: yield gr.update(value={"text": "", "files": []}), history, session_id; return
+
+    # Cấu hình bối cảnh cho AI
     user_name = profile.username if profile else "Guest"
-    messages = [{"role": "system", "content": f"Master: {user_name}."}]
-    for msg in history: messages.append({"role": msg["role"], "content": msg["content"]})
-    messages.append({"role": "user", "content": text})
+    system_msg = {"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {user_name}."}
+    
+    # Tạo prompt từ lịch sử (Memory)
+    messages = [system_msg] + history + [{"role": "user", "content": text}]
+    
     new_history = history + [{"role": "user", "content": text}]
     ans = ""
+    
     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
+        
+        # LƯU VÀO SQL (Memory thực thụ)
         if profile:
-            data = {"username": profile.username, "session_id": session_id, "title": text[:30], "history": new_history + [{"role": "assistant", "content": ans}]}
+            final_history = new_history + [{"role": "assistant", "content": ans}]
+            data = {
+                "username": profile.username, 
+                "session_id": session_id, 
+                "title": text[:30], 
+                "history": final_history
+            }
             supabase.table("chats").upsert(data, on_conflict="session_id").execute()
+            
     except Exception as e:
         yield gr.update(), new_history + [{"role": "assistant", "content": f"Lỗi: {str(e)}"}], session_id
 
-# 5. UI (Bỏ tham số head gây lỗi)
-with gr.Blocks() as demo:
-    # Chèn Script xóa Header/Footer thông qua HTML ẩn
+# 5. UI
+with gr.Blocks(css=custom_css) as demo:
     gr.HTML(js_cleaner, visible=False)
-    
     session_id = gr.State(str(uuid.uuid4()))
     db_state = gr.State([])
 
@@ -90,7 +103,7 @@ with gr.Blocks() as demo:
             gr.HTML("<div class='logo-area'>NODEAI GENESIS</div>")
             gr.LoginButton(value="Đăng nhập", size="sm")
             new_chat_btn = gr.Button("＋ WORKSPACE MỚI", variant="primary")
-            gr.Markdown("### 🗄️ Memory")
+            gr.Markdown("### 🗄️ SQL Memory")
             history_list = gr.Dataset(components=[gr.Textbox(visible=False)], samples=[], type="index")
 
         with gr.Column(scale=4):
@@ -104,9 +117,15 @@ with gr.Blocks() as demo:
         return data, gr.Dataset(samples=[[item['title']] for item in data])
 
     demo.load(refresh_data, None, [db_state, history_list])
+    
+    # Khi nhấn vào lịch sử, Chatbot sẽ hiển thị Memory từ SQL
     history_list.select(fn=load_selected_chat, inputs=[db_state], outputs=[chatbot, session_id])
-    msg_input.submit(chat_engine, [msg_input, chatbot, session_id], [msg_input, chatbot, session_id]).then(refresh_data, None, [db_state, history_list])
+    
+    msg_input.submit(chat_engine, [msg_input, chatbot, session_id], [msg_input, chatbot, session_id]).then(
+        refresh_data, None, [db_state, history_list]
+    )
+    
     new_chat_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
 
 if __name__ == "__main__":
-    demo.queue().launch(css=custom_css)
+    demo.queue().launch()