Update app.py

app.py

@@ -107,13 +107,14 @@ if query and st.session_state.documents_loaded:
         # Initialize Google Generative AI
         llm = GoogleGenerativeAI(model='gemini-1.0-pro', google_api_key="AIzaSyC1AvHnvobbycU8XSCXh-gRq3DUfG0EP98")
 
-        # Create a PromptTemplate for the QA chain
+
+        #Create a PromptTemplate for the QA chain
         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",
+            chain_type="stuff",
             llm=llm,
             chain_type_kwargs={"prompt": qa_prompt}
         )