Update app.py

app.py

@@ -1,131 +1,105 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-from supabase import create_client, Client
 import os
 import uuid
 
-# 1. SETUP
-URL = os.environ.get("SUPABASE_URL")
-KEY = os.environ.get("SUPABASE_KEY")
+# 1. CẤU HÌNH AI
 HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN")
-
-try:
-    supabase: Client = create_client(URL, KEY) if URL and KEY else None
-except:
-    supabase = None
-
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
-# 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'].forEach(s => {
-            const el = document.querySelector(s);
-            if (el) el.remove();
-        });
-    });
-    observer.observe(document.body, { childList: true, subtree: true });
-" style="display:none;">
-"""
-
-custom_css = """
-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.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; 
-    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; }
-"""
-
-# 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("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):
+# 2. MẢNG MEMORY TRONG SCRIPT (Biến cục bộ qua gr.State)
+# Memory này sẽ tồn tại suốt phiên làm việc của Master
+def chat_engine(message, history, profile: gr.OAuthProfile | None):
     text = message.get("text", "").strip()
-    if not text: yield gr.update(value={"text": "", "files": []}), history, session_id; return
+    if not text: 
+        yield gr.update(value={"text": "", "files": []}), history
+        return
 
-    # Cấu hình bối cảnh cho AI
-    user_name = profile.username if profile else "Guest"
-    system_msg = {"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {user_name}."}
+    # Khởi tạo tin nhắn hệ thống
+    user_name = profile.name if profile else "Master"
+    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {user_name}."}]
+    
+    # Đổ mảng memory từ history vào tin nhắn gửi đi
+    for msg in history:
+        messages.append({"role": msg["role"], "content": msg["content"]})
     
-    # Tạo prompt từ lịch sử (Memory)
-    messages = [system_msg] + history + [{"role": "user", "content": text}]
+    messages.append({"role": "user", "content": text})
     
+    # Cập nhật mảng memory tạm thời (history)
     new_history = history + [{"role": "user", "content": text}]
     ans = ""
     
-    yield gr.update(value={"text": "", "files": []}), new_history + [{"role": "assistant", "content": "..."}], session_id
+    # Phản hồi streaming
+    yield gr.update(value={"text": "", "files": []}), new_history + [{"role": "assistant", "content": "..."}]
     
     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:
-            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()
-            
+                yield gr.update(), new_history + [{"role": "assistant", "content": ans}]
     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: {str(e)}"}]
+
+# 3. GIAO DIỆN & CSS (Giữ phong cách sạch sẽ)
+custom_css = """
+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.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; 
+    font-size: 22px; text-align: center; font-weight: bold; margin-bottom: 20px; 
+}
+.custom-warning { text-align: center; color: #94a3b8; font-size: 12px; margin: 10px 0; font-style: italic; }
+"""
+
+js_cleaner = """
+<img src="x" onerror="
+    const observer = new MutationObserver(() => {
+        ['footer', '.built-with', 'header', '#header'].forEach(s => {
+            const el = document.querySelector(s); if (el) el.remove();
+        });
+    });
+    observer.observe(document.body, { childList: true, subtree: true });
+" style="display:none;">
+"""
 
-# 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([])
-
+    
     with gr.Row():
+        # Sidebar (Chỉ để trang trí hoặc bổ sung tính năng sau)
         with gr.Column(scale=1, elem_classes="sidebar-node"):
             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("### 🗄️ SQL Memory")
-            history_list = gr.Dataset(components=[gr.Textbox(visible=False)], samples=[], type="index")
+            new_chat_btn = gr.Button("＋ CLEAR MEMORY", variant="stop")
+            gr.Markdown("---")
+            gr.Markdown("💡 **Script Memory**: Đang hoạt động cục bộ. Tốc độ cực nhanh nhưng sẽ reset khi tải lại trang.")
 
+        # Vùng Chat chính
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(show_label=False, height="76vh")
+            # chatbot lưu trữ mảng history ngay trong component
+            chatbot = gr.Chatbot(show_label=False, height="78vh", type="messages")
+            
             gr.HTML("<div class='custom-warning'>NodeAI không thể chắc chắn 100%, hãy rà soát lại khi bạn có chuyện gì quan trọng!</div>")
-            msg_input = gr.MultimodalTextbox(show_label=False, placeholder="Hỏi NodeAI...", container=False)
-
-    def refresh_data(profile: gr.OAuthProfile | None):
-        if not profile: return [], gr.Dataset(samples=[])
-        data = get_db_history(profile)
-        return data, gr.Dataset(samples=[[item['title']] for item in data])
+            
+            msg_input = gr.MultimodalTextbox(
+                show_label=False, 
+                placeholder="Nhập nội dung tại đây...", 
+                container=False
+            )
 
-    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]
+    # Xử lý sự kiện
+    # Truyền chatbot trực tiếp vào làm history
+    msg_input.submit(
+        chat_engine, 
+        [msg_input, chatbot], 
+        [msg_input, chatbot]
     )
     
-    new_chat_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
+    # Nút xóa sạch mảng memory trong script
+    new_chat_btn.click(lambda: [], None, chatbot)
 
 if __name__ == "__main__":
     demo.queue().launch()