bug fix

app.py

@@ -15,6 +15,9 @@ from s3bucket import upload_to_s3
 
 vector_database_name = "Adina_Vector_Database"
 temp_pdf_folder = "temp-pdf-files"
+vector_database_path = (
+    f"{os.environ.get('VECTOR_DATABASE_PATH', '.')}/{vector_database_name}"
+)
 
 RETRIEVER = None
 
@@ -61,14 +64,14 @@ def initialize_vector_db():
     vector_database = FAISS.from_texts(
         ["Adina Cosmetic Ingredients"], OpenAIEmbeddings()
     )
-    vector_database.save_local(f"/data/{vector_database_name}")
+    vector_database.save_local(vector_database_path)
     return vector_database
 
 
 def load_vector_db():
-    if os.path.exists(f"{vector_database_name}"):
+    if os.path.exists(vector_database_path):
         return FAISS.load_local(
-            f"{vector_database_name}",
+            vector_database_path,
             OpenAIEmbeddings(),
             allow_dangerous_deserialization=True,
         )
@@ -80,7 +83,7 @@ def append_to_vector_db(docs: list = []):
     existing_vector_db = load_vector_db()
     new_vector_db = FAISS.from_documents(docs, OpenAIEmbeddings())
     existing_vector_db.merge_from(new_vector_db)
-    existing_vector_db.save_local(f"{vector_database_name}")
+    existing_vector_db.save_local(vector_database_path)
     RETRIEVER = existing_vector_db.as_retriever()