Update app.py

app.py

@@ -37,7 +37,7 @@ def method_get_vectorstore(document_chunks):
     vector_store = Chroma.from_documents(document_chunks, embeddings)
     return vector_store
 
-def get_context_retriever_chain(vector_store):
+def get_context_retriever_chain(vector_store,question):
     # Initialize the retriever
     retriever = vector_store.as_retriever()
     
@@ -61,7 +61,7 @@ def get_context_retriever_chain(vector_store):
         | StrOutputParser()
     )
     
-    return after_rag_chain
+    return after_rag_chain.invoke(question)
 
 def main():
     st.set_page_config(page_title="Chat with websites", page_icon="🤖")
@@ -88,10 +88,8 @@ def main():
                 doc_splits = method_get_text_chunks(raw_text)
                 # create vector store
                 vector_store = method_get_vectorstore(doc_splits)
-                # st.write(doc_splits)
-                retriever_chain = get_context_retriever_chain(vector_store)
                 # Generate response using the RAG pipeline
-                answer = retriever_chain.invoke(question)
+                answer = get_context_retriever_chain(vector_store,question)
                 # Display the generated answer
                 st.text_area("Answer", value=answer, height=300, disabled=True)