Update download_repo.py

download_repo.py

@@ -14,6 +14,7 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder,
         # Construct the URL for the release's zip file
         encoded_project_id = urllib.parse.quote(project_id, safe="")
         url = f"{api_url}/projects/{encoded_project_id}/repository/archive.zip?sha={version}"
+        print("URL: ", url)
 
         logging.info(f"Constructed URL: {url}")
 
@@ -23,8 +24,10 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder,
 
         if response.status_code == 200:
             content_disposition = response.headers.get('content-disposition')
+            print("Content_disposition:", content_disposition)
             if content_disposition:
                 filename = content_disposition.split('filename=')[-1].strip('\"')
+                print("FILENAME: ", filename)
             else:
                 filename = 'archive.zip'  # Fallback to a default name if not found
 
@@ -35,6 +38,7 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder,
 
         existing_files = hf_api.list_repo_files(repo_id=hf_space_name, repo_type='space')
         target_path = f"{target_folder}/{filename}"
+        print("TARGET PATH: ", target_path)
 
         logging.info(f"Target Path: '{target_path}'")
         logging.info(f"Existing Files: {[repr(file) for file in existing_files]}")
@@ -43,7 +47,9 @@ 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)
+            repo_content= io.BytesIO(response.content)
+            print(repo_content)
+            _upload_file_to_hfspace(repo_content, 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,12 +61,11 @@ 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(content, 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_or_fileobj=content,
             path_in_repo=target_path,
             repo_id=hf_space_name,
             repo_type="space"