Migrate snapshot folder files to parent

download_model.py

@@ -9,37 +9,31 @@ MODEL_CACHE_DIR = "/app/model_cache"
 print("⏳ Downloading the SentenceTransformer model...")
 model_path = snapshot_download(repo_id=MODEL_REPO, cache_dir=MODEL_CACHE_DIR)
 
-# Find the correct snapshot directory
-# snapshots_dir = os.path.join(model_path, "snapshots")
-snapshots_dir = model_path
-print("Model path: ", snapshots_dir)
-if os.path.exists(snapshots_dir):
-    snapshot_folders = os.listdir(snapshots_dir)
-    if snapshot_folders:
-        snapshot_dir = os.path.join(snapshots_dir, snapshot_folders[0])  # Get first snapshot folder (e.g., migrate config file)
-        # Move all files and directories from the snapshot folder to /app/model_cache
-        for item in os.listdir(snapshot_folders):
-            source_path = os.path.join(snapshot_folders, item)
-            dest_path = os.path.join(MODEL_CACHE_DIR, item)
-            if os.path.exists(dest_path):
-                shutil.rmtree(dest_path) if os.path.isdir(dest_path) else os.remove(dest_path)
-            shutil.move(source_path, dest_path)
-        print(f"📂 Moving model files from {snapshot_folders} to {MODEL_CACHE_DIR}...")
-        # Move all model files (including subdirectories)
-        for item in os.listdir(snapshot_dir):
-            source_path = os.path.join(snapshot_dir, item)
-            dest_path = os.path.join(MODEL_CACHE_DIR, item)
-            if os.path.isdir(source_path):
-                shutil.move(source_path, dest_path)
-            else:
-                shutil.move(source_path, dest_path)
-        
-        print(f"✅ Model extracted and flattened in {MODEL_CACHE_DIR}")
-    else:
-        print("❌ No snapshot folder found!")
-        exit(1)
+# Find the correct model path and move files
+print("Model path: ", model_path)
+
+if os.path.exists(model_path):
+    # Ensure the destination directory is clean
+    if os.path.exists(MODEL_CACHE_DIR):
+        shutil.rmtree(MODEL_CACHE_DIR)
+    os.makedirs(MODEL_CACHE_DIR, exist_ok=True)
+
+    print(f"📂 Moving model files from {model_path} to {MODEL_CACHE_DIR}...")
+
+    # Move all files and directories from model_path to MODEL_CACHE_DIR
+    for item in os.listdir(model_path):
+        src_path = os.path.join(model_path, item)
+        dest_path = os.path.join(MODEL_CACHE_DIR, item)
+
+        # Move directories and files while preserving structure
+        if os.path.isdir(src_path):
+            shutil.move(src_path, dest_path)
+        else:
+            shutil.move(src_path, dest_path)
+
+    print(f"✅ Model extracted and flattened in {MODEL_CACHE_DIR}")
 else:
-    print("❌ No snapshots directory found!")
+    print("❌ Model download failed! Directory not found:", model_path)
     exit(1)
 
 # Verify structure