Update app.py

app.py

@@ -21,7 +21,7 @@ def get_json_data(url):
         print(f"Error fetching JSON data: {e}")
         return None
 
-def check_nsfw(json_data, profile: Optional[gr.OAuthProfile]):
+def check_nsfw(json_data, profile):
     if(profile.preferred_username in TRUSTED_UPLOADERS):
         return True
     if json_data["nsfw"]:
@@ -86,10 +86,10 @@ def download_file(url, filename, folder="."):
     except requests.exceptions.RequestException as e:
         raise gr.Error(f"Error downloading file: {e}")
 
-def process_url(url, do_download=True, folder="."):
+def process_url(url, profile, do_download=True, folder="."):
     json_data = get_json_data(url)
     if json_data:
-        if check_nsfw(json_data):
+        if check_nsfw(json_data, profile):
             info = extract_info(json_data)
             if info:
                 if(do_download):
@@ -179,7 +179,7 @@ def extract_huggingface_username(username):
 
 
 def check_civit_link(profile: Optional[gr.OAuthProfile], url):
-    info, _ = process_url(url, do_download=False)
+    info, _ = process_url(url, profile, do_download=False)
     hf_username = extract_huggingface_username(info['creator'])
     attributes_methods = dir(profile)
     
@@ -210,7 +210,7 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
     
     folder = str(uuid.uuid4())
     os.makedirs(folder, exist_ok=False)
-    info, downloaded_files = process_url(url, folder=folder)
+    info, downloaded_files = process_url(url, profile, folder=folder)
     create_readme(info, downloaded_files, folder=folder)
     try:
         api = HfApi(token=os.environ["HUGGING_FACE_HUB_TOKEN"])