Update app.py

app.py

@@ -29,23 +29,37 @@ large_text_theme = Base(
 
 
 CONFIG_PATH = "config.json"
+PERSISTENT_DIR = "/data"
+
+FORCE_UPDATE_FLAG = False
 
 DEFAULT_LLM = "Jatin19K/unsloth-q5_k_m-mistral-nemo-instruct-2407"
 DEFAULT_VECTOR_STORE = "CFIR"
 
+
+EMBED_MODEL_PATH = os.path.join(PERSISTENT_DIR, "bge_onnx")
+VECTOR_STORE_DIR = os.path.join(PERSISTENT_DIR, "vector_stores")
+
+
 token = os.getenv("HF_TOKEN")
 dataset_id = os.getenv("DATASET_ID")
 
+def download_data_if_needed():
+    if not os.path.exists(EMBED_MODEL_PATH) or not os.path.exists(VECTOR_STORE_DIR):
+        FORCE_UPDATE_FLAG = True
 
+    if FORCE_UPDATE_FLAG:
+        snapshot_download(
+                repo_id=dataset_id,
+                repo_type="dataset",
+                token=token,
+                local_dir=PERSISTENT_DIR
+            )
+        print("Data downloaded successfully.")
+    else:
+        print("Data exists.")
 
-local_dir = snapshot_download(
-            repo_id=dataset_id,
-            repo_type="dataset",
-            token=token
-        )
-
-EMBED_MODEL_PATH = os.path.join(local_dir, "bge_onnx")
-VECTOR_STORE_DIR = os.path.join(local_dir, "vector_stores")
+download_data_if_needed()
 
 class ModelManager:
     def __init__(self):