Update app.py

app.py

@@ -3,97 +3,79 @@ from huggingface_hub import InferenceClient
 import os
 import uuid
 
-# 1. Cấu hình Model
+# 1. Khởi tạo
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
 client = InferenceClient(model="Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
 custom_css = """
-body, .gradio-container { background-color: #131314 !important; color: #e3e3e3 !important; }
-.sidebar { background-color: #1e1f20 !important; border-right: 1px solid #333 !important; height: 100vh; padding: 15px !important; }
-.gemini-title { color: #8ab4f8; font-size: 24px; font-weight: bold; margin-bottom: 20px; }
-.nav-btn { text-align: left !important; background: #333 !important; color: white !important; margin-bottom: 10px; border-radius: 20px !important; }
-.history-btn { 
-    width: 100%; text-align: left !important; background: transparent !important; 
-    border: none !important; color: #ccc !important; cursor: pointer; padding: 5px 10px !important;
-}
-.history-btn:hover { background: #2d2e30 !important; color: white !important; }
-.footer-tag { font-size: 0.75rem; color: #9aa0a6; text-align: center; margin-top: 15px; }
+body, .gradio-container { background-color: #131314 !important; color: #eee !important; }
+.sidebar { background-color: #1e1f20 !important; border-right: 1px solid #333 !important; padding: 15px !important; }
+.nav-btn { background: #333 !important; color: white !important; border-radius: 20px !important; margin-bottom: 15px !important; }
+.history-btn { text-align: left !important; background: transparent !important; border: none !important; color: #ccc !important; cursor: pointer; }
+.history-btn:hover { background: #2d2e30 !important; }
 """
 
-def chat_fn(message, history, session_id, chat_store):
-    if not message: return "", history, session_id, chat_store
+def chat_fn(message, history, current_id, chat_store):
+    if not message: return "", history, current_id, chat_store
     user_text = message if isinstance(message, str) else message.get("text", "")
 
-    # Gọi API
-    messages = [{"role": "system", "content": "Bạn là NodeAI Pro phong cách Gemini."}]
+    # Cập nhật Store nếu là lần đầu chat trong phiên này
+    if not history:
+        chat_store[current_id] = {"title": user_text[:20] + "...", "data": []}
+
+    messages = [{"role": "user", "content": user_text}]
     for turn in history:
-        content = turn["content"][0]["text"] if isinstance(turn["content"], list) else turn["content"]
-        messages.append({"role": turn["role"], "content": content})
-    messages.append({"role": "user", "content": user_text})
+        messages.insert(-1, {"role": turn["role"], "content": turn["content"][0]["text"]})
 
     try:
-        response_text = ""
+        res = ""
         for token in client.chat_completion(messages=messages, max_tokens=1024, stream=True):
             delta = token.choices[0].delta.content
             if delta:
-                response_text += delta
-                current_history = history + [
+                res += delta
+                new_history = history + [
                     {"role": "user", "content": [{"type": "text", "text": user_text}]},
-                    {"role": "assistant", "content": [{"type": "text", "text": response_text}]}
+                    {"role": "assistant", "content": [{"type": "text", "text": res}]}
                 ]
-                yield "", current_history, session_id, chat_store
-        
-        # LƯU TOÀN BỘ VÀO STORE
-        chat_store[session_id] = {"title": user_text[:20] + "...", "data": current_history}
+                # Lưu dữ liệu vào bộ nhớ
+                chat_store[current_id]["data"] = new_history
+                yield "", new_history, current_id, chat_store
     except Exception as e:
-        yield "", history + [{"role": "assistant", "content": [{"type": "text", "text": str(e)}]}], session_id, chat_store
-
-def load_old_chat(session_id_to_load, chat_store):
-    # Lấy dữ liệu từ store dựa trên ID đã chọn
-    old_data = chat_store.get(session_id_to_load, {}).get("data", [])
-    return old_data, session_id_to_load
+        yield "", history, current_id, chat_store
 
-def start_new_chat():
-    return [], str(uuid.uuid4())
+# Hàm để khi bấm vào nút cũ thì hiện lại chat
+def load_chat(sid, store):
+    return store[sid]["data"], sid
 
 with gr.Blocks(title="NodeAI Gemini") as demo:
     chat_store = gr.State({}) 
-    session_id = gr.State(str(uuid.uuid4()))
+    current_id = gr.State(str(uuid.uuid4()))
 
     with gr.Row():
         # SIDEBAR
         with gr.Column(scale=1, elem_classes="sidebar"):
-            gr.Markdown("Gemini", elem_classes="gemini-title")
-            new_btn = gr.Button("+ Cuộc trò chuyện mới", elem_classes="nav-btn")
+            gr.Markdown("## Gemini")
+            new_btn = gr.Button("+ Trò chuyện mới", elem_classes="nav-btn")
             gr.Markdown("---")
-            gr.Markdown("### Gần đây")
             
-            # Danh sách các nút lịch sử
-            history_container = gr.Column()
-
+            # ��ÂY LÀ CHỖ TẠO NÚT
             @gr.render(inputs=chat_store)
-            def render_history(store):
-                for sid, info in store.items():
-                    # Tạo một nút cho mỗi cuộc hội thoại cũ
-                    btn = gr.Button(f"🕒 {info['title']}", elem_classes="history-btn")
-                    # Khi nhấn vào nút này, load lại chat đó
-                    btn.click(load_old_chat, [gr.State(sid), chat_store], [chatbot, session_id])
-
-            gr.Markdown("<div style='height: 30vh;'></div>")
-            gr.LoginButton(size="sm")
+            def show_history(store):
+                for sid in store.keys():
+                    btn = gr.Button(f"💬 {store[sid]['title']}", elem_classes="history-btn")
+                    btn.click(load_chat, [gr.State(sid), chat_store], [chatbot, current_id])
 
-        # MAIN
+        # MAIN CHAT
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(label=None, height=600)
+            chatbot = gr.Chatbot(height=600)
             with gr.Row():
-                msg_input = gr.Textbox(placeholder="Nhập yêu cầu...", show_label=False, container=False, scale=9)
-                submit_btn = gr.Button("🚀", scale=1, variant="primary")
-            gr.Markdown("NodeAI Pro v4.5 | Powered by Qwen 2.5", elem_classes="footer-tag")
+                msg_input = gr.Textbox(placeholder="Nhập tin nhắn...", show_label=False, scale=9)
+                submit_btn = gr.Button("🚀", scale=1)
 
-    # Xử lý
-    msg_input.submit(chat_fn, [msg_input, chatbot, session_id, chat_store], [msg_input, chatbot, session_id, chat_store])
-    submit_btn.click(chat_fn, [msg_input, chatbot, session_id, chat_store], [msg_input, chatbot, session_id, chat_store])
-    new_btn.click(start_new_chat, None, [chatbot, session_id])
+    # Logic
+    msg_input.submit(chat_fn, [msg_input, chatbot, current_id, chat_store], [msg_input, chatbot, current_id, chat_store])
+    submit_btn.click(chat_fn, [msg_input, chatbot, current_id, chat_store], [msg_input, chatbot, current_id, chat_store])
+    new_btn.click(lambda: ([], str(uuid.uuid4())), None, [chatbot, current_id])
 
 if __name__ == "__main__":
     demo.launch(css=custom_css)