Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import uuid
 import json
 
-# Import an toàn
+# Thư viện bổ trợ
 try: from duckduckgo_search import DDGS
 except: DDGS = None
 try: from gtts import gTTS
@@ -14,7 +14,7 @@ except: gTTS = None
 USER_ID, USER_PASS = "admin", "123456"
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
-DB_FILE = "nodeai_platinum_v3.json"
+DB_FILE = "nodeai_ultimate_memory.json"
 
 def load_db():
     if os.path.exists(DB_FILE):
@@ -27,100 +27,107 @@ 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 (FIX LỖI LAYOUT & MIC RÕ RÀNG)
+# 2. GUI DESIGN (FIX TOÀN DIỆN)
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@700&family=Plus+Jakarta+Sans:wght@400;600&display=swap');
 
+/* Nền chủ đạo */
 .gradio-container {
-    background: radial-gradient(circle at top right, #0d0d1a, #050505) !important;
+    background: #050505 !important;
     font-family: 'Plus Jakarta Sans', sans-serif !important;
 }
 
+/* Sidebar Pro */
 .sidebar-pro {
     position: fixed !important;
-    left: 10px !important; top: 10px !important;
-    width: 280px !important; height: calc(100vh - 20px) !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;
+    left: 0 !important; top: 0 !important;
+    width: 260px !important; height: 100vh !important;
+    background: #0a0a0f !important;
+    border-right: 1px solid #1a1a20 !important;
     z-index: 1000 !important;
     padding: 20px !important;
 }
 
 .nodeai-logo {
     font-family: 'Orbitron', sans-serif !important;
-    background: linear-gradient(90deg, #00f2fe, #7ef2ff);
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-    font-size: 26px !important;
-    text-align: center; margin-bottom: 25px;
+    color: #00f2fe; font-size: 24px; text-align: center; margin-bottom: 30px;
 }
 
+/* Vùng Chat */
 .chat-col {
-    margin-left: 300px !important;
-    width: calc(100% - 320px) !important;
+    margin-left: 260px !important;
+    width: calc(100% - 260px) !important;
     height: 100vh !important;
+    padding: 20px 40px !important;
 }
 
-.gr-chatbot { background: transparent !important; border: none !important; margin-bottom: 150px !important; }
+.gr-chatbot { 
+    background: transparent !important; 
+    border: none !important; 
+    height: 70vh !important;
+    margin-bottom: 120px !important;
+}
 
-/* THANH NHẬP LIỆU CỐ ĐỊNH - ĐÃ FIX LỖI DỊCH CHUYỂN */
-.control-panel-fixed {
+/* KHÓA CHẶT THANH ĐIỀU KHIỂN Ở DƯỚI ĐÁY */
+.input-area-fixed {
     position: fixed !important;
-    bottom: 25px !important;
-    left: 310px !important;
-    right: 25px !important;
-    background: rgba(20, 20, 35, 0.9) !important;
-    backdrop-filter: blur(15px) !important;
-    border: 1px solid rgba(0, 242, 254, 0.4) !important;
-    border-radius: 35px !important;
-    padding: 12px 25px !important;
-    z-index: 9999 !important;
+    bottom: 30px !important;
+    left: 300px !important;
+    right: 40px !important;
+    z-index: 2000 !important;
     display: flex !important;
-    align-items: center !important;
-    gap: 15px !important;
-    box-shadow: 0 10px 40px rgba(0,0,0,0.5) !important;
+    align-items: flex-end !important;
+    gap: 10px !important;
 }
 
-.input-box { flex: 1 !important; border: none !important; background: transparent !important; }
-
-/* NÚT MIC TO VÀ RÕ RÀNG */
-.mic-container {
-    min-width: 150px !important;
-    background: rgba(255, 255, 255, 0.05) !important;
-    border-radius: 20px !important;
+/* Ô nhập liệu */
+.input-container {
+    flex: 1 !important;
+    background: #121217 !important;
+    border: 1px solid #2a2a35 !important;
+    border-radius: 15px !important;
     padding: 5px !important;
 }
 
+/* Mic gọn gàng */
+.mic-box {
+    width: 60px !important;
+    min-width: 60px !important;
+    height: 60px !important;
+    background: #121217 !important;
+    border: 1px solid #2a2a35 !important;
+    border-radius: 15px !important;
+    overflow: hidden !important;
+    display: flex !important;
+    justify-content: center !important;
+    align-items: center !important;
+}
+
+/* Ẩn bớt các nhãn thừa để tránh nhảy layout */
+.mic-box label, .mic-box .gr-block-label { display: none !important; }
+
 footer { display: none !important; }
 """
 
 # 3. CORE LOGIC
-def search_web(query):
-    if not DDGS: return ""
-    try:
-        with DDGS() as ddgs:
-            results = [f"[{r['title']}]({r['href']}): {r['body']}" for r in ddgs.text(query, max_results=2)]
-        return "\n".join(results)
-    except: return ""
-
 def chat_fn(message, history, session_id, store):
     user_text = message.get("text", "") if isinstance(message, dict) else message
-    if not user_text and not message.get("files", []): return {"text": ""}, history, session_id, store, None
+    files = message.get("files", []) if isinstance(message, dict) else []
     
-    if session_id not in store:
-        store[session_id] = {"title": (user_text[:20] if user_text else "Audio Task"), "history": []}
+    if not user_text and not files: return {"text": ""}, history, session_id, store, None
     
-    web_info = search_web(user_text) if any(x in user_text.lower() for x in ["tìm", "tra", "web"]) else ""
+    if session_id not in store:
+        store[session_id] = {"title": user_text[:20] if user_text else "New Workspace", "history": []}
     
-    api_messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: Trần Đình Phúc. Web: {web_info}"}]
+    api_messages = [{"role": "system", "content": "Bạn là NodeAI Genesis. Master: Trần Đình Phúc."}]
     for msg in history: api_messages.append({"role": msg["role"], "content": msg["content"]})
-    api_messages.append({"role": "user", "content": user_text})
+    
+    # Hỗ trợ gửi text hoặc file/audio
+    api_messages.append({"role": "user", "content": user_text if user_text else "Phân tích nội dung này"})
     
     try:
         ans = ""
-        new_h = history + [{"role": "user", "content": user_text}]
+        new_h = history + [{"role": "user", "content": user_text if user_text else "📎 File/Audio"}]
         for token in client.chat_completion(messages=api_messages, max_tokens=2000, stream=True):
             if token.choices[0].delta.content:
                 ans += token.choices[0].delta.content
@@ -131,7 +138,7 @@ def chat_fn(message, history, session_id, store):
         save_db(store)
         yield {"text": ""}, final_h, session_id, store, ans
     except Exception as e:
-        yield {"text": ""}, history + [{"role": "assistant", "content": str(e)}], session_id, store, None
+        yield {"text": ""}, history + [{"role": "assistant", "content": f"Lỗi: {str(e)}"}], session_id, store, None
 
 def tts_engine(text):
     if not text or not gTTS: return None
@@ -146,37 +153,38 @@ with gr.Blocks(title="NodeAI Genesis") as demo:
     chat_store = gr.State(load_db())
     session_id = gr.State(str(uuid.uuid4()))
     
-    with gr.Row():
-        # SIDEBAR
-        with gr.Column(elem_classes="sidebar-pro"):
-            gr.HTML("<div class='nodeai-logo'>NODEAI</div>")
-            new_btn = gr.Button("＋ NEW WORKSPACE", variant="primary")
-            with gr.Column():
-                @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.click(lambda s=sid: (store[s]["history"], s), None, [chatbot, session_id])
+    # Sidebar
+    with gr.Column(elem_classes="sidebar-pro"):
+        gr.HTML("<div class='nodeai-logo'>NODEAI</div>")
+        new_btn = gr.Button("＋ NEW CHAT", variant="primary")
+        with gr.Column():
+            @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.click(lambda s=sid: (store[s]["history"], s), None, [chatbot, session_id])
+
+    # Main Chat Area
+    with gr.Column(elem_classes="chat-col"):
+        chatbot = gr.Chatbot(show_label=False)
+        audio_out = gr.Audio(autoplay=True, visible=False)
         
-        # MAIN CONTENT
-        with gr.Column(elem_classes="chat-col"):
-            chatbot = gr.Chatbot(show_label=False, height="75vh")
-            audio_out = gr.Audio(autoplay=True, visible=False)
-            
-            # THANH ĐIỀU KHIỂN
-            with gr.Row(elem_classes="control-panel-fixed"):
-                with gr.Column(scale=7):
-                    msg_input = gr.MultimodalTextbox(
-                        show_label=False, placeholder="Hỏi NodeAI hoặc gửi ảnh...", container=False, elem_classes="input-box"
-                    )
-                with gr.Column(scale=3, elem_classes="mic-container"):
-                    # ĐÂY LÀ MIC CỦA BẠN: Bấm vào đây để nói!
-                    mic_btn = gr.Audio(sources=["microphone"], type="filepath", label="Ghi âm", show_label=True)
-
-    # Xử lý sự kiện
+        # FIXED INPUT AREA (Chống nhảy layout)
+        with gr.Row(elem_classes="input-area-fixed"):
+            with gr.Column(elem_classes="input-container", scale=10):
+                msg_input = gr.MultimodalTextbox(
+                    show_label=False, 
+                    placeholder="Hỏi Phúc bất cứ điều gì...", 
+                    container=False
+                )
+            with gr.Column(elem_classes="mic-box", scale=1):
+                # Nút Mic thu nhỏ gọn gàng ở bên cạnh
+                mic_btn = gr.Audio(sources=["microphone"], type="filepath", container=False)
+
+    # Trình xử lý sự kiện
     msg_input.submit(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)
     
-    def process_mic(path): return {"text": "Dịch và trả lời âm thanh này", "files": [path]}
+    def process_mic(path): return {"text": "Dịch âm thanh này", "files": [path]}
     mic_btn.stop_recording(process_mic, mic_btn, msg_input)
     
     new_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])