Update app.py

app.py

@@ -16,6 +16,13 @@ API_URL_DEV = "https://api-inference.huggingface.co/models/black-forest-labs/FLU
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 timeout = 100
 
+def check_ubuse(prompt,word_list=["little girl"]):
+    for word in word_list:
+        if word in prompt:
+            print(f"Abuse! prompt {prompt} wiped!")
+            return "Fuck You Bastard!"
+    return prompt
+    
 def enhance_prompt(prompt, model="mistralai/Mistral-7B-Instruct-v0.1", style="photo-realistic"):
     
     client = Client("K00B404/Mistral-Nemo-custom")
@@ -111,7 +118,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
         return None, None, None
 
     key = random.randint(0, 999)
-
+    prompt = check_ubuse(prompt)
     #prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
     print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')