Update app.py

app.py

@@ -108,14 +108,14 @@ if query and st.session_state.documents_loaded:
         llm = GoogleGenerativeAI(model='gemini-1.0-pro', google_api_key="AIzaSyC1AvHnvobbycU8XSCXh-gRq3DUfG0EP98")
 
         # Create a PromptTemplate for the QA chain
-        qa_prompt = PromptTemplate(template="Answer the following question based on the context provided:\n\n{context}\n\nQuestion: {question}\nAnswer:", input_variables=["context", "question"])
+        qa_prompt = PromptTemplate(template="Answer the following question based on the context provided:\n\nContext: {context}\n\nQuestion: {question}\n\nAnswer:", input_variables=["context", "question"])
 
         # Create the retrieval QA chain
         qa_chain = RetrievalQA.from_chain_type(
             retriever=st.session_state.vector_store.as_retriever(),
             chain_type="map_reduce",
             llm=llm,
-            prompt=qa_prompt
+            chain_type_kwargs={"prompt": qa_prompt}
         )
 
         response = qa_chain({"question": query})