Update app.py

app.py

@@ -69,7 +69,7 @@ def text_splitter(text):
 #COHERE_API_KEY = os.getenv("COHERE_API_KEY")
 #HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
 
-def get_conversational_chain():
+def get_conversational_chain(retriever):
     prompt_template = """
     Given the following extracted parts of a long document and a question, create a final answer.
     Answer the question as detailed as possible from the provided context, make sure to provide all the details, if the answer is not in
@@ -94,7 +94,7 @@ def get_conversational_chain():
 
     pt = ChatPromptTemplate.from_template(prompt_template)
     # Retrieve and generate using the relevant snippets of the blog.
-    retriever = db.as_retriever()
+    #retriever = db.as_retriever()
     rag_chain = (
     {"context": retriever, "question": RunnablePassthrough()}
     | pt
@@ -110,7 +110,9 @@ def embedding(chunk,query):
     db = Chroma.from_documents(chunk,embeddings)
     doc = db.similarity_search(query)
     print(doc)
-    chain = get_conversational_chain()
+    #Retrieve and generate using the relevant snippets of the blog.
+    retriever = doc.as_retriever()
+    chain = get_conversational_chain(retriever)
     response = chain.invoke(query)
     print(response)
     return response