Update app.py

app.py

@@ -56,8 +56,16 @@ def load_pdfs():
 
             # Create embeddings and vector store for the current tenant
             embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+
+            # Directory to store the vector database
+            db_directory = "path/to/store/chroma_db"  # Update with your desired path
+
             if st.session_state.tenant not in st.session_state.vectordb:
-                st.session_state.vectordb[st.session_state.tenant] = Chroma.from_documents(documents=splits, embedding=embeddings)
+                st.session_state.vectordb[st.session_state.tenant] = Chroma.from_documents(
+                    documents=splits,
+                    embedding=embeddings,
+                    persist_directory=db_directory
+                )
             else:
                 # Update the existing vector store for the tenant
                 st.session_state.vectordb[st.session_state.tenant].add_documents(splits)