Update app.py

app.py

@@ -4,16 +4,16 @@ import os
 import uuid
 import json
 
-# Thư viện hỗ trợ
+# Import an toàn
 try: from duckduckgo_search import DDGS
 except: DDGS = None
 try: from gtts import gTTS
 except: gTTS = None
 
-# 1. CẤU HÌNH & MODEL
+# 1. CẤU HÌNH
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
-DB_FILE = "nodeai_genesis_final.json"
+DB_FILE = "nodeai_platinum_v31.json"
 
 def load_db():
     if os.path.exists(DB_FILE):
@@ -26,16 +26,17 @@ 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 Layout & Orange Theme)
+# 2. GUI DESIGN (FIX TRÀN MÀN HÌNH)
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@700&family=Plus+Jakarta+Sans:wght@400;600&display=swap');
 
 .gradio-container {
     background: #08080c !important;
     font-family: 'Plus Jakarta Sans', sans-serif !important;
+    overflow-x: hidden !important;
 }
 
-/* Sidebar */
+/* Sidebar linh hoạt */
 .sidebar-node {
     position: fixed !important;
     left: 0 !important; top: 0 !important;
@@ -51,44 +52,49 @@ custom_css = """
     background: linear-gradient(90deg, #00f2fe, #4facfe);
     -webkit-background-clip: text;
     -webkit-text-fill-color: transparent;
-    font-size: 22px; text-align: center; margin-bottom: 20px;
+    font-size: 20px; text-align: center; margin-bottom: 20px;
 }
 
-/* Nút Cam Pro */
 .orange-btn {
     background: linear-gradient(90deg, #ff7e5f, #feb47b) !important;
-    border: none !important;
-    color: white !important;
-    font-weight: bold !important;
-    margin-top: 10px !important;
+    border: none !important; color: white !important; font-weight: bold !important;
 }
 
-/* Chat Area */
+/* Chat Area - Sửa lỗi tràn màn hình */
 .chat-main {
     margin-left: 260px !important;
     width: calc(100% - 260px) !important;
-    height: 100vh !important;
-    padding: 20px 40px !important;
+    min-height: 100vh !important;
+    padding: 10px 20px !important;
+    box-sizing: border-box !important;
 }
 
 .gr-chatbot { 
     background: transparent !important; 
     border: none !important; 
     height: 75vh !important;
-    margin-bottom: 100px !important;
+    margin-bottom: 120px !important;
 }
 
-/* Thanh nhập liệu cố định */
+/* Thanh nhập liệu - Tự động co giãn theo màn hình */
 .input-fixed-bottom {
     position: fixed !important;
-    bottom: 25px !important;
-    left: 285px !important;
-    right: 25px !important;
+    bottom: 20px !important;
+    left: 280px !important; /* Cách sidebar */
+    right: 20px !important; /* Cách lề phải */
     background: #16161e !important;
-    border: 1px solid rgba(0, 242, 254, 0.2) !important;
-    border-radius: 16px !important;
+    border: 1px solid rgba(0, 242, 254, 0.3) !important;
+    border-radius: 15px !important;
     padding: 5px 15px !important;
     z-index: 2000 !important;
+    box-sizing: border-box !important;
+}
+
+/* Responsive cho điện thoại */
+@media screen and (max-width: 768px) {
+    .sidebar-node { display: none !important; }
+    .chat-main { margin-left: 0 !important; width: 100% !important; }
+    .input-fixed-bottom { left: 15px !important; right: 15px !important; }
 }
 
 footer { display: none !important; }
@@ -97,34 +103,36 @@ footer { display: none !important; }
 # 3. CORE LOGIC
 def chat_fn(message, history, session_id, store):
     user_text = message.get("text", "")
-    files = message.get("files", [])
-    
-    if not user_text and not files: return {"text": ""}, history, session_id, store, None
+    if not user_text and not message.get("files", []): 
+        yield {"text": ""}, history, session_id, store, ""
+        return
     
     if session_id not in store:
-        store[session_id] = {"title": user_text[:20] if user_text else "Task AI", "history": []}
+        store[session_id] = {"title": user_text[:20] if user_text else "New Workspace", "history": []}
     
     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 if user_text else "Gửi file"})
+    api_messages.append({"role": "user", "content": user_text})
     
     try:
         ans = ""
-        new_h = history + [{"role": "user", "content": user_text if user_text else "📎 File"}]
+        new_h = history + [{"role": "user", "content": user_text}]
         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
-                yield {"text": ""}, new_h + [{"role": "assistant", "content": ans}], session_id, store, None
+                # Chỉ gửi text về chatbot, không gửi về audio_out ở đây
+                yield {"text": ""}, new_h + [{"role": "assistant", "content": ans}], session_id, store, ""
         
         final_h = new_h + [{"role": "assistant", "content": ans}]
         store[session_id]["history"] = final_h
         save_db(store)
+        # Khi xong hẳn, gửi văn bản cuối cùng vào tts_buffer (một Textbox ẩn)
         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": str(e)}], session_id, store, ""
 
 def tts_engine(text):
-    if not text or not gTTS: return None
+    if not text or len(text) < 2 or not gTTS: return None
     try:
         tts = gTTS(text=text[:300], lang='vi')
         tts.save("voice.mp3")
@@ -135,14 +143,11 @@ def tts_engine(text):
 with gr.Blocks(title="NodeAI Genesis") as demo:
     chat_store = gr.State(load_db())
     session_id = gr.State(str(uuid.uuid4()))
+    tts_buffer = gr.Textbox(visible=False) # Dùng làm trung gian để fix lỗi audio
     
-    # Sidebar
     with gr.Column(elem_classes="sidebar-node"):
         gr.HTML("<div class='logo-text'>NODEAI GENESIS</div>")
-        
-        # ĐĂNG NHẬP CHÍNH CHỦ HUGGING FACE
-        gr.LoginButton(value="Đăng nhập HF")
-        
+        gr.LoginButton(value="Đăng nhập")
         new_btn = gr.Button("＋ NEW WORKSPACE", variant="primary", elem_classes="orange-btn")
         
         with gr.Column():
@@ -152,7 +157,6 @@ with gr.Blocks(title="NodeAI Genesis") as demo:
                     btn = gr.Button(f"◈ {store[sid]['title']}", variant="ghost")
                     btn.click(lambda s=sid: (store[s]["history"], s), None, [chatbot, session_id])
 
-    # Main area
     with gr.Column(elem_classes="chat-main"):
         chatbot = gr.Chatbot(show_label=False) 
         audio_out = gr.Audio(autoplay=True, visible=False)
@@ -160,16 +164,20 @@ with gr.Blocks(title="NodeAI Genesis") as demo:
         with gr.Row(elem_classes="input-fixed-bottom"):
             msg_input = gr.MultimodalTextbox(
                 show_label=False, 
-                placeholder="Hỏi NodeAI bất cứ điều gì...", 
+                placeholder="Hỏi Master Phúc bất cứ điều gì...", 
                 container=False
             )
 
+    # LUỒNG XỬ LÝ (QUAN TRỌNG ĐỂ FIX LỖI)
     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)
+        chat_fn, 
+        [msg_input, chatbot, session_id, chat_store], 
+        [msg_input, chatbot, session_id, chat_store, tts_buffer] # Trả kết quả TTS vào buffer
+    ).then(
+        tts_engine, tts_buffer, audio_out # TTS chỉ đọc từ buffer, không đọc từ Chatbot
+    )
     
     new_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, session_id])
 
 if __name__ == "__main__":
-    # Đã xóa login_button ở đây vì đã dùng gr.LoginButton() ở trên
     demo.launch(css=custom_css)