Update app.py

app.py

@@ -17,14 +17,9 @@ client = InferenceClient(token=HF_TOKEN)
 def home():
     return "Welcome to the Image Background Remover!"
 
-# Function to generate a black image
-def generate_black_image(height, width):
-    black_image = Image.new("RGB", (width, height), color="black")
-    return black_image
-
 # Simple content moderation function
 def is_prompt_explicit(prompt):
-    explicit_keywords = ["sexual", "nudity", "erotic", "explicit", "porn", "pornographic", "xxx", "hentai", "fetish", "sex", "sensual", "nude", "strip", "stripping", "adult", "lewd", "provocative", "obscene", "vulgar", "intimacy", "intimate", " lust", "arouse", "seductive", "seduction", "kinky", "bdsm", "dominatrix", "bondage", "hardcore", "softcore", "topless", "bottomless", "threesome", "orgy", "incest", "taboo", "masturbation", "genital", "penis", "vagina", "breast", "boob", "nipple", "butt", "anal", "oral", "ejaculation", "climax", "moan", "foreplay", "intercourse", "naked", "exposed", "suicide", "self-harm", " overdose", "poison", "hang", " end life", "kill myself", "noose", "depression", "hopeless", "worthless", "die", "death", "harm myself"]  # Add more keywords as needed
+    explicit_keywords = ["sexual", "nudity", "erotic", "explicit", "porn", "pornographic", "xxx", "hentai", "fetish", "sex", "sensual", "nude", "strip", "stripping", "adult", "lewd", "provocative", "obscene", "vulgar", "intimacy", "intimate", "lust", "arouse", "seductive", "seduction", "kinky", "bdsm", "dominatrix", "bondage", "hardcore", "softcore", "topless", "bottomless", "threesome", "orgy", "incest", "taboo", "masturbation", "genital", "penis", "vagina", "breast", "boob", "nipple", "butt", "anal", "oral", "ejaculation", "climax", "moan", "foreplay", "intercourse", "naked", "exposed", "suicide", "self-harm", "overdose", "poison", "hang", "end life", "kill myself", "noose", "depression", "hopeless", "worthless", "die", "death", "harm myself"]  # Add more keywords as needed
     for keyword in explicit_keywords:
         if keyword.lower() in prompt.lower():
             return True
@@ -70,14 +65,9 @@ def generate_api():
     try:
         # Check for explicit content
         if is_prompt_explicit(prompt):
-            # Generate and return a black image
-            black_image = generate_black_image(height, width)
-            img_byte_arr = BytesIO()
-            black_image.save(img_byte_arr, format='PNG')  # Save as PNG
-            img_byte_arr.seek(0)  # Reset byte stream
-
+            # Return the pre-defined "thinkgood.png" image
             return send_file(
-                img_byte_arr,
+                "thinkgood.png",
                 mimetype='image/png',
                 as_attachment=False,
                 download_name='thinkgood.png'