Update download_repo.py

download_repo.py

@@ -9,7 +9,7 @@ import logging
 # Set up logging
 logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
 
-def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder, hf_api):
+def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder, hf_api, hf_space_name):
     try:
         # Construct the URL for the release's zip file
         encoded_project_id = urllib.parse.quote(project_id, safe="")
@@ -33,7 +33,7 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder,
             logging.error(f"Failed to download the file. HTTP Status: {response.status_code}")
             return
 
-        existing_files = hf_api.list_repo_files(repo_id=HF_SPACE_NAME, repo_type='space')
+        existing_files = hf_api.list_repo_files(repo_id=hf_space_name, repo_type='space')
         target_path = f"{target_folder}/{filename}"
 
         logging.info(f"Target Path: '{target_path}'")
@@ -43,7 +43,7 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder,
             logging.warning(f"File '{target_path}' already exists in the repository. Skipping upload...")
             print(f"File '{target_path}' already exists in the repository. Skipping upload...")
         else:
-            _upload_file_to_hfspace(response, hf_api, target_path, HF_SPACE_NAME)
+            _upload_file_to_hfspace(response, hf_api, target_path, hf_space_name)
 
     except FileNotFoundError:
         logging.error("The config.json file was not found. Please ensure it exists in the project directory.")
@@ -55,14 +55,14 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder,
         logging.error(f"An unexpected error occurred: {e}")
         print(f"An unexpected error occurred: {e}")
 
-def _upload_file_to_hfspace(response, hf_api, target_path, HF_SPACE_NAME):
+def _upload_file_to_hfspace(response, hf_api, target_path, hf_space_name):
     try:
         archive_bytes = io.BytesIO(response.content)
         logging.info(f"Preparing to upload file to '{target_path}'")
         hf_api.upload_file(
             path_or_fileobj=archive_bytes,
             path_in_repo=target_path,
-            repo_id=HF_SPACE_NAME,
+            repo_id=hf_space_name,
             repo_type="space"
         )
         logging.info(f"File successfully uploaded to '{target_path}'")