Update app.py

app.py

@@ -48,9 +48,60 @@ pipeline.to("cpu")
 pipeline.compile()  # Biên dịch mô hình nếu hỗ trợ
 
 # Tối ưu hóa từ điển NSFW - sử dụng set để tìm kiếm nhanh hơn
+NSFW_HIGH = {
+    "nude", "naked", "sex", "porn", "xxx", "fuck", "dick", "cock", "pussy", "vagina", "penis",
+    "boobs", "tits", "breasts", "bra", "panties", "underwear", "lingerie", "orgasm", "cum",
+    "blowjob", "handjob", "masturbate", "rape", "gangbang", "incest", "hentai", "lewd",
+    "erotic", "kinky", "bondage", "bdsm", "squirt", "creampie", "threesome", "orgy", "yaoi",
+    "yuri", "futanari", "cunnilingus", "fellatio", "anal", "paizuri", "bukkake", "guro",
+    "vore", "tentacle", "netorare", "cuckold", "exhibitionism", "voyeurism", "poop", "pee",
+    "poo", "shit", "piss", "scat", "diarrhea", "vomit", "gore", "blood", "murder",
+    "torture", "suicide", "decapitation", "mutilation", "drugs", "cocaine", "heroin",
+    "lsd", "ecstasy", "vlxx"
+}
+
+NSFW_MEDIUM = {
+    "bikini", "swimwear", "sexy", "succubus", "leather", "latex", "stockings", "miniskirt",
+    "cleavage", "thighs", "ass", "butt", "skirt", "dress", "topless", "wet", "moaning",
+    "spread", "legs apart", "tight", "revealing", "provocative", "suggestive", "flirty"
+}
 
+NSFW_PHRASES = {
+    "spreading legs", "removing bra", "pulling panties", "sucking dick", "licking pussy",
+    "penetrating", "fucking scene", "hard cock", "wet pussy", "big tits", "exposed breasts",
+    "nipples visible", "ass spread", "thigh gap", "camel toe", "pussy lips", "cum on face",
+    "blowjob scene", "anal sex", "titty fuck", "gang rape", "group sex", "public sex",
+    "hidden camera", "peeing girl", "pooping girl", "covered in blood", "cutting flesh",
+    "snorting cocaine", "injecting heroin", "hallucinating", "smoking weed"
+}
+
+SENSITIVE_CONTEXT = {
+    "spread", "removing", "pulling", "sucking", "licking", "penetrating",
+    "fucking", "hard", "wet", "exposed", "visible", "ass", "tight", "revealing"
+}
 
+# Tối ưu hóa hàm NSFW detection
+@lru_cache(maxsize=1024)
+def detect_nsfw(prompt: str):
+    prompt_lower = prompt.lower()
+    
+    # Kiểm tra từ khóa cao
+    words = set(re.findall(r'\b\w+\b', prompt_lower))
+    if words & NSFW_HIGH:
+        return True
+    
+    # Kiểm tra cụm từ NSFW
+    for phrase in NSFW_PHRASES:
+        if phrase in prompt_lower:
+            return True
+    
     # Kiểm tra từ khóa trung bình + ngữ cảnh nhạy cảm
+    medium_matches = words & NSFW_MEDIUM
+    if medium_matches and (words & SENSITIVE_CONTEXT):
+        return True
+    
+    return False
+
 def infer(
     prompt: str,
     negative_prompt: str,
@@ -70,6 +121,9 @@ def infer(
     print(f"🚫 Negative Prompt: {negative_prompt or '[None]'}")
     
     # Tối ưu hóa kiểm tra NSFW
+    if detect_nsfw(prompt):
+        raise gr.Error("⚠️ Prompt contains NSFW content. Please use a safe prompt.")
+    
     if randomize_seed:
         seed = random.randint(0, np.iinfo(np.int32).max)
         print(f"🎲 Random Seed generated: {seed}")