Update app.py

app.py

@@ -84,11 +84,24 @@ all_splits = text_splitter.split_documents(docs)
 # -------------------------------
 
 # Create a Chroma vector store using the document splits
-vectorstore = Chroma.from_documents(
-    documents=all_splits, 
-    embedding=OpenAIEmbeddings(), 
-    persist_directory="./chroma_db"
-)
+persist_directory = "./chroma_db"
+
+# Check if the persist directory exists and contains data.
+if os.path.exists(persist_directory) and os.listdir(persist_directory):
+    # Load the persisted vector store
+    vectorstore = Chroma(
+        persist_directory=persist_directory,
+        embedding_function=OpenAIEmbeddings()
+    )
+    print("Loaded vector store from persist directory.")
+else:
+    # Create a new vector store from your document splits and persist it.
+    vectorstore = Chroma.from_documents(
+        documents=all_splits, 
+        embedding=OpenAIEmbeddings(), 
+        persist_directory=persist_directory
+    )
+    print("Created new vector store and persisted embeddings.")
 
 # Create a BM25 retriever from the document splits
 bm25_retriever = BM25Retriever.from_documents(all_splits)