testtt

app.py

@@ -1,7 +1,7 @@
 import requests
 import os
 import gradio as gr
-from huggingface_hub import update_repo_visibility, whoami, upload_folder, create_repo, upload_file, update_repo_visibility, file_exists
+from huggingface_hub import update_repo_visibility, whoami, upload_folder, create_repo, upload_file, update_repo_visibility
 import subprocess
 
 import gradio as gr
@@ -43,15 +43,12 @@ 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, api_key=None):
+def get_files_by_username(username):
     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, headers=headers, timeout=30)
+        response = requests.get(url, timeout=30)
         data = response.json()
         # Add current page items to the list
         for model in data['items']:
@@ -70,14 +67,10 @@ def get_files_by_username(username, api_key=None):
         url = metadata.get('nextPage', None)
     return output
         
-def get_files_by_model_id(model_id, api_key=None):
+def get_files_by_model_id(model_id):
     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, headers=headers)
+        response = requests.get(api_url)
         response.raise_for_status()
         model = response.json()
 
@@ -100,10 +93,10 @@ def get_files_by_model_id(model_id, api_key=None):
 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, api_key)
+        files = get_files_by_model_id(model_id)
     elif url.startswith("https://civitai.com/user/"):
         username = url.split('/')[4]
-        files = get_files_by_username(username, api_key)
+        files = get_files_by_username(username)
     else:
         raise gr.Error("Unknown CivitAI URL format, please provide model URL or user profile URL")
     
@@ -117,13 +110,6 @@ def process_url(url, profile, user_repo_id, oauth_token, folder, api_key=None):
         try:
             download_url = data['downloadUrl']
             filename = dl_path.split('/')[-1]
-            if file_exists(
-                repo_id = user_repo_id,
-                filename = dl_path,
-                token = oauth_token
-            ):
-                gr.Info(f"Skipping {filename}, folder exists {dl_path}")
-                continue
             gr.Info(f"Downloading {filename} ({current_file}/{total_files})")
             download_file(download_url, dl_path, folder, api_key)
             # Upload the model and card