Preparing for presentation

app.py

@@ -6,7 +6,9 @@ import gradio as gr
 
 hf_token = os.getenv("HF_TOKEN")
 model_id = int(os.getenv("Model"))
-nsfw_filter = int(os.getenv("Safe"))
+nsfw_filter_enabled = int(os.getenv("Safe"))
+naughty_words = os.getenv("NaughtyWords").split()
+override = os.getenv("Override")
 
 #stable-diffusion-xl-base-1.0  0 - base model
 #Colossus_Project_XL           1 - better people
@@ -21,9 +23,6 @@ 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;
@@ -37,12 +36,12 @@ css = """
 
 def generate(prompt, neg_prompt, samp_steps, cfg_scale, batch_size, seed, height, width, progress=gr.Progress(track_tqdm=True)):
     prompt = prompt.lower()
-    if nsfw_filter:
-        if prompt[:10] == "krebzonide":
-            prompt = prompt[10:]
+    if nsfw_filter_enabled:
+        if prompt[:len(override)] == override:
+            prompt = prompt[len(override):]
         else:
-            neg_prompt = neg_prompt + ", child, nsfw, nipples, nude, underwear"
-            for word in naughtyWords:
+            neg_prompt = neg_prompt + ", child, nsfw, nude, underwear"
+            for word in naughty_words:
                 if prompt.find(word) >= 0:
                     return None, 58008
     if seed < 0:
@@ -90,11 +89,7 @@ examples = [
     ['cinematic photo of a woman sitting at a cafe, 35mm photograph, film, bokeh, professional, 4k, detailed face',
     'drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly'],
     ['analog film photo of old woman on the streets of london, faded film, desaturated, 35mm photo, grainy, vignette, vintage, Kodachrome, Lomography, stained, highly detailed, found footage',
-    'painting, drawing, illustration, glitch, deformed, mutated, cross-eyed, ugly, disfigured']#,
-    #['nude photo of a 20 year old model in the back seat of a car, detailed face',
-    #'big boobs'],
-    #['nude photo of a 20 year old man, penis and testicles, dick and balls, erection',
-    #'woman']
+    'painting, drawing, illustration, glitch, deformed, mutated, cross-eyed, ugly, disfigured']
 ]
 
 with gr.Blocks(css=css) as demo: