Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import gradio as gr
 
 model_id = int(os.getenv("Model"))
+nsfw_filter = int(os.getenv("Safe"))
 
 #stable-diffusion-xl-base-1.0  0 - base model
 #Colossus_Project_XL           1 - better people
@@ -21,6 +22,9 @@ model_url_list = ["stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base
                  "SG161222/RealVisXL_V2.0/blob/main/RealVisXL_V2.0.safetensors",
                  "Krebzonide/AcornIsSpinning_acornXLV1/blob/main/acornIsSpinning_acornxlV1.safetensors"]
 
+naughtyWords = ["nude", "nsfw", "naked", "porn", "boob", "tit", "nipple", "vagina", "pussy", "panties", "underwear", "upskirt", "bottomless", "topless", "petite", "xxx"]
+
+
 css = """
 .btn-green {
   background-image: linear-gradient(to bottom right, #6dd178, #00a613) !important;
@@ -33,6 +37,10 @@ css = """
 """
 
 def generate(prompt, neg_prompt, samp_steps, cfg_scale, batch_size, seed, height, width, progress=gr.Progress(track_tqdm=True)):
+    if nsfw_filter:
+        for word in naughtyWords:
+            if prompt.find(word) > 0:
+                return None 58008
     if seed < 0:
         seed = random.randint(1,999999)
     images = pipe(