Update app.py

app.py

@@ -2,13 +2,20 @@ import gradio as gr
 import requests
 import json
 import time
+from profanity_check import predict  # For NSFW detection
 
 url = "https://nexra.aryahcr.cc/api/image/complements"
 headers = {
     "Content-Type": "application/json"
 }
 
+def is_nsfw(text):
+    return predict([text])[0] == 1  
+
 def generate_image(prompt, model):
+    if is_nsfw(prompt):
+        return "Error: Prompt contains inappropriate content."
+
     data = {
         "prompt": prompt,
         "model": model,