Update app.py

app.py

@@ -4,12 +4,11 @@ import os
 import uuid
 import json
 
-# 1. CẤU HÌNH MODEL (Sử dụng model ổn định nhất của Hugging Face)
+# 1. CẤU HÌNH MODEL (Model Zephyr cực kỳ ổn định cho Chat)
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-# Model này cực kỳ ổn định và ít khi bị lỗi "Not supported"
-MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.3" 
+MODEL_ID = "HuggingFaceH4/zephyr-7b-beta" 
 client = InferenceClient(model=MODEL_ID, token=HF_TOKEN)
-DB_FILE = "nodeai_pro_vault.json"
+DB_FILE = "nodeai_ultimate_vault.json"
 
 def load_vault():
     if os.path.exists(DB_FILE):
@@ -22,21 +21,31 @@ def save_vault(data):
     with open(DB_FILE, "w", encoding="utf-8") as f:
         json.dump(data, f, ensure_ascii=False, indent=4)
 
-# 2. UI GLASSMORPHISM PLATINUM
+# 2. UI GLASSMORPHISM DIAMOND
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@800&family=Plus+Jakarta+Sans:wght@400;600&display=swap');
-.gradio-container { background: radial-gradient(circle at 50% 0%, #0f172a 0%, #020617 100%) !important; font-family: 'Plus Jakarta Sans', sans-serif !important; }
-.sidebar-node { position: fixed !important; left: 0 !important; top: 0 !important; width: 280px !important; height: 100vh !important; background: rgba(15, 23, 42, 0.7) !important; backdrop-filter: blur(20px) !important; border-right: 1px solid rgba(255, 255, 255, 0.1) !important; z-index: 1000 !important; padding: 25px !important; }
+.gradio-container { background: radial-gradient(circle at 50% 0%, #0f172a 0%, #020617 100%) !important; font-family: 'Plus Jakarta Sans', sans-serif !important; color: white !important; }
+.sidebar-node { position: fixed !important; left: 0 !important; top: 0 !important; width: 280px !important; height: 100vh !important; background: rgba(15, 23, 42, 0.7) !important; backdrop-filter: blur(25px) !important; border-right: 1px solid rgba(255, 255, 255, 0.1) !important; z-index: 1000 !important; padding: 25px !important; }
 .logo-text { font-family: 'Orbitron', sans-serif !important; background: linear-gradient(90deg, #38bdf8, #818cf8); -webkit-background-clip: text; -webkit-text-fill-color: transparent; font-size: 22px; text-align: center; margin-bottom: 25px; }
-.orange-btn { background: linear-gradient(135deg, #f97316 0%, #fb923c 100%) !important; border: none !important; color: white !important; font-weight: 600 !important; border-radius: 12px !important; box-shadow: 0 4px 15px rgba(249, 115, 22, 0.3) !important; }
+.orange-btn { background: linear-gradient(135deg, #f97316 0%, #fb923c 100%) !important; border: none !important; color: white !important; font-weight: 600 !important; border-radius: 12px !important; cursor: pointer !important; }
 .chat-main { margin-left: 280px !important; width: calc(100% - 280px) !important; height: 100vh !important; padding: 0 20px !important; box-sizing: border-box !important; }
 .gr-chatbot { background: transparent !important; border: none !important; height: 75vh !important; margin-bottom: 110px !important; }
-.input-fixed-bottom { position: fixed !important; bottom: 25px !important; left: 50% !important; transform: translateX(calc(-50% + 140px)) !important; width: calc(100% - 340px) !important; max-width: 800px !important; background: rgba(30, 41, 59, 0.9) !important; border: 1px solid rgba(56, 189, 248, 0.3) !important; border-radius: 20px !important; padding: 5px 15px !important; z-index: 2000 !important; }
+.input-fixed-bottom { position: fixed !important; bottom: 25px !important; left: 50% !important; transform: translateX(calc(-50% + 140px)) !important; width: calc(100% - 340px) !important; max-width: 800px !important; background: rgba(30, 41, 59, 0.9) !important; backdrop-filter: blur(10px) !important; border: 1px solid rgba(56, 189, 248, 0.3) !important; border-radius: 20px !important; padding: 5px 15px !important; z-index: 2000 !important; }
 footer { display: none !important; }
 @media screen and (max-width: 768px) { .sidebar-node { display: none !important; } .chat-main { margin-left: 0 !important; width: 100% !important; } .input-fixed-bottom { transform: translateX(-50%) !important; width: 92% !important; } }
 """
 
-# 3. LOGIC XỬ LÝ ĐA PHƯƠNG TIỆN (Multimodal)
+# 3. LOGIC XỬ LÝ FILE VÀ CHAT
+def read_file_content(file_path):
+    try:
+        # Nếu là file văn bản, đọc nội dung bên trong
+        if file_path.lower().endswith(('.txt', '.py', '.lua', '.html', '.css', '.js', '.json')):
+            with open(file_path, "r", encoding="utf-8") as f:
+                return f"\n--- Nội dung file {os.path.basename(file_path)} ---\n{f.read()}\n----------------------\n"
+        return f"\n[Gửi kèm file: {os.path.basename(file_path)} (Định dạng này AI chỉ ghi nhận tên)]\n"
+    except:
+        return ""
+
 def chat_fn(message, history, session_id, profile: gr.OAuthProfile | None):
     username = profile.username if profile else "Master_Phuc"
     user_text = message.get("text", "")
@@ -46,32 +55,26 @@ def chat_fn(message, history, session_id, profile: gr.OAuthProfile | None):
         yield {"text": ""}, history, session_id, ""
         return
 
-    # Xử lý nội dung gửi đi
-    prompt_content = user_text
-    if files:
-        file_list = [os.path.basename(f) for f in files]
-        prompt_content += f"\n\n[Hệ thống: Người dùng đã gửi các file: {', '.join(file_list)}. Hãy phân tích nếu đó là code hoặc văn bản.]"
+    # Tổng hợp nội dung: Văn bản + Nội dung bên trong file
+    full_prompt = user_text
+    for f in files:
+        full_prompt += read_file_content(f)
 
     vault = load_vault()
     if username not in vault: vault[username] = {}
     if session_id not in vault[username]:
         vault[username][session_id] = {"title": user_text[:20] if user_text else "File Analysis", "history": []}
 
-    # Xây dựng hội thoại cho model text-only nhưng thông minh
-    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {username}. Bạn cực giỏi về lập trình Roblox và Web."}]
+    # Format tin nhắn chuẩn Zephyr
+    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {username}. Bạn hỗ trợ chuyên sâu về Roblox (Lua) và Web Development. Hãy đọc kỹ nội dung file nếu Master gửi kèm."}]
     for msg in history:
-        # Đảm bảo nội dung là string để tránh lỗi model text-only
-        content = msg["content"]
-        if isinstance(content, list):
-            content = content[0].get("text", "") if content else ""
-        messages.append({"role": msg["role"], "content": content})
+        messages.append({"role": msg["role"], "content": msg["content"]})
     
-    messages.append({"role": "user", "content": prompt_content})
+    messages.append({"role": "user", "content": full_prompt})
 
     try:
         ans = ""
-        # Hiển thị tin nhắn người dùng (kèm ảnh) lên UI
-        # Gradio tự động hiển thị ảnh nếu chúng ta gửi object message vào history
+        # Hiển thị tin nhắn (kèm ảnh/icon file) lên giao diện
         new_h = history + [{"role": "user", "content": message}]
         
         for token in client.chat_completion(messages=messages, max_tokens=2048, stream=True):
@@ -84,10 +87,10 @@ def chat_fn(message, history, session_id, profile: gr.OAuthProfile | None):
         save_vault(vault)
         yield {"text": ""}, final_h, session_id, ans
     except Exception as e:
-        yield {"text": ""}, history + [{"role": "assistant", "content": f"Lỗi kết nối: {str(e)}"}], session_id, ""
+        yield {"text": ""}, history + [{"role": "assistant", "content": f"Lỗi hệ thống: {str(e)}"}], session_id, ""
 
-# 4. GIAO DIỆN
-with gr.Blocks(title="NodeAI Genesis Pro") as demo:
+# 4. DỰNG GIAO DIỆN
+with gr.Blocks(title="NodeAI Ultimate") as demo:
     session_id = gr.State(str(uuid.uuid4()))
     tts_buffer = gr.Textbox(visible=False)
     vault_state = gr.State(load_vault())
@@ -95,7 +98,7 @@ with gr.Blocks(title="NodeAI Genesis Pro") as demo:
     with gr.Row():
         with gr.Column(elem_classes="sidebar-node"):
             gr.HTML("<div class='logo-text'>NODEAI GENESIS</div>")
-            gr.LoginButton(value="Xác thực Nick")
+            gr.LoginButton(value="Xác thực")
             new_btn = gr.Button("＋ NEW WORKSPACE", elem_classes="orange-btn")
             @gr.render(inputs=[vault_state]) 
             def render_history(current_vault, profile: gr.OAuthProfile | None):
@@ -111,10 +114,8 @@ with gr.Blocks(title="NodeAI Genesis Pro") as demo:
             audio_out = gr.Audio(autoplay=True, visible=False)
             with gr.Row(elem_classes="input-fixed-bottom"):
                 msg_input = gr.MultimodalTextbox(
-                    show_label=False, 
-                    placeholder="Gửi file hoặc nhắn tin cho AI...", 
-                    container=False,
-                    file_types=["image", ".pdf", ".txt", ".py", ".lua"] # Hỗ trợ file Roblox Lua
+                    show_label=False, placeholder="Gửi script Roblox hoặc tin nhắn...", container=False,
+                    file_types=["image", ".lua", ".py", ".txt", ".js"]
                 )
 
     msg_input.submit(