Update generator.py

generator.py

@@ -29,7 +29,12 @@ def format_sources(docs):
             sources.add(filename)
     return "\n".join(f"- {src}" for src in sorted(sources))
 
-def answer_query(query, model="Gemini Pro", temperature=0.2):
+def answer_query(query, model="Gemini Pro", temperature=0.2, history=None):
+    # Khởi tạo lịch sử nếu None
+    if history is None:
+        history = []
+
+    # Lấy và lọc tài liệu
     all_docs = retrieve_docs(query)
     if not all_docs:
         return "Không tìm thấy tài liệu liên quan để trả lời.", []
@@ -37,23 +42,31 @@ 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])
 
+    # Định dạng lịch sử hội thoại
+    history_text = ""
+    if history:
+        history_text = "Lịch sử hội thoại:\n"
+        for exchange in history:
+            history_text += f"Người dùng: {exchange['user']}\nTrợ lý: {exchange['assistant']}\n\n"
+
+    # Tạo prompt với lịch sử hội thoại
     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:
+{history_text}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:
 - 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ồ.
+- Đọc kĩ các tài liệu để đưa ra câu trả lời liên quan và chính xác nhất
 - 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:
 """
 
+    # Gọi API Gemini
     url = f"https://generativelanguage.googleapis.com/v1/models/gemini-2.0-flash:generateContent?key={API_KEY}"
     headers = {"Content-Type": "application/json"}
     payload = {
@@ -64,10 +77,11 @@ Trả lời:
     response = requests.post(url, headers=headers, json=payload)
     data = response.json()
 
+    # Xử lý phản hồi từ API
     try:
         answer = data['candidates'][0]['content']['parts'][0]['text']
     except Exception as e:
-        print("🔴 Response từ Gemini:", data)
+        print("🔴 Phản hồi từ Gemini:", data)
         answer = "Lỗi khi gọi Gemini API: " + str(e)
 
-    return answer, docs
+    return answer, docs