debugging streamlit deployment

Medkallel · Oct 14, 2024 · a42baaf · a42baaf
1 parent 1edc32b
commit a42baaf
Showing 1 changed file with 7 additions and 8 deletions.
diff --git 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(