Update app.py

app.py

@@ -53,7 +53,7 @@ def get_vectorstore(text_chunks):
     with st.spinner("Creating vectorstore..."):
         metadatas = [{"source": f"{i}-pl"} for i in range(len(text_chunks))]
         embeddings = OpenAIEmbeddings()
-        vectorstore = Chroma.from_texts(texts=text_chunks, embedding=embeddings, persist_directory="./chroma_db", metadatas=metadatas)
+        vectorstore = Chroma.from_texts(texts=text_chunks, embedding=embeddings, persist_directory="./data/vectorstore", metadatas=metadatas)
         return vectorstore
 
 def get_conversation_chain(vectorstore):
@@ -99,10 +99,10 @@ def main():
         st.session_state.chat_history = None
     
     if st.session_state.conversation is None:
-        if os.path.isdir("./chroma_db"):
-            if os.listdir("./chroma_db"):
+        if os.path.isdir("./data/vectorstore"):
+            if os.listdir("./data/vectorstore"):
                 with st.spinner("Loading vector store..."):
-                    vectorstore = Chroma(persist_directory="./chroma_db", embedding_function=OpenAIEmbeddings())
+                    vectorstore = Chroma(persist_directory="./data/vectorstore", embedding_function=OpenAIEmbeddings())
                     st.session_state.conversation = get_conversation_chain(vectorstore)
             else:
                 retrain_model()