Add civitai-to-hf-uploader

app.py

@@ -43,12 +43,15 @@ def download_file(url, file_path, folder, api_key=None):
         raise gr.Error(f"Error downloading file: {str(e)}")
 
         
-def get_files_by_username(username):
+def get_files_by_username(username, api_key=None):
     url = f"https://civitai.com/api/v1/models?username={username}&limit=100&nsfw=true"
     output = {}
+    headers = {}
+    if api_key:
+        headers['Authorization'] = f'Bearer {api_key}'
 
     while url:
-        response = requests.get(url, timeout=30)
+        response = requests.get(url, headers=headers, timeout=30)
         data = response.json()
         # Add current page items to the list
         for model in data['items']:
@@ -67,10 +70,14 @@ def get_files_by_username(username):
         url = metadata.get('nextPage', None)
     return output
         
-def get_files_by_model_id(model_id):
+def get_files_by_model_id(model_id, api_key=None):
     api_url = f"https://civitai.com/api/v1/models/{model_id}"
+    headers = {}
+    if api_key:
+        headers['Authorization'] = f'Bearer {api_key}'
+        
     try:
-        response = requests.get(api_url)
+        response = requests.get(api_url, headers=headers)
         response.raise_for_status()
         model = response.json()
 
@@ -93,10 +100,10 @@ def get_files_by_model_id(model_id):
 def process_url(url, profile, user_repo_id, oauth_token, folder, api_key=None):
     if url.startswith("https://civitai.com/models/"):
         model_id = url.split('/')[4]
-        files = get_files_by_model_id(model_id)
+        files = get_files_by_model_id(model_id, api_key)
     elif url.startswith("https://civitai.com/user/"):
         username = url.split('/')[4]
-        files = get_files_by_username(username)
+        files = get_files_by_username(username, api_key)
     else:
         raise gr.Error("Unknown CivitAI URL format, please provide model URL or user profile URL")