Update app.py

app.py

@@ -387,6 +387,18 @@ def rag_workflow(query):
     return response.content
 
 
+def get_chroma_vectorstore2(embedding_model):
+    # Define the persist_directory path
+    vectorstore_path = "/home/user/data"
+    
+    # Ensure the directory exists
+    os.makedirs(vectorstore_path, exist_ok=True)  # Creates it if it doesn't exist
+    print(f"Using persist_directory: {vectorstore_path}")
+    
+    # Initialize the Chroma vectorstore with the specified persist_directory
+    vectorstore = Chroma(persist_directory=vectorstore_path, embedding_function=embedding_model)
+    return vectorstore
+
 def initialize():
     global vector_store, chunks, llm
 
@@ -431,18 +443,6 @@ def initialize():
 initialize()
 
 
-def get_chroma_vectorstore2(embedding_model):
-    # Define the persist_directory path
-    vectorstore_path = "/home/user/data"
-    
-    # Ensure the directory exists
-    os.makedirs(vectorstore_path, exist_ok=True)  # Creates it if it doesn't exist
-    print(f"Using persist_directory: {vectorstore_path}")
-    
-    # Initialize the Chroma vectorstore with the specified persist_directory
-    vectorstore = Chroma(persist_directory=vectorstore_path, embedding_function=embedding_model)
-    return vectorstore
-
 # Example usage
 # embedding_model should be your embedding function or model instance
 # vectorstore = get_chroma_vectorstore(embedding_model)