W4 D1 HW

app.py

@@ -51,9 +51,9 @@ hf_embeddings = HuggingFaceEndpointEmbeddings(
     huggingfacehub_api_token=HF_TOKEN,
 )
 
-if os.path.exists("./data/vectorstore"):
+if os.path.exists(VECTORSTORE_PATH):
     vectorstore = FAISS.load_local(
-        "./data/vectorstore", 
+        VECTORSTORE_PATH, 
         hf_embeddings, 
         allow_dangerous_deserialization=True # this is necessary to load the vectorstore from disk as it's stored as a `.pkl` file.
     )
@@ -61,13 +61,13 @@ if os.path.exists("./data/vectorstore"):
     print("Loaded Vectorstore")
 else:
     print("Indexing Files")
-    os.makedirs("./data/vectorstore", exist_ok=True)
+    os.makedirs(VECTORSTORE_PATH, exist_ok=True)
     for i in range(0, len(split_documents), 32):
         if i == 0:
             vectorstore = FAISS.from_documents(split_documents[i:i+32], hf_embeddings)
             continue
         vectorstore.add_documents(split_documents[i:i+32])
-    vectorstore.save_local("./data/vectorstore")
+    vectorstore.save_local(VECTORSTORE_PATH)
 
 hf_retriever = vectorstore.as_retriever()