Update app.py

app.py

@@ -10,6 +10,11 @@ import urllib3
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 def generate_image(prompt, negative_prompt, use_negative_embedding, scheduler, steps, width, height, cfg, restore_faces, seed):
+    forbidden_words = os.getenv("FORBIDDEN_WORDS").split(", ")
+    # Check if the prompt contains any of the forbidden words
+    for word in forbidden_words:
+        if word in prompt:
+            raise Exception(f"The prompt contains a forbidden word: {word}")
     request_time = datetime.datetime.now()
     restore_faces = bool(restore_faces)
     use_negative_embedding = bool(use_negative_embedding)