Update app.py

app.py

@@ -6,11 +6,12 @@ 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
+# Biến toàn cục lưu lịch sử hội thoại cho model và hiển thị
 chat_history_ids = None
+history = []  # lưu [("User: ...", "Bot: ..."), ...]
 
 def chatbot(msg):
-    global chat_history_ids
+    global chat_history_ids, history
     
     # Encode câu hỏi mới
     new_input_ids = tokenizer.encode(msg + tokenizer.eos_token, return_tensors='pt')
@@ -24,19 +25,25 @@ 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,  # tăng để giữ nhiều context
+        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)
-    return f"Dr. Mom AI: {reply}"
+    
+    # Lưu vào lịch sử
+    history.append(("Bạn: " + msg, "Dr. Mom AI: " + reply))
+    
+    # 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
 
 # Giao diện Gradio
 demo = gr.Interface(
     fn=chatbot,
     inputs=gr.Textbox(label="Bạn hỏi gì nè?"),
-    outputs=gr.Textbox(label="Dr. Mom AI trả lời"),
+    outputs=gr.Textbox(label="Lịch sử hội thoại", lines=15),
     title="Dr. Mom AI",
     theme="default"
 )