Update app.py

app.py

@@ -13,13 +13,16 @@ import torch
 from diffusers import DiffusionPipeline
 
 bad_words = json.loads(os.getenv('BAD_WORDS', "[]"))
+bad_words_negative = json.loads(os.getenv('BAD_WORDS_NEGATIVE', "[]"))
+default_negative = os.getenv("default_negative","")
 
-#print(bad_words)
-
-def check_text(prompt):
+def check_text(prompt, negative=""):
     for i in bad_words:
         if i in prompt:
             return True
+    for i in bad_words_negative:
+        if i in negative:
+            return True
     return False
 
 DESCRIPTION = """# RealVisXL_V3.0"""
@@ -77,7 +80,7 @@ def generate(prompt: str,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True)):
-    if check_text(prompt):
+    if check_text(prompt, negative_prompt):
         return 'NSFW Detection'
     return _generate(
         prompt,
@@ -110,8 +113,8 @@ def _generate(
     generator = torch.Generator().manual_seed(seed)
 
     if not use_negative_prompt:
-        negative_prompt = None  # type: ignore
-    
+        negative_prompt = ""  # type: ignore
+    negative_prompt += default_negative    
     images = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,