Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN")
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
-# 2. GIAO DIỆN CSS (Sẽ dùng trong .launch)
+# 2. GIAO DIỆN CSS
 custom_css = """
 footer, .built-with, header, #header { display: none !important; }
 .gradio-container { max-width: 100% !important; padding: 0 !important; }
@@ -31,37 +31,37 @@ js_cleaner = """
 " style="display:none;">
 """
 
-# 3. CHAT ENGINE (Cấu trúc Memory dạng Tuple cho bản cũ)
+# 3. CHAT ENGINE (Sử dụng định dạng {'role': '...', 'content': '...'})
 def chat_engine(message, history, profile: gr.OAuthProfile | None):
     text = message.get("text", "").strip()
     if not text: 
-        yield "", history
+        yield gr.update(value={"text": "", "files": []}), history
         return
 
     user_name = profile.name if profile else "Master"
-    # Chuyển đổi history từ Tuple sang mảng hội thoại cho AI
-    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {user_name}."}]
     
-    for user_msg, bot_msg in history:
-        if user_msg: messages.append({"role": "user", "content": user_msg})
-        if bot_msg: messages.append({"role": "assistant", "content": bot_msg})
+    # Tạo danh sách tin nhắn gửi đi cho AI (Bao gồm System Prompt)
+    messages = [{"role": "system", "content": f"Bạn là NodeAI Genesis. Master: {user_name}."}]
     
+    # history bây giờ là danh sách các dict, chỉ cần cộng vào là xong
+    messages.extend(history)
     messages.append({"role": "user", "content": text})
     
-    # Chuẩn bị history mới
-    new_history = history + [[text, ""]]
+    # Cập nhật history cho chatbot hiển thị
+    new_history = history + [{"role": "user", "content": text}]
     ans = ""
     
+    # Gửi trạng thái đang chờ
+    yield gr.update(value={"text": "", "files": []}), new_history + [{"role": "assistant", "content": "..."}]
+    
     try:
         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
-                # Cập nhật tin nhắn cuối cùng trong mảng history
-                new_history[-1][1] = ans
-                yield "", new_history
+                # Gửi history kèm tin nhắn assistant đang được cập nhật
+                yield gr.update(), new_history + [{"role": "assistant", "content": ans}]
     except Exception as e:
-        new_history[-1][1] = f"Lỗi: {str(e)}"
-        yield "", new_history
+        yield gr.update(), new_history + [{"role": "assistant", "content": f"Lỗi: {str(e)}"}]
 
 # 4. GIAO DIỆN UI
 with gr.Blocks() as demo:
@@ -73,23 +73,23 @@ with gr.Blocks() as demo:
             gr.LoginButton(value="Đăng nhập", size="sm")
             new_chat_btn = gr.Button("＋ CLEAR MEMORY", variant="stop")
             gr.Markdown("---")
-            gr.Markdown("💡 **Script Memory**: Đang chạy bản ổn định.")
+            gr.Markdown("💡 **Status**: Script Memory (Messages Format)")
 
         with gr.Column(scale=4):
-            # Xóa tham số type="messages" để tránh lỗi
+            # Không đặt type="messages" nữa vì bản này tự hiểu khi nhận list dict
             chatbot = gr.Chatbot(show_label=False, height="78vh")
             
             gr.HTML("<div class='custom-warning'>NodeAI không thể chắc chắn 100%, hãy rà soát lại khi bạn có chuyện gì quan trọng!</div>")
             
             msg_input = gr.MultimodalTextbox(
                 show_label=False, 
-                placeholder="Hỏi NodeAI...", 
+                placeholder="Nhập nội dung Master cần hỏi...", 
                 container=False
             )
 
     # Xử lý sự kiện
     msg_input.submit(chat_engine, [msg_input, chatbot], [msg_input, chatbot])
-    new_chat_btn.click(lambda: None, None, chatbot)
+    new_chat_btn.click(lambda: [], None, chatbot) # Trả về mảng rỗng để xóa chat
 
 # 5. LAUNCH
 if __name__ == "__main__":