Update app.py

app.py

@@ -56,7 +56,7 @@ def save_usage_data(data):
 # بارگذاری اولیه داده‌ها
 usage_data_cache = load_usage_data()
 
-# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
+# --- بارگذاری مدل تشخیص محتوای نامناسب (Image Safety) ---
 print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
@@ -133,18 +133,56 @@ ASPECT_RATIOS_MAP = {
     "شخصی‌سازی (Custom)": "Custom"
 }
 
+# --- لیست کلمات ممنوعه پیشرفته (Strict Mode) ---
 BANNED_WORDS = [
-    "nude", "naked", "sex", "porn", "undressed", "nsfw", "erotic", "xxx", 
-    "breast", "nipple", "genital", "vagina", "penis", "ass", "butt", "sexual",
-    "lingerie", "bikini", "swimwear", "underwear", "fetish", "topless",
-    "exhibitionism", "hentai", "ecchi", "18+"
+    # General NSFW
+    "nsfw", "nude", "naked", "sex", "porn", "erotic", "xxx", "18+", "adult",
+    "explicit", "uncensored", "sexual", "lewd", "sensual", "lust", "horny",
+    
+    # Anatomy (Explicit)
+    "breast", "breasts", "nipple", "nipples", "vagina", "pussy", "cunt",
+    "penis", "dick", "cock", "genital", "genitals", "groin", "pubic",
+    "ass", "butt", "buttocks", "anus", "anal", "rectum",
+    
+    # Acts / Concepts
+    "intercourse", "masturbation", "orgasm", "blowjob", "bj", "cum", "sperm",
+    "ejaculation", "penetration", "fucking", "sucking", "licking",
+    
+    # Clothing / Lingerie
+    "lingerie", "bikini", "swimwear", "underwear", "panties", "bra", "thong",
+    "topless", "bottomless", "undressed", "unclothed", "skimpy", "transparent",
+    "see-through",
+    
+    # Fetish / Styles
+    "fetish", "bdsm", "bondage", "latex", "hentai", "ecchi", "ahegao",
+    "exhibitionism", "voyeur", "harem",
+    
+    # Violence / Gore (Optional but recommended for strict safety)
+    "gore", "bloody", "blood", "kill", "murder", "dead", "torture", "abuse"
 ]
 
 def check_text_safety(text):
+    """
+    بررسی وجود کلمات ممنوعه در متن انگلیسی.
+    از متد whole word matching ساده استفاده می‌کند تا کلماتی مثل 'assassin' اشتباهاً حذف نشوند.
+    """
+    if not text: return True
     text_lower = text.lower()
+    
+    # نرمال‌سازی متن برای جستجوی دقیق‌تر
+    # اضافه کردن فاصله به اول و آخر متن برای تشخیص کلمات اول و آخر جمله
+    padded_text = f" {text_lower} "
+    
+    # حذف علائم نگارشی برای جلوگیری از دور زدن فیلتر (مثلاً sex.)
+    for char in [".", ",", "!", "?", "-", "_", "(", ")", "[", "]", "{", "}"]:
+        padded_text = padded_text.replace(char, " ")
+    
     for word in BANNED_WORDS:
-        if f" {word} " in f" {text_lower} ": 
+        # جستجو به صورت کلمه مستقل
+        if f" {word} " in padded_text: 
+            print(f"Safety Triggered: Found banned word '{word}'")
             return False
+            
     return True
 
 def translate_prompt(text):
@@ -282,20 +320,22 @@ def infer(
                 None, seed, get_quota_exceeded_html(), 
                 gr.update(visible=False), gr.update(visible=True)
             )
-        
-        # اگر اعتبار داشت، حالا کم می‌کنیم
-        consume_quota(fingerprint)
-
-    # --- ادامه منطق اصلی ساخت تصویر ---
+    
+    # --- بررسی‌های اولیه (تصویر و متن) قبل از کسر اعتبار ---
     if input_image is None:
         return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید."), gr.update(visible=True), gr.update(visible=False)
 
     if is_image_nsfw(input_image):
         return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود."), gr.update(visible=True), gr.update(visible=False)
 
+    # ترجمه و بررسی متن (مهم: قبل از کسر اعتبار)
     english_prompt = translate_prompt(prompt)
     if not check_text_safety(english_prompt):
-        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است."), gr.update(visible=True), gr.update(visible=False)
+        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز، جنسی یا خشونت‌آمیز است. لطفاً متن را اصلاح کنید."), gr.update(visible=True), gr.update(visible=False)
+
+    # اگر همه چیز امن بود، حالا اعتبار را کسر می‌کنیم
+    if subscription_status != 'paid':
+        consume_quota(fingerprint)
 
     adapter_internal_name = LORA_MAPPING.get(lora_adapter_persian)
     if adapter_internal_name:
@@ -306,7 +346,8 @@ def infer(
 
     generator = torch.Generator(device=device).manual_seed(seed)
     
-    safety_negative = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood"
+    # اضافه کردن کلمات منفی ایمنی به صورت پیش‌فرض به پرامپت منفی
+    safety_negative = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood, explicit"
     base_negative = "worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
     final_negative_prompt = f"{safety_negative}, {base_negative}"
 
@@ -334,6 +375,7 @@ def infer(
             true_cfg_scale=guidance_scale,
         ).images[0]
 
+        # بررسی نهایی تصویر تولید شده
         if is_image_nsfw(result):
             return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد."), gr.update(visible=True), gr.update(visible=False)
         
@@ -354,10 +396,9 @@ def infer(
     except Exception as e:
         error_str = str(e)
         # -------------------------------------------------------------
-        #  مدیریت خطای GPU QUOTA برای نمایش Modal سفارشی (بخش درخواستی)
+        #  مدیریت خطای GPU QUOTA
         # -------------------------------------------------------------
         if "quota" in error_str.lower() or "exceeded" in error_str.lower():
-             # خطا را به بالا پرتاب می‌کنیم تا اسکنر جاوااسکریپت آن را بگیرد
              raise e 
         
         return None, seed, get_error_html(f"خطا در پردازش: {error_str}"), gr.update(visible=True), gr.update(visible=False)
@@ -395,7 +436,7 @@ js_upgrade_func = """
 }
 """
 
-# --- جاوااسکریپت سراسری (ترکیب شده: اعتبار سنجی + هندلینگ خطای GPU) ---
+# --- جاوااسکریپت سراسری ---
 js_global_content = """
 <script>
 document.addEventListener('DOMContentLoaded', () => {
@@ -631,7 +672,7 @@ document.addEventListener('DOMContentLoaded', () => {
 </script>
 """
 
-# --- CSS Updated (ترکیب استایل‌های مودال، بج و گرادیو) ---
+# --- CSS Updated ---
 css_code = """
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
@@ -650,7 +691,7 @@ css_code = """
     font-family: 'Vazirmatn', sans-serif !important;
 }
 
-/* --- IP Reset Guide CSS (مودال سفارشی) --- */
+/* --- IP Reset Guide CSS (مودال) --- */
 :root {
     --guide-bg: rgba(255, 255, 255, 0.98);
     --guide-border: rgba(102, 126, 234, 0.2);