Update app.py

app.py

@@ -3,45 +3,54 @@ from huggingface_hub import InferenceClient
 from supabase import create_client, Client
 import os
 import uuid
-import json
-import base64
 
-# 1. CẤU HÌNH CORE (Lấy từ Secrets của Hugging Face)
+# 1. CẤU HÌNH CORE & SUPABASE
 URL = os.environ.get("SUPABASE_URL")
 KEY = os.environ.get("SUPABASE_KEY")
 HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN")
 
-# Khởi tạo siêu máy chủ
-supabase: Client = create_client(URL, KEY)
+# Định danh mặc định
+MASTER_ID = "lamgamepro"
+
+# Khởi tạo Memory (Supabase) an toàn
+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. HÀM XỬ LÝ DATABASE (Supabase Vĩnh Cửu)
-def get_user_history(username):
+# 2. HỆ THỐNG MEMORY (Tích hợp sâu vào script)
+def get_db_history():
+    if not supabase: return []
     try:
-        res = supabase.table("chats").select("*").eq("username", username).order("created_at", desc=True).execute()
+        res = supabase.table("chats").select("*").eq("username", MASTER_ID).order("created_at", desc=True).execute()
         return res.data
-    except: return []
+    except Exception as e:
+        print(f"Lỗi tải DB: {e}")
+        return []
 
-def save_to_db(username, session_id, title, history):
+def save_to_db(session_id, title, history):
+    if not supabase: return
     try:
-        data = {"username": username, "session_id": session_id, "title": title, "history": history}
+        data = {"username": MASTER_ID, "session_id": session_id, "title": title, "history": history}
         supabase.table("chats").upsert(data, on_conflict="session_id").execute()
-    except Exception as e: print(f"DB Error: {e}")
+    except Exception as e:
+        print(f"Lỗi lưu DB: {e}")
 
-# 3. GIAO DIỆN SIÊU CẤP (GLASSMORPHISM PRO)
+# 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');
 :root { --primary: #38bdf8; --accent: #f97316; }
-.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.7) !important; backdrop-filter: blur(20px) !important; border-right: 1px solid rgba(56, 189, 248, 0.2) !important; height: 100vh !important; }
-.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: 20px 0; }
-.chat-bubble { border-radius: 15px !important; border: 1px solid rgba(255,255,255,0.1) !important; }
-.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; }
-.orange-btn { background: linear-gradient(135deg, #f97316 0%, #fb923c 100%) !important; color: white !important; font-weight: bold !important; border: none !important; }
+.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; }
 """
 
-# 4. LOGIC XỬ LÝ ĐA PHƯƠNG TIỆN (Multimodal)
+# 4. LOGIC PHÂN TÍCH ĐA PHƯƠNG TIỆN (Code/Text/Ảnh)
 def process_content(message):
     text = message.get("text", "")
     files = message.get("files", [])
@@ -52,92 +61,97 @@ def process_content(message):
         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()}"
+                    combined_prompt += f"\n\n[=== NỘI DUNG FILE: {name} ===]\n{file.read()}"
             except: pass
         elif f.lower().endswith(('.png', '.jpg', '.jpeg')):
-            combined_prompt += f"\n\n[ Master đã gửi 1 ảnh: {name} ]"
+            combined_prompt += f"\n\n[Hệ thống: Master đã gửi kèm ảnh {name}]"
     return combined_prompt
 
-def chat_engine(message, history, session_id, profile: gr.OAuthProfile | None):
-    username = profile.username if profile else "Master_Phuc"
+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
+
+    # Chuẩn bị Prompt cho AI
+    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master của bạn là {MASTER_ID}. Bạn là chuyên gia về Roblox Studio Lua, Three.js và Web Development."}]
     
-    if not full_prompt.strip(): yield "", history, session_id, "" ; return
+    # Đọc lịch sử (Chuẩn Gradio 6)
+    for msg in history:
+        content = msg["content"]
+        if isinstance(content, str):
+            messages.append({"role": msg["role"], "content": content})
 
-    # Build Prompt
-    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {username}. Bạn là chuyên gia Roblox và Web. Hãy phân tích kỹ code Master gửi."}]
-    for msg in history: messages.append({"role": msg["role"], "content": msg["content"]})
     messages.append({"role": "user", "content": full_prompt})
-
+    
+    # Hiển thị trên UI
+    display_history = history + [{"role": "user", "content": message.get("text") or f"Đã gửi {len(message.get('files', []))} file đính kèm"}]
     ans = ""
-    # Cập nhật UI ngay lập tức
-    display_history = history + [{"role": "user", "content": message.get("text") or "Gửi file"}]
     
     try:
+        # Xóa thanh nhập liệu ngay khi bắt đầu xử lý
+        yield gr.update(value={"text": "", "files": []}), display_history + [{"role": "assistant", "content": "..."}], session_id
+        
         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 "", display_history + [{"role": "assistant", "content": ans}], session_id, ""
+                yield gr.update(), display_history + [{"role": "assistant", "content": ans}], session_id
         
-        # Lưu vào Supabase sau khi hoàn thành stream
+        # Lưu vào Supabase khi hoàn thành
         final_history = display_history + [{"role": "assistant", "content": ans}]
-        title = message.get("text", "Workspace Analysis")[:25]
-        save_to_db(username, session_id, title, final_history)
-        yield "", final_history, session_id, ans
+        title = message.get("text", "Phân tích file")[:25] if message.get("text") else "Phân tích Code"
+        save_to_db(session_id, title, final_history)
+        
+        yield gr.update(), final_history, session_id
     except Exception as e:
-        yield "", history + [{"role": "assistant", "content": f"⚠️ Lỗi: {str(e)}"}], session_id, ""
+        yield gr.update(), history + [{"role": "assistant", "content": f"⚠️ Lỗi kết nối AI: {str(e)}"}], session_id
 
-# 5. DỰNG WEB APP
-with gr.Blocks(title="NodeAI Diamond Pro", css=custom_css) as demo:
+# 5. XÂY DỰNG GIAO DIỆN CHÍNH
+with gr.Blocks(title="NodeAI Genesis Pro", css=custom_css) as demo:
     session_id = gr.State(str(uuid.uuid4()))
-    tts_buffer = gr.Textbox(visible=False)
     db_state = gr.State([])
 
     with gr.Row():
-        # Sidebar History
+        # Cột Sidebar: Lịch sử Memory
         with gr.Column(scale=1, elem_classes="sidebar-node"):
-            gr.HTML("<div class='logo-area'><h1>NODEAI GENESIS</h1></div>")
-            gr.LoginButton()
-            new_chat = gr.Button("＋ NEW WORKSPACE", elem_classes="orange-btn")
+            gr.HTML("<div class='logo-area'>NODEAI GENESIS</div>")
+            new_chat_btn = gr.Button("＋ WORKSPACE MỚI", elem_classes="orange-btn")
+            gr.Markdown("### 🗄️ Database Memory")
             
-            gr.Markdown("### 🛠 Lịch sử Cloud")
             @gr.render(inputs=[db_state])
-            def render_history(items, profile: gr.OAuthProfile | None):
+            def render_history(items):
                 with gr.Column():
+                    if not items:
+                        gr.Markdown("*Chưa có dữ liệu hoặc lỗi kết nối Supabase.*")
                     for item in items:
                         b = gr.Button(f"◈ {item['title']}", variant="ghost")
-                        b.click(lambda h=item['history'], s=item['session_id']: (h, s), 
-                                None, [chatbot, session_id])
+                        b.click(lambda h=item['history'], s=item['session_id']: (h, s), None, [chatbot, session_id])
 
-        # Main Chat Area
+        # Cột Chatbot: Tương tác chính
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(show_label=False, bubble_full_width=False, height="75vh")
-            audio_player = gr.Audio(autoplay=True, visible=False)
+            chatbot = gr.Chatbot(type="messages", show_label=False, height="80vh")
             
             with gr.Row(elem_classes="input-area"):
                 msg_input = gr.MultimodalTextbox(
                     show_label=False, 
-                    placeholder="Dán script Roblox hoặc hỏi Master Phúc...",
-                    file_types=["image", ".lua", ".py", ".txt"],
+                    placeholder="Dán mã Roblox Lua, tải ảnh hoặc hỏi đáp tại đây...",
+                    file_types=["image", ".lua", ".py", ".txt", ".js", ".html", ".css"],
                     container=False
                 )
 
-    # Khởi tạo dữ liệu
-    def refresh(profile: gr.OAuthProfile | None):
-        user = profile.username if profile else "Master_Phuc"
-        return get_user_history(user)
-
-    demo.load(refresh, None, db_state)
+    # Khởi động: Load dữ liệu từ Supabase
+    def refresh_db(): return get_db_history()
+    demo.load(refresh_db, None, db_state)
     
-    # Event Handlers
+    # Xử lý sự kiện gửi tin nhắn
     msg_input.submit(
-        chat_engine, [msg_input, chatbot, session_id], [msg_input, chatbot, session_id, tts_buffer]
-    ).then(refresh, None, db_state).then(
-        lambda x: f"https://translate.google.com/translate_tts?ie=UTF-8&tl=vi&client=tw-ob&q={x[:200]}" if x else None,
-        tts_buffer, audio_player
-    )
+        chat_engine, 
+        [msg_input, chatbot, session_id], 
+        [msg_input, chatbot, session_id]
+    ).then(refresh_db, None, db_state) # Cập nhật lại danh sách bên trái
     
-    new_chat.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
+    # Nút tạo Chat mới
+    new_chat_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
 
 if __name__ == "__main__":
     demo.launch()