Update app.py

app.py

@@ -115,9 +115,11 @@ def main():
         value=4  
     )
     retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=300, chunk_overlap=30)
+    rag_chain = create_conversational_rag_chain(retriever)
+
     
     if user_question := st.text_input("Ask a question about your documents:"):
-        handle_userinput(user_question,retriever)
+        handle_userinput(user_question,retriever,rag_chain)
         
     
     
@@ -130,7 +132,7 @@ def main():
     
 
 
-def handle_userinput(user_question,retriever):
+def handle_userinput(user_question,retriever,rag_chain):
     st.session_state.messages.append({"role": "user", "content": user_question})
     st.chat_message("user").write(user_question)
     docs = retriever.invoke(user_question)
@@ -143,7 +145,7 @@ def handle_userinput(user_question,retriever):
     
     doc_txt = [doc.page_content for doc in docs]
     
-    rag_chain = create_conversational_rag_chain(retriever)
+    
     response = rag_chain.invoke({"context": doc_txt, "question": user_question})
     st.session_state.messages.append({"role": "assistant", "content": response})
     st.chat_message("assistant").write(response)