diff --git a/src/Gift_Recommendation_Bot.py b/src/Gift_Recommendation_Bot.py index ff0d1cd..2add55a 100644 --- a/src/Gift_Recommendation_Bot.py +++ b/src/Gift_Recommendation_Bot.py @@ -131,14 +131,13 @@ def download_file (path,url): model=EMBEDDINGS_MODEL_NAME, api_key=st.secrets["TOGETHER_API_KEY"] ) shutil.rmtree(VECSTORE_PERSIST_DIRECTORY) - st.write(os.listdir()) - # if not os.path.exists(VECSTORE_PERSIST_DIRECTORY): - # os.makedirs(VECSTORE_PERSIST_DIRECTORY) - # if not os.path.exists(VECSTORE_PERSIST_DIRECTORY+CHROMA_SUBDIR_NAME): - # os.makedirs(VECSTORE_PERSIST_DIRECTORY+CHROMA_SUBDIR_NAME) - # download_file(VECSTORE_PERSIST_DIRECTORY+"/chroma.sqlite3",CHROMA_SQLITE3) - # for key in VECTORSTORE_LINKS: - # download_file(VECSTORE_PERSIST_DIRECTORY+CHROMA_SUBDIR_NAME+"/"+key,VECTORSTORE_LINKS[key]) + if not os.path.exists(VECSTORE_PERSIST_DIRECTORY): + os.makedirs(VECSTORE_PERSIST_DIRECTORY) + if not os.path.exists(VECSTORE_PERSIST_DIRECTORY+CHROMA_SUBDIR_NAME): + os.makedirs(VECSTORE_PERSIST_DIRECTORY+CHROMA_SUBDIR_NAME) + download_file(VECSTORE_PERSIST_DIRECTORY+"/chroma.sqlite3",CHROMA_SQLITE3) + for key in VECTORSTORE_LINKS: + download_file(VECSTORE_PERSIST_DIRECTORY+CHROMA_SUBDIR_NAME+"/"+key,VECTORSTORE_LINKS[key]) st.write(os.listdir(VECSTORE_PERSIST_DIRECTORY)) st.write(os.listdir(VECSTORE_PERSIST_DIRECTORY+CHROMA_SUBDIR_NAME)) vectorstore = Chroma(