Update app.py

app.py

@@ -73,16 +73,16 @@ def postprocess_answer(answer):
     
     return answer
 
-# Sửa lại prompt template với context và query
+# Sửa lại prompt template sử dụng 'question' thay vì 'query'
 template = """Hãy trả lời câu hỏi dựa trên nội dung sau:
 {context}
 
-Câu hỏi: {query}
+Câu hỏi: {question}
 Trả lời bằng tiếng Việt một cách tự nhiên:"""
 
 QA_PROMPT = PromptTemplate(
     template=template,
-    input_variables=["context", "query"]  # Thêm context vào input variables
+    input_variables=["context", "question"]  # Đổi thành 'question'
 )
 
 def main():
@@ -101,15 +101,16 @@ def main():
         
         llm = load_llm()
         
-        # Sửa lại cách khởi tạo RetrievalQA
+        # Thêm tham số input_key để ánh xạ đúng tên biến
         qa_chain = RetrievalQA.from_chain_type(
             llm=llm,
             chain_type="stuff",
             retriever=vectorstore.as_retriever(search_kwargs={"k": 4}),
             return_source_documents=True,
+            input_key="question",  # Thêm ánh xạ input key
             chain_type_kwargs={
                 "prompt": QA_PROMPT,
-                "document_variable_name": "context"  # Chỉ định tên biến chứa documents
+                "document_variable_name": "context"
             }
         )
         
@@ -117,7 +118,8 @@ def main():
         if query:
             with st.spinner("Đang tổng hợp câu trả lời..."):
                 try:
-                    result = qa_chain({"query": query})
+                    # Truyền input dưới dạng dictionary với key đúng
+                    result = qa_chain({"question": query})  # Đổi thành 'question'
                     raw_answer = result["result"]
                     answer = postprocess_answer(raw_answer)