Update app.py

app.py

@@ -22,10 +22,10 @@ def get_json_data(url):
         return None
 
 def check_nsfw(json_data, profile):
-    if(profile.preferred_username in TRUSTED_UPLOADERS):
-        return True
     if json_data["nsfw"]:
         return False
+    if(profile.preferred_username in TRUSTED_UPLOADERS):
+        return True
     for model_version in json_data["modelVersions"]:
         for image in model_version["images"]:
             if image["nsfw"] != "None":
@@ -43,12 +43,15 @@ def extract_info(json_data):
                         
                         # Then append all image URLs to the list
                         for image in model_version["images"]:
-                            urls_to_download.append({
-                                "url": image["url"],
-                                "filename": os.path.basename(image["url"]),
-                                "type": "imageName",
-                                "prompt": image["meta"]["prompt"] if image["meta"] is not None and "prompt" in image["meta"] else ""
-                            })
+                            if image["nsfw"] != "None":
+                                pass #ugly before checking the actual logic
+                            else:
+                                urls_to_download.append({
+                                    "url": image["url"],
+                                    "filename": os.path.basename(image["url"]),
+                                    "type": "imageName",
+                                    "prompt": image["meta"]["prompt"] if image["meta"] is not None and "prompt" in image["meta"] else ""
+                                })
                         
                         info = {
                             "urls_to_download": urls_to_download,
@@ -286,8 +289,10 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, link_civit=False
 def bulk_upload(profile: Optional[gr.OAuthProfile], urls, link_civit=False, progress=gr.Progress(track_tqdm=True)):
     for url in urls.split("\n"):
         if(url):
-            yield upload_civit_to_hf(profile, url, link_civit)
-
+            try:
+                yield upload_civit_to_hf(profile, url, link_civit)
+            except:
+                gr.Info(f"Something wrong happened uploading {link_civit}, skipping that model")
 css = '''
 #login {
     font-size: 0px;