Update app.py

app.py

@@ -418,8 +418,36 @@ def initialize():
     llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
 
 
+
+    # Path to the vectorstore directory inside /home/user/app
+    vectorstore_path = os.path.join(os.getcwd(), "vectorstore")
+    print(f"Vectorstore Path: {vectorstore_path}")
+    
+    # Ensure the directory exists
+    os.makedirs(vectorstore_path, exist_ok=True)
+    
+    vectorstore = get_chroma_vectorstore2(EMBEDDING_MODEL_NAME)
+
 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)
+
+
 # Gradio utils
 def check_input_text(text):
     if not text: