Update app.py

app.py

@@ -6,17 +6,16 @@ import torch
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-small")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-small")
 
-# Biến toàn cục lưu lịch sử hội thoại cho model và hiển thị
+# Biến toàn cục để lưu lịch sử cho mô hình
 chat_history_ids = None
-history = []  # lưu [("User: ...", "Bot: ..."), ...]
 
-def chatbot(msg):
-    global chat_history_ids, history
+def chatbot(msg, history):
+    global chat_history_ids
     
     # Encode câu hỏi mới
     new_input_ids = tokenizer.encode(msg + tokenizer.eos_token, return_tensors='pt')
     
-    # Nếu chưa có lịch sử, khởi tạo; nếu có thì nối thêm
+    # Nếu chưa có lịch sử thì khởi tạo; nếu có thì nối thêm
     if chat_history_ids is None:
         bot_input_ids = new_input_ids
     else:
@@ -24,28 +23,30 @@ def chatbot(msg):
     
     # Sinh câu trả lời dựa trên toàn bộ lịch sử
     chat_history_ids = model.generate(
-        bot_input_ids, 
-        max_length=1000,  
+        bot_input_ids,
+        max_length=1000,
         pad_token_id=tokenizer.eos_token_id
     )
     
     # Decode câu trả lời (chỉ lấy phần mới sinh ra)
-    reply = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
-    
-    # Lưu vào lịch sử
-    history.append(("Bạn: " + msg, "Dr. Mom AI: " + reply))
+    reply = tokenizer.decode(
+        chat_history_ids[:, bot_input_ids.shape[-1]:][0],
+        skip_special_tokens=True
+    )
     
-    # Hiển thị toàn bộ lịch sử hội thoại
-    chat_log = "\n".join([f"{u}\n{b}" for u, b in history])
-    return chat_log
+    # Lưu vào history (Gradio dùng list of tuples: (user, bot))
+    history.append((msg, reply))
+    return history, history
 
 # Giao diện Gradio
-demo = gr.Interface(
-    fn=chatbot,
-    inputs=gr.Textbox(label="Bạn hỏi gì nè?"),
-    outputs=gr.Textbox(label="Lịch sử hội thoại", lines=15),
-    title="Dr. Mom AI",
-    theme="default"
-)
+with gr.Blocks(title="Dr. Mom AI") as demo:
+    gr.Markdown("## 🤖 Dr. Mom AI Chatbot")
+    
+    chatbot_ui = gr.Chatbot(label="Lịch sử hội thoại")
+    msg = gr.Textbox(label="Bạn hỏi gì nè?", placeholder="Nhập tin nhắn rồi nhấn Enter...")
+    clear = gr.Button("🔄 Reset hội thoại")
+    
+    msg.submit(chatbot, [msg, chatbot_ui], [chatbot_ui, chatbot_ui])
+    clear.click(lambda: None, None, chatbot_ui, queue=False)
 
 demo.launch()