Update app.py

app.py

@@ -5,6 +5,7 @@ import subprocess
 from huggingface_hub import InferenceClient
 from io import BytesIO
 from PIL import Image
+import re
 
 # Initialize the Flask app
 app = Flask(__name__)
@@ -20,23 +21,35 @@ mutated fingers, poorly drawn fingers, disfigured fingers,
 too many fingers, deformed hands, extra hands, malformed hands, 
 blurry hands, disproportionate fingers"""
 
+# Define the list of explicit keywords
+EXPLICIT_KEYWORDS = [
+    "sexual", "sex", "boobs", "boob", "breasts", "cleavage", "penis", "phallus", "porn", "pornography", "hentai", 
+    "fetish", "nude", "nudity", "provocative", "obscene", "vulgar", "intimate", "kinky", "hardcore", 
+    "threesome", "orgy", "masturbation", "masturbate", "genital", "genitals", "vagina", "vaginal", 
+    "anus", "anal", "butt", "buttocks", "butthole", "ass", "prostate", "erection", "cum", "ejaculation", 
+    "sperm", "semen", "naked", "bare", "lingerie", "thong", "striptease", "stripper", 
+    "seductive", "sensual", "explicit", "lewd", "taboo", "NSFW", "bdsm", "dominatrix", "submission", 
+    "intercourse", "penetration", "orgasm", "fuck", "fucking", "fuckers", "fucker", "slut", "whore", 
+    "prostitute", "hooker", "escort", "camgirl", "camwhore", "sugar daddy", "sugar baby", "adult content", 
+    "sexually explicit", "arousal", "lust", "depraved", "hardcore porn", "softcore", "erotic", "erotica",
+    "roleplay", "incest", "taboo", "voyeur", "exhibitionist", "peeping", "dildo", "sex toy", "vibrator", 
+    "suicide", "self-harm", "depression", "kill myself", "worthless", "abuse", "violence", "rape", 
+    "sexual violence", "molestation", "pedophilia", "child porn", "underage", "illegal content"
+]
+
+# Function to scan the entire prompt for explicit keywords
+def scan_prompt(prompt, keywords):
+    # Create a regex pattern to match any keyword (case insensitive)
+    pattern = r'\b(?:' + '|'.join(re.escape(keyword) for keyword in keywords) + r')\b'
+    # Find all matches in the prompt
+    matches = re.findall(pattern, prompt, flags=re.IGNORECASE)
+    # Return True if matches are found, and the list of matched keywords
+    return bool(matches), matches
+
 @app.route('/')
 def home():
     return "Welcome to the Image Background Remover!"
 
-# Simple content moderation function
-def is_prompt_explicit(prompt):
-    # Streamlined keyword list to avoid unnecessary restrictions
-    explicit_keywords = [
-        "sexual","boobs","boob","penis","porn", "hentai", "fetish", "nude", "provocative", "obscene", "vulgar", "intimate", "kinky", "hardcore",
-        "threesome", "orgy", "masturbation", "genital", "suicide", 
-        "self-harm", "depression", "kill myself", "worthless","vagina","intercourse","fuck","fucking","fuckers","fucker"
-    ]
-    for keyword in explicit_keywords:
-        if keyword.lower() in prompt.lower():
-            return True
-    return False
-
 # Function to generate an image from a text prompt
 def generate_image(prompt, negative_prompt=None, height=512, width=512, model="stabilityai/stable-diffusion-2-1", num_inference_steps=50, guidance_scale=7.5, seed=None):
     try:
@@ -75,14 +88,16 @@ def generate_api():
         return jsonify({"error": "Prompt is required"}), 400
 
     try:
-        # Check for explicit content
-        if is_prompt_explicit(prompt):
-            # Return the pre-defined "thinkgood.png" image
+        # Check for explicit content using scan_prompt
+        is_nsfw, found_keywords = scan_prompt(prompt, EXPLICIT_KEYWORDS)
+        if is_nsfw:
+            print(f"Explicit keywords found: {found_keywords}")
+            # Return the pre-defined "nsfw.jpg" image
             return send_file(
                 "nsfw.jpg",
                 mimetype='image/png',
                 as_attachment=False,
-                download_name='thinkgood.png'
+                download_name='nsfw_detected.png'
             )
 
         # Call the generate_image function with the provided parameters