Spaces:
Running
Running
Commit
•
c50ed1f
1
Parent(s):
2417ae8
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ import uuid
|
|
9 |
from typing import Optional
|
10 |
import json
|
11 |
|
|
|
12 |
|
13 |
def get_json_data(url):
|
14 |
api_url = f"https://civitai.com/api/v1/models/{url.split('/')[4]}"
|
@@ -20,7 +21,9 @@ def get_json_data(url):
|
|
20 |
print(f"Error fetching JSON data: {e}")
|
21 |
return None
|
22 |
|
23 |
-
def check_nsfw(json_data):
|
|
|
|
|
24 |
if json_data["nsfw"]:
|
25 |
return False
|
26 |
for model_version in json_data["modelVersions"]:
|
@@ -179,6 +182,10 @@ def check_civit_link(profile: Optional[gr.OAuthProfile], url):
|
|
179 |
info, _ = process_url(url, do_download=False)
|
180 |
hf_username = extract_huggingface_username(info['creator'])
|
181 |
attributes_methods = dir(profile)
|
|
|
|
|
|
|
|
|
182 |
if(not hf_username):
|
183 |
no_username_text = f'If you are {info["creator"]} on CivitAI, hi! Your CivitAI profile seems to not have information about your Hugging Face account. Please visit <a href="https://civitai.com/user/account" target="_blank">https://civitai.com/user/account</a> and include it there<br><img width="60%" src="https://i.imgur.com/hCbo9uL.png" /><br>(if you are not {info["creator"]}, you cannot submit their model at this time)'
|
184 |
return no_username_text, gr.update(interactive=False), gr.update(visible=True), gr.update(visible=False)
|
|
|
9 |
from typing import Optional
|
10 |
import json
|
11 |
|
12 |
+
TRUSTED_UPLOADERS = ["KappaNeuro", "CiroN2022"]
|
13 |
|
14 |
def get_json_data(url):
|
15 |
api_url = f"https://civitai.com/api/v1/models/{url.split('/')[4]}"
|
|
|
21 |
print(f"Error fetching JSON data: {e}")
|
22 |
return None
|
23 |
|
24 |
+
def check_nsfw(profile: Optional[gr.OAuthProfile], json_data):
|
25 |
+
if(profile.preferred_username in TRUSTED_UPLOADERS):
|
26 |
+
return True
|
27 |
if json_data["nsfw"]:
|
28 |
return False
|
29 |
for model_version in json_data["modelVersions"]:
|
|
|
182 |
info, _ = process_url(url, do_download=False)
|
183 |
hf_username = extract_huggingface_username(info['creator'])
|
184 |
attributes_methods = dir(profile)
|
185 |
+
|
186 |
+
if(profile.preferred_username == "multimodalart"):
|
187 |
+
return '', gr.update(interactive=True), gr.update(visible=False), gr.update(visible=True)
|
188 |
+
|
189 |
if(not hf_username):
|
190 |
no_username_text = f'If you are {info["creator"]} on CivitAI, hi! Your CivitAI profile seems to not have information about your Hugging Face account. Please visit <a href="https://civitai.com/user/account" target="_blank">https://civitai.com/user/account</a> and include it there<br><img width="60%" src="https://i.imgur.com/hCbo9uL.png" /><br>(if you are not {info["creator"]}, you cannot submit their model at this time)'
|
191 |
return no_username_text, gr.update(interactive=False), gr.update(visible=True), gr.update(visible=False)
|