Update app3.py

app3.py

@@ -7,6 +7,7 @@ from langchain_community.vectorstores import Chroma
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
+from langchain.chains import RetrievalQA 
 
 st.set_page_config(page_title="Document Genie", layout="wide")
 
@@ -56,7 +57,8 @@ def get_pdf(pdf_docs,query):
     chunks=text_splitter.split_text(text)
     embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
     vector = Chroma.from_documents(chunk, embeddings)
-    docs = vector.similarity_search(query)
+    #docs = vector.similarity_search(query)
+    docs = vector_store.as_retriever(search_type='similarity', search_kwargs={'k': 3}) 
     chain = get_conversational_chain()
     response = chain({"input_documents": docs, "question": query}, return_only_outputs=True)
     return response