Ais commited on
Commit
b67a3d2
·
verified ·
1 Parent(s): 9c1c873

Update app/download_adapter.py

Browse files
Files changed (1) hide show
  1. app/download_adapter.py +1 -10
app/download_adapter.py CHANGED
@@ -2,22 +2,17 @@ import os
2
  import gdown
3
  import re
4
 
5
- # ✅ Google Drive folder with adapter versions like 'version 1', 'version 2', etc.
6
  DRIVE_FOLDER_URL = "https://drive.google.com/drive/folders/1S9xT92Zm9rZ4RSCxAe_DLld8vu78mqW4"
7
- LOCAL_DEST = "adapter" # Where we'll copy the latest version
8
 
9
  def download_latest_adapter():
10
  print("🔽 Downloading adapter folder from Google Drive...")
11
-
12
- # Download everything from the Drive folder into temp dir
13
  gdown.download_folder(url=DRIVE_FOLDER_URL, output="gdrive_tmp", quiet=False, use_cookies=False)
14
 
15
- # Find all folders named "version X"
16
  all_versions = sorted(
17
  [d for d in os.listdir("gdrive_tmp") if re.match(r"version \d+", d)],
18
  key=lambda x: int(x.split()[-1])
19
  )
20
-
21
  if not all_versions:
22
  raise ValueError("❌ No version folders found in Google Drive folder.")
23
 
@@ -25,10 +20,7 @@ def download_latest_adapter():
25
  src = os.path.join("gdrive_tmp", latest)
26
  print(f"✅ Latest adapter found: {latest}")
27
 
28
- # Ensure destination exists
29
  os.makedirs(LOCAL_DEST, exist_ok=True)
30
-
31
- # Copy files to destination
32
  for file in os.listdir(src):
33
  src_file = os.path.join(src, file)
34
  dest_file = os.path.join(LOCAL_DEST, file)
@@ -36,6 +28,5 @@ def download_latest_adapter():
36
 
37
  print(f"✅ Adapter copied to: {LOCAL_DEST}")
38
 
39
- # ✅ Run automatically if script is executed directly
40
  if __name__ == "__main__":
41
  download_latest_adapter()
 
2
  import gdown
3
  import re
4
 
 
5
  DRIVE_FOLDER_URL = "https://drive.google.com/drive/folders/1S9xT92Zm9rZ4RSCxAe_DLld8vu78mqW4"
6
+ LOCAL_DEST = "adapter"
7
 
8
  def download_latest_adapter():
9
  print("🔽 Downloading adapter folder from Google Drive...")
 
 
10
  gdown.download_folder(url=DRIVE_FOLDER_URL, output="gdrive_tmp", quiet=False, use_cookies=False)
11
 
 
12
  all_versions = sorted(
13
  [d for d in os.listdir("gdrive_tmp") if re.match(r"version \d+", d)],
14
  key=lambda x: int(x.split()[-1])
15
  )
 
16
  if not all_versions:
17
  raise ValueError("❌ No version folders found in Google Drive folder.")
18
 
 
20
  src = os.path.join("gdrive_tmp", latest)
21
  print(f"✅ Latest adapter found: {latest}")
22
 
 
23
  os.makedirs(LOCAL_DEST, exist_ok=True)
 
 
24
  for file in os.listdir(src):
25
  src_file = os.path.join(src, file)
26
  dest_file = os.path.join(LOCAL_DEST, file)
 
28
 
29
  print(f"✅ Adapter copied to: {LOCAL_DEST}")
30
 
 
31
  if __name__ == "__main__":
32
  download_latest_adapter()