Update app.py

app.py

@@ -76,17 +76,17 @@ documents = text_splitter.split_documents(documents)
 
 
 def load_or_create_vector_database():
-    # Check if the vector database already exists
     if os.path.exists("./data"):
-        # If it exists, load and return it
-        vectordb = Chroma.load("./data")
+        vectordb = Chroma.from_directory("./data")
     else:
-        # If it doesn't exist, create and persist the database
         documents = []  # Load your documents here
         vectordb = Chroma.from_documents(documents, embedding=OpenAIEmbeddings(), persist_directory="./data")
         vectordb.persist()
     return vectordb
 
+
+
+
 # Define Streamlit app
 def main():
     st.title("AI Chat Application")