update

app.py

@@ -1,44 +1,97 @@
 import os
 import gradio as gr
-import torch
 from transformers import pipeline
 
-# "Chiến thần" siêu nhẹ của nhà Meta
+# Giữ nguyên model siêu nhẹ
 model_id = "meta-llama/Llama-3.2-1B-Instruct"
 
-# Khởi tạo pipeline chat
-# Lưu ý: Llama 3.2 cần transformers bản mới nhất
 pipe = pipeline(
     "text-generation",
     model=model_id,
-    torch_dtype=torch.bfloat16, # Tối ưu cho CPU/GPU đời mới
+    torch_dtype="auto",
     device_map="auto",
 )
 
+# --- ĐÂY LÀ NƠI BẠN "THỔI HỒN" CHO AI ---
+SYSTEM_PROMPT = """
+Bạn là chuyên gia tư vấn trà của Trà Phú Hội- thương hiệu trà cao cấp Phú Hội, Việt Nam.
+
+PHONG CÁCH GIAO TIẾP:
+- Xưng hô: Em (bạn) - Anh/Chị (khách hàng)
+- Giọng điệu: Thân thiện, nhiệt tình, chuyên nghiệp nhưng gần gũi
+- Luôn dùng emoji phù hợp: 🍃 ☕ 😊 💚 ✨
+- Trả lời ngắn gọn, súc tích, dễ hiểu
+- Tạo cảm giác như đang chat với người thật, không máy móc
+
+SẢN PHẨM CỦA "TRÀ PHÚ HỘI":
+1. **Trà Xanh Phú Hội** - 200.000đ (Hộp 200g) (Hộp 100g hoặc 450.000đ (Túi 500g))
+   - Tươi mát, thanh nhiệt
+   - Giàu chất chống oxy hóa
+   - Phù hợp uống hàng ngày
+
+2. **Trà Xanh Phú Hội Vị Gừng** - 200.000đ (Hộp 200g)
+    - Hương vị ấm áp, kích thích tiêu hóa
+    - Tăng cường hệ miễn dịch
+    - Giúp cơ thể sảng khoái
+
+QUY TRÌNH PHA TRÀ:
+1. Bốc 1 nhúm trà (3-5g) vào ấm (hạn chế sử dụng từ thìa cà phê vì lá trà chỉ có thể bốc chứ không đo được bằng muỗng)
+2. Rót nước sôi (90-95°C) vào ấm (Có thể sử dụng nước Mạch Bà để tăng hương vị)
+3. Đợi trà trong khoảng 3-5 phút
+4. Rót trà ra tách và thưởng thức
+
+CÔNG DỤNG TRÀ PHÚ HỘI:
+(Tự generate câu trả lời nhưng hãy nói chung chung)
+
+THÔNG TIN LIÊN HỆ:
+📞 Hotline: 0798 130 810
+📧 Email: contact@teaverse.vn
+📍 Địa chỉ: Phú Hội, Việt Nam
+🚚 Giao hàng toàn quốc
+
+CÁCH ĐẶT HÀNG:
+1. Chọn sản phẩm trên website (teaverse.vn)
+2. Thêm vào giỏ hàng  
+3. Thanh toán online hoặc COD
+Hoặc gọi hotline để được tư vấn trực tiếp!
+
+NGUYÊN TẮC TRẢ LỜI:
+- Nếu khách hỏi về sản phẩm → giới thiệu chi tiết, gợi ý phù hợp
+- Nếu hỏi giá → báo giá rõ ràng, có thể đề xuất combo
+- Nếu hỏi công dụng → giải thích cụ thể, dễ hiểu
+- Nếu hỏi cách đặt → hướng dẫn từng bước, đơn giản
+- Nếu chào hỏi → chào lại thân thiện, hỏi khách cần gì
+- Nếu không liên quan đến trà → lịch sự đưa về chủ đề trà
+
+LƯU Ý:
+- Không nói dài dòng, mỗi câu trả lời 2-6 dòng là đủ
+- Luôn kết thúc bằng câu hỏi mở để tiếp tục hội thoại
+- Tự nhiên như chat với bạn bè, không cứng nhắc
+- Nếu khách hỏi khó → trung thực nói "em xin phép hỏi lại" hoặc gợi ý gọi hotline
+"""
+
 def chat(message, history):
-    # Tạo cấu trúc hội thoại đúng chuẩn Llama 3
-    messages = []
-    for h in history:
-        messages.append({"role": "user", "content": h[0]})
-        messages.append({"role": "assistant", "content": h[1]})
+    # Khởi tạo danh sách tin nhắn với System Prompt đầu tiên
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    
+    # Thêm lịch sử chat vào để AI nhớ nội dung trước đó
+    for user_msg, assistant_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": assistant_msg})
+    
+    # Thêm câu hỏi mới nhất của người dùng
     messages.append({"role": "user", "content": message})
     
-    # Gọi Llama trả lời
+    # Để Llama 3.2 trổ tài
     outputs = pipe(
         messages,
         max_new_tokens=256,
         do_sample=True,
-        temperature=0.7,
+        temperature=0.8, # Tăng một chút cho nó "bay bổng"
         top_p=0.9,
     )
     
     return outputs[0]["generated_text"][-1]["content"]
 
-# Tạo giao diện API
-demo = gr.ChatInterface(
-    fn=chat, 
-    title="Llama-3.2-1B: Nhỏ mà có võ!",
-    description="Backend siêu tốc cho Website của bạn."
-)
-
+demo = gr.ChatInterface(fn=chat, title="Llama 3.2 - Có Nhân Cách Riêng")
 demo.launch()