Add civitai-to-hf-uploader

app.py

@@ -10,7 +10,6 @@ from typing import Optional
 import json
 def download_file(url, file_path, folder):
     headers = {}
-    gr.Info("Downloading file from "+ url + " to "+ file_path)
     try:
         response = requests.get(url, headers=headers)
         response.raise_for_status()
@@ -32,8 +31,7 @@ def download_file(url, file_path, folder):
 
     with open(os.path.join(folder, file_path), 'wb') as f:
         f.write(response.content)
-    
-    gr.Info("Downloaded "+ file_path)
+
         
 def get_files_by_username(username):
     url = f"https://civitai.com/api/v1/models?username={username}&limit=100"
@@ -82,10 +80,15 @@ def process_url(url, profile, do_download=True, folder="."):
     
     if do_download:
         downloaded_files = {}
+        total_files = len(files)
+        current_file = 1
         for dl_path, download_url in files.items():
             try:
+                filename = dl_path.split('/')[-1]
+                gr.Info(f"Downloading {filename} ({current_file}/{total_files})")
                 download_file(download_url, dl_path, folder)
                 downloaded_files[dl_path] = download_url
+                current_file += 1
             except Exception as e:
                 gr.Warning(f"Failed to download {dl_path}: {str(e)}")
         return downloaded_files
@@ -129,6 +132,8 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
         files = process_url(url, profile, folder=folder)
         if not files or len(files.keys()) == 0:
             return gr.Error("No files were downloaded. Please check the URL and try again.")
+        
+        gr.Info(f"Downloaded {len(files)} files")
             
         results = []
 
@@ -142,6 +147,7 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
             gr.Info(f"Repository {user_repo_id} already exists, will update it:"+ str(e))
 
         gr.Info(f"Uploading models to {user_repo_id}...")
+        gr.Info(f"This may take a while, please don't close the page")
         
         # Upload the model and card
         upload_folder(