Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import uuid
 import json
 
-# Import an toàn
+# Import thư viện hỗ trợ
 try:
     from duckduckgo_search import DDGS
 except ImportError:
@@ -15,12 +15,11 @@ try:
 except ImportError:
     gTTS = None
 
-# 1. CẤU HÌNH (Đổi sang model ổn định nhất)
+# 1. CẤU HÌNH HỆ THỐNG
 USER_ID, USER_PASS = "admin", "123456"
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-# Sử dụng model Qwen2.5-7B-Instruct để đảm bảo chạy 100% trên mọi Server
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
-DB_FILE = "nodeai_genesis_memory.json"
+DB_FILE = "nodeai_platinum_memory.json"
 
 def load_db():
     if os.path.exists(DB_FILE):
@@ -33,12 +32,12 @@ def save_db(data):
     with open(DB_FILE, "w", encoding="utf-8") as f:
         json.dump(data, f, ensure_ascii=False, indent=4)
 
-# 2. GUI DESIGN (Cyberpunk Neon)
+# 2. GUI DESIGN (PLATINUM CYBERPUNK)
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;700&family=Plus+Jakarta+Sans:wght@300;400;600&display=swap');
 
 .gradio-container {
-    background: radial-gradient(circle at top right, #1a1a2e, #0f0f1a, #050505) !important;
+    background: radial-gradient(circle at top right, #0d0d1a, #050505) !important;
     font-family: 'Plus Jakarta Sans', sans-serif !important;
 }
 
@@ -46,52 +45,55 @@ custom_css = """
     position: fixed !important;
     left: 10px !important; top: 10px !important;
     width: 280px !important; height: calc(100vh - 20px) !important;
-    background: rgba(15, 15, 26, 0.8) !important;
-    backdrop-filter: blur(15px) !important;
-    border: 1px solid rgba(255, 255, 255, 0.1) !important;
+    background: rgba(10, 10, 15, 0.9) !important;
+    backdrop-filter: blur(20px) !important;
+    border: 1px solid rgba(0, 242, 254, 0.2) !important;
     border-radius: 20px !important;
     z-index: 999 !important;
-    padding: 25px !important;
-    overflow-y: auto !important;
+    padding: 20px !important;
 }
 
 .nodeai-logo {
     font-family: 'Orbitron', sans-serif !important;
-    background: linear-gradient(90deg, #00f2fe, #4facfe);
+    background: linear-gradient(90deg, #00f2fe, #7ef2ff);
     -webkit-background-clip: text;
     -webkit-text-fill-color: transparent;
-    font-size: 28px !important;
+    font-size: 26px !important;
     font-weight: 800 !important;
     text-align: center;
-    margin-bottom: 20px;
+    margin-bottom: 25px;
 }
 
 .chat-col {
     margin-left: 310px !important;
     width: calc(100% - 330px) !important;
     height: 100vh !important;
-    padding: 20px !important;
 }
 
 .gr-chatbot { background: transparent !important; border: none !important; }
 
-/* Làm đẹp khu vực nhập liệu và Mic */
-.input-row {
-    display: flex !important;
-    gap: 10px !important;
-    align-items: center !important;
+/* Control Panel cho Input */
+.control-panel {
     background: rgba(255, 255, 255, 0.03) !important;
     border: 1px solid rgba(255, 255, 255, 0.1) !important;
-    border-radius: 30px !important;
-    padding: 5px 15px !important;
+    border-radius: 25px !important;
+    padding: 10px !important;
+    margin-bottom: 20px !important;
 }
 
-.input-box { flex: 1 !important; border: none !important; background: transparent !important; }
+.hist-btn {
+    background: rgba(255, 255, 255, 0.03) !important;
+    border: 1px solid rgba(0, 242, 254, 0.1) !important;
+    border-radius: 12px !important;
+    margin-bottom: 8px !important;
+    transition: 0.3s;
+}
+.hist-btn:hover { border-color: #00f2fe !important; background: rgba(0, 242, 254, 0.1) !important; }
 
 footer { display: none !important; }
 """
 
-# 3. LOGIC XỬ LÝ
+# 3. CORE LOGIC
 def search_web(query):
     if not DDGS: return ""
     try:
@@ -101,22 +103,17 @@ def search_web(query):
     except: return ""
 
 def chat_fn(message, history, session_id, store):
-    # Xử lý nếu message là text hoặc từ microphone
-    user_text = ""
-    files = []
-    
-    if isinstance(message, dict):
-        user_text = message.get("text", "")
-        files = message.get("files", [])
+    user_text = message.get("text", "") if isinstance(message, dict) else message
+    files = message.get("files", []) if isinstance(message, dict) else []
     
     if not user_text and not files: return {"text": ""}, history, session_id, store, None
     
     if session_id not in store:
-        store[session_id] = {"title": user_text[:20] if user_text else "New Chat", "history": []}
+        store[session_id] = {"title": user_text[:20] if user_text else "Nova Task", "history": []}
     
-    web_data = search_web(user_text) if any(x in user_text.lower() for x in ["tìm", "tra", "web", "giá"]) else ""
+    web_info = search_web(user_text) if any(x in user_text.lower() for x in ["tìm", "tra", "web", "giá"]) else ""
     
-    api_messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: Trần Đình Phúc. Web: {web_data}"}]
+    api_messages = [{"role": "system", "content": f"Bạn là NodeAI Platinum. Master: Trần Đình Phúc. Web Context: {web_info}"}]
     for msg in history:
         api_messages.append({"role": msg["role"], "content": msg["content"]})
     
@@ -130,12 +127,12 @@ def chat_fn(message, history, session_id, store):
                 ans += token.choices[0].delta.content
                 yield {"text": ""}, new_h + [{"role": "assistant", "content": ans}], session_id, store, None
         
-        final_history = new_h + [{"role": "assistant", "content": ans}]
-        store[session_id]["history"] = final_history
+        final_h = new_h + [{"role": "assistant", "content": ans}]
+        store[session_id]["history"] = final_h
         save_db(store)
-        yield {"text": ""}, final_history, session_id, store, ans
+        yield {"text": ""}, final_h, session_id, store, ans
     except Exception as e:
-        yield {"text": ""}, history + [{"role": "assistant", "content": f"Lỗi: {str(e)}"}], session_id, store, None
+        yield {"text": ""}, history + [{"role": "assistant", "content": f"Neural Error: {str(e)}"}], session_id, store, None
 
 def tts_engine(text):
     if not text or not gTTS: return None
@@ -146,7 +143,7 @@ def tts_engine(text):
     except: return None
 
 # 4. INTERFACE
-with gr.Blocks(title="NodeAI Genesis") as demo:
+with gr.Blocks(title="NodeAI Platinum") as demo:
     chat_store = gr.State(load_db())
     session_id = gr.State(str(uuid.uuid4()))
     
@@ -159,32 +156,36 @@ with gr.Blocks(title="NodeAI Genesis") as demo:
                 @gr.render(inputs=chat_store)
                 def render_history(store):
                     for sid in reversed(list(store.keys())):
-                        btn = gr.Button(f"◈ {store[sid]['title']}", variant="ghost")
+                        btn = gr.Button(f"◈ {store[sid]['title']}", elem_classes="hist-btn")
                         btn.click(lambda s=sid: (store[s]["history"], s), None, [chatbot, session_id])
         
-        # MAIN CHAT
+        # MAIN CONTENT
         with gr.Column(elem_classes="chat-col"):
-            chatbot = gr.Chatbot(show_label=False, height="75vh")
+            chatbot = gr.Chatbot(show_label=False, height="72vh")
             audio_out = gr.Audio(autoplay=True, visible=False)
             
-            # Khu vực nhập liệu có Mic
-            with gr.Row(elem_classes="input-row"):
-                msg_input = gr.MultimodalTextbox(
-                    show_label=False, 
-                    placeholder="Command NodeAI hoặc dùng Mic...", 
-                    container=False,
-                    elem_classes="input-box",
-                    sources=["upload", "microphone"] # Kích hoạt nút Mic
-                )
-            
-            gr.HTML("<p style='text-align:center; color:rgba(255,255,255,0.1); font-size:11px; margin-top:10px;'>GENESIS V30.3 • BY TRẦN ĐÌNH PHÚ</p>")
-
+            with gr.Row(elem_classes="control-panel"):
+                with gr.Column(scale=9):
+                    msg_input = gr.MultimodalTextbox(
+                        show_label=False, placeholder="Command NodeAI Genesis...", container=False
+                    )
+                with gr.Column(scale=1, min_width=60):
+                    # Mic rời cực kỳ nhạy để trình duyệt nhận diện quyền
+                    mic_btn = gr.Audio(sources=["microphone"], type="filepath", label="Mic", show_label=False)
+
+            gr.HTML("<p style='text-align:center; color:rgba(0, 242, 254, 0.2); font-size:11px;'>PLATINUM V30.5 • POWERED BY TRẦN ĐÌNH PHÚ</p>")
+
+    # Xử lý nhập liệu
     msg_input.submit(
-        chat_fn, 
-        [msg_input, chatbot, session_id, chat_store], 
-        [msg_input, chatbot, session_id, chat_store, audio_out]
+        chat_fn, [msg_input, chatbot, session_id, chat_store], [msg_input, chatbot, session_id, chat_store, audio_out]
     ).then(tts_engine, audio_out, audio_out)
     
+    # Xử lý từ Mic (Khi thu âm xong tự động gửi)
+    def process_mic(path):
+        return {"text": "Dịch nội dung từ file âm thanh này và trả lời", "files": [path]}
+    
+    mic_btn.stop_recording(process_mic, mic_btn, msg_input)
+    
     new_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
 
 if __name__ == "__main__":