Update generator.py

generator.py

@@ -37,22 +37,22 @@ def answer_query(query, model="Gemini Pro", temperature=0.2):
     docs = filter_relevant_docs(all_docs, query)
     context = "\n\n".join([doc.page_content for doc in docs])
 
-    prompt = f"""Dựa trên tài liệu sau, hãy trả lời câu hỏi theo phong cách trang trọng, lịch sự và chuyên nghiệp:
-
+    prompt = f"""
+Dựa trên tài liệu sau, hãy trả lời câu hỏi theo phong cách trang trọng, lịch sự và chuyên nghiệp:
 {context}
-
 Câu hỏi: {query}
 
 Yêu cầu:
-- Sử dụng từ ngữ lịch sự ("Bạn cần...", "Vui lòng...", "Sau khi...")
-- Không sử dụng từ nói miệng như "nhé", "nha", "ok".
-- Câu trúc câu đầy đủ, rõ ràng.
-- Chỉ trả lời dựa trên thông tin trong tài liệu. Nếu không tìm thấy thông tin liên quan, trả lời: "Thông tin không có trong tài liệu được cung cấp."
-- Không tự thêm "Nguồn tham khảo" trong phần trả lời.
-- Không thêm "Dựa vào tài liệu đã cung cấp,..."
-- Hãy trả lời câu hỏi của người dùng như con Chatbot thông thường
+- Luôn sử dụng từ ngữ lịch sự ("Bạn cần...", "Vui lòng...", "Sau khi...").
+- Tránh dùng từ nói miệng như "nhé", "nha", "ok".
+- Câu trả lời cần đầy đủ, rõ ràng, không mơ hồ.
+- Chỉ sử dụng thông tin có trong tài liệu. Nếu không có thông tin liên quan, trả lời: "Thông tin không có trong tài liệu được cung cấp."
+- Không được nhắc đến việc đã tham khảo hay nguồn tài liệu.
+- Nếu người dùng yêu cầu, vui lòng cung cấp câu trả lời bằng ngôn ngữ khác hoặc viết code chính xác, đầy đủ theo yêu cầu.
+- Trả lời như một chatbot thông thường, không cần giải thích về quy trình hay tài liệu.
 
-Trả lời:"""
+Trả lời:
+"""
 
     url = f"https://generativelanguage.googleapis.com/v1/models/gemini-2.0-flash:generateContent?key={API_KEY}"
     headers = {"Content-Type": "application/json"}