Update app.py

app.py

@@ -4,117 +4,53 @@ import numpy as np
 import spaces
 import torch
 import random
-from PIL import Image
+from PIL import Image, ImageFilter
 from typing import Iterable
-from gradio.themes.utils import colors
+from gradio.themes import Soft
+from gradio.themes.utils import colors, fonts, sizes
 from deep_translator import GoogleTranslator
 from transformers import pipeline
-import datetime
-import time
-from collections import defaultdict
-import threading
+from datetime import datetime
 
-# --- تعریف تم (بدون تغییر) ---
+# --- متغیرهای سراسری برای مدیریت محدودیت ---
+USAGE_TRACKER = {}  # ذخیره: {ip: {'count': 0, 'date': 'YYYY-MM-DD'}}
+DAILY_LIMIT = 5     # محدودیت تعداد عکس در روز برای کاربران رایگان
+
+# --- تعریف تم ---
 colors.steel_blue = colors.Color(
-    name="steel_blue", c50="#EBF3F8", c100="#D3E5F0", c200="#A8CCE1", c300="#7DB3D2", c400="#529AC3", c500="#4682B4", c600="#3E72A0", c700="#36638C", c800="#2E5378", c900="#264364", c950="#1E3450",
+    name="steel_blue",
+    c50="#EBF3F8",
+    c100="#D3E5F0",
+    c200="#A8CCE1",
+    c300="#7DB3D2",
+    c400="#529AC3",
+    c500="#4682B4",
+    c600="#3E72A0",
+    c700="#36638C",
+    c800="#2E5378",
+    c900="#264364",
+    c950="#1E3450",
 )
 
-# --- سیستم مدیریت اعتبار کاربران (بخش جدید) ---
-USER_DATA = {}
-USER_LOCKS = defaultdict(threading.Lock)
-DAILY_CREDIT_LIMIT = 5
-
-def check_and_use_credit(fingerprint: str) -> bool:
-    """اعتبار کاربر را بررسی و در صورت وجود، یکی کم می‌کند."""
-    if not fingerprint:
-        return False
-    with USER_LOCKS[fingerprint]:
-        today = datetime.date.today().isoformat()
-        user = USER_DATA.get(fingerprint, {'credits': DAILY_CREDIT_LIMIT, 'last_reset': today})
-
-        # اگر تاریخ گذشته بود، اعتبار را ریست کن
-        if user['last_reset'] != today:
-            user['credits'] = DAILY_CREDIT_LIMIT
-            user['last_reset'] = today
-
-        if user['credits'] > 0:
-            user['credits'] -= 1
-            USER_DATA[fingerprint] = user
-            print(f"Credit used for {fingerprint}. Remaining: {user['credits']}")
-            return True
-        else:
-            print(f"No credits left for {fingerprint}.")
-            return False
-
-def get_user_status_api(fingerprint: str) -> dict:
-    """وضعیت فعلی کاربر را برای نمایش در UI برمی‌گرداند."""
-    if not fingerprint:
-        return {'credits': 0, 'next_reset_timestamp': 0}
-        
-    with USER_LOCKS[fingerprint]:
-        today = datetime.date.today()
-        user = USER_DATA.get(fingerprint, {'credits': DAILY_CREDIT_LIMIT, 'last_reset': today.isoformat()})
-
-        if user['last_reset'] != today.isoformat():
-            user['credits'] = DAILY_CREDIT_LIMIT
-            user['last_reset'] = today.isoformat()
-            USER_DATA[fingerprint] = user
-
-        tomorrow = today + datetime.timedelta(days=1)
-        next_reset_timestamp = int(time.mktime(tomorrow.timetuple()))
-        
-        return {'credits': user['credits'], 'next_reset_timestamp': next_reset_timestamp}
-
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# --- بارگذاری سیستم امنیتی دوگانه (بدون تغییر) ---
-print("Loading Safety Checkers...")
-safety_classifier_1 = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
-safety_classifier_2 = pipeline("image-classification", model="AdamCodd/vit-base-nsfw-detector", device=-1)
+# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
+print("Loading Safety Checker...")
+safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
 def is_image_nsfw(image):
     if image is None: return False
     try:
-        results1 = safety_classifier_1(image)
-        for result in results1:
-            if result['label'] == 'nsfw' and result['score'] > 0.5:
-                print(f"Safety Check 1 Failed: {result['score']}")
-                return True
-        results2 = safety_classifier_2(image)
-        for result in results2:
-            label, score = result['label'].lower(), result['score']
-            if label == 'nsfw' and score > 0.3:
-                print(f"Safety Check 2 (NSFW) Failed: {score}")
-                return True
-            if label in ['sexy', 'porn', 'hentai'] and score > 0.4:
-                print(f"Safety Check 2 (Partial) Failed: {label} - {score}")
+        results = safety_classifier(image)
+        for result in results:
+            if result['label'] == 'nsfw' and result['score'] > 0.75:
                 return True
         return False
     except Exception as e:
         print(f"Safety check error: {e}")
-        return True
-
-BANNED_WORDS = [
-    "nsfw", "nude", "naked", "sex", "porn", "erotic", "xxx", "18+", "uncensored", "breast", "nipple", "areola", "cleavage", "topless", "open chest", "genital", "vagina", "penis", "dick", "cock", "pussy", "ass", "butt", "anus", "lingerie", "bikini", "swimwear", "underwear", "panties", "bra", "fetish", "bdsm", "bondage", "exhibitionism", "voyeur", "hentai", "ecchi", "ahegao", "paizuri", "undressed", "stripping", "naked body", "exposed skin", "sheer", "see-through", "rape", "violence", "blood", "gore", "sexual"
-]
-
-def check_text_safety(text):
-    if not text: return True
-    text_lower = text.lower()
-    for word in BANNED_WORDS:
-        if word in text_lower:
-            print(f"Banned word found: {word}")
-            return False
-    return True
-
-def translate_prompt(text):
-    if not text: return ""
-    try:
-        return GoogleTranslator(source='auto', target='en').translate(text)
-    except Exception as e:
-        print(f"Translation Error: {e}")
-        return text
+        return False
 
-# --- بارگذاری مدل اصلی ویرایش تصویر (بدون تغییر) ---
+# --- بارگذاری مدل اصلی ---
 from diffusers import FlowMatchEulerDiscreteScheduler
 from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
@@ -123,15 +59,19 @@ from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-print("Loading Generation Pipeline...")
+print("Loading pipeline...")
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "Qwen/Qwen-Image-Edit-2509",
     transformer=QwenImageTransformer2DModel.from_pretrained(
-        "linoyts/Qwen-Image-Edit-Rapid-AIO", subfolder='transformer', torch_dtype=dtype, device_map='cuda'
+        "linoyts/Qwen-Image-Edit-Rapid-AIO",
+        subfolder='transformer',
+        torch_dtype=dtype,
+        device_map='cuda'
     ),
     torch_dtype=dtype
 ).to(device)
 
+# بارگذاری LoRA ها
 pipe.load_lora_weights("autoweeb/Qwen-Image-Edit-2509-Photo-to-Anime", weight_name="Qwen-Image-Edit-2509-Photo-to-Anime_000001000.safetensors", adapter_name="anime")
 pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Multiple-angles", weight_name="镜头转换.safetensors", adapter_name="multiple-angles")
 pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Light_restoration", weight_name="移除光影.safetensors", adapter_name="light-restoration")
@@ -145,406 +85,861 @@ pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 MAX_SEED = np.iinfo(np.int32).max
 
 LORA_MAPPING = {
-    "تبدیل عکس به انیمه": "anime", "تغییر زاویه دید": "multiple-angles", "اصلاح نور و سایه": "light-restoration", "نورپردازی مجدد (Relight)": "relight", "نورپردازی چند زاویه‌ای": "multi-angle-lighting", "روتوش پوست": "edit-skin", "صحنه بعدی (سینمایی)": "next-scene", "افزایش کیفیت (Upscale)": "upscale-image"
+    "تبدیل عکس به انیمه": "anime",
+    "تغییر زاویه دید": "multiple-angles",
+    "اصلاح نور و سایه": "light-restoration",
+    "نورپردازی مجدد (Relight)": "relight",
+    "نورپردازی چند زاویه‌ای": "multi-angle-lighting",
+    "روتوش پوست": "edit-skin",
+    "صحنه بعدی (سینمایی)": "next-scene",
+    "افزایش کیفیت (Upscale)": "upscale-image"
 }
+
 ASPECT_RATIOS_LIST = [
-    "خودکار (پیش‌فرض)", "۱:۱ (مربع - 1024x1024)", "۱۶:۹ (افقی - 1344x768)", "۹:۱۶ (عمودی - 768x1344)", "شخصی‌سازی (Custom)"
+    "خودکار (پیش‌فرض)",
+    "۱:۱ (مربع - 1024x1024)",
+    "۱۶:۹ (افقی - 1344x768)",
+    "۹:۱۶ (عمودی - 768x1344)",
+    "شخصی‌سازی (Custom)"
 ]
+
 ASPECT_RATIOS_MAP = {
-    "خودکار (پیش‌فرض)": "Auto", "۱:۱ (مربع - 1024x1024)": (1024, 1024), "۱۶:۹ (افقی - 1344x768)": (1344, 768), "۹:۱۶ (عمودی - 768x1344)": (768, 1344), "شخصی‌سازی (Custom)": "Custom"
+    "خودکار (پیش‌فرض)": "Auto",
+    "۱:۱ (مربع - 1024x1024)": (1024, 1024),
+    "۱۶:۹ (افقی - 1344x768)": (1344, 768),
+    "۹:۱۶ (عمودی - 768x1344)": (768, 1344),
+    "شخصی‌سازی (Custom)": "Custom"
 }
 
+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+"
+]
+
+def check_text_safety(text):
+    text_lower = text.lower()
+    for word in BANNED_WORDS:
+        if f" {word} " in f" {text_lower} ": 
+            return False
+    return True
+
+def translate_prompt(text):
+    if not text:
+        return ""
+    try:
+        translated = GoogleTranslator(source='auto', target='en').translate(text)
+        return translated
+    except Exception as e:
+        print(f"Translation Error: {e}")
+        return text
+
 def update_dimensions_on_upload(image):
-    if image is None: return 1024, 1024
-    w, h = image.size
-    new_w, new_h = (1024, int(1024 * h / w)) if w > h else (int(1024 * w / h), 1024)
-    return (new_w // 8) * 8, (new_h // 8) * 8
+    if image is None:
+        return 1024, 1024
+    original_width, original_height = image.size
+    if original_width > original_height:
+        new_width = 1024
+        aspect_ratio = original_height / original_width
+        new_height = int(new_width * aspect_ratio)
+    else:
+        new_height = 1024
+        aspect_ratio = original_width / original_height
+        new_width = int(new_height * aspect_ratio)
+    new_width = (new_width // 8) * 8
+    new_height = (new_height // 8) * 8
+    return new_width, new_height
+
+def update_sliders_visibility(choice):
+    if choice == "شخصی‌سازی (Custom)":
+        return gr.update(visible=True), gr.update(visible=True)
+    else:
+        return gr.update(visible=False), gr.update(visible=False)
 
 def get_error_html(message):
-    return f"""<div style="background-color: #fee2e2; border: 1px solid #ef4444; color: #b91c1c; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;"><span style="font-size: 1.2em;">⛔</span> {message}</div>"""
+    return f"""
+    <div style="background-color: #fee2e2; border: 1px solid #ef4444; color: #b91c1c; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;">
+        <span style="font-size: 1.2em;">⛔</span>
+        {message}
+    </div>
+    """
 
 def get_success_html(message):
-    return f"""<div style="background-color: #dcfce7; border: 1px solid #22c55e; color: #15803d; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;"><span style="font-size: 1.2em;">✅</span> {message}</div>"""
+    return f"""
+    <div style="background-color: #dcfce7; border: 1px solid #22c55e; color: #15803d; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;">
+        <span style="font-size: 1.2em;">✅</span>
+        {message}
+    </div>
+    """
+
+def check_and_update_quota(request: gr.Request, is_vip: bool):
+    """
+    بررسی محدودیت کاربر و بروزرسانی شمارنده.
+    اگر VIP باشد، محدودیت ندارد.
+    اگر رایگان باشد، بر اساس IP محدودیت ۵ بار در روز اعمال می‌شود.
+    """
+    if is_vip:
+        return True, ""
+        
+    try:
+        client_ip = request.client.host
+    except:
+        client_ip = "unknown"
+        
+    today = datetime.now().strftime("%Y-%m-%d")
+    
+    # مقداردهی اولیه برای IP جدید
+    if client_ip not in USAGE_TRACKER:
+        USAGE_TRACKER[client_ip] = {'count': 0, 'date': today}
+    
+    user_data = USAGE_TRACKER[client_ip]
+    
+    # ریست کردن اگر روز تغییر کرده باشد
+    if user_data['date'] != today:
+        user_data['count'] = 0
+        user_data['date'] = today
+        
+    # بررسی سقف مجاز
+    if user_data['count'] >= DAILY_LIMIT:
+        return False, f"شما به سقف مجاز روزانه ({DAILY_LIMIT} تصویر) رسیده‌اید. فردا مجدداً تلاش کنید یا حساب خود را ارتقا دهید."
+        
+    return True, ""
+
+def increment_quota(request: gr.Request, is_vip: bool):
+    """
+    افزایش شمارنده پس از موفقیت آمیز بودن عملیات
+    """
+    if is_vip: return
+    try:
+        client_ip = request.client.host
+        if client_ip in USAGE_TRACKER:
+            USAGE_TRACKER[client_ip]['count'] += 1
+    except:
+        pass
 
-@spaces.GPU(duration=45)
+@spaces.GPU(duration=30)
 def infer(
-    input_image, prompt, lora_adapter_persian, seed, randomize_seed,
-    guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height,
-    is_paid_user, user_fingerprint,  # ورودی‌های جدید برای مدیریت اعتبار
+    input_image,
+    prompt,
+    lora_adapter_persian,
+    seed,
+    randomize_seed,
+    guidance_scale,
+    steps,
+    aspect_ratio_selection,
+    custom_width,
+    custom_height,
+    is_vip_state, # ورودی جدید برای وضعیت VIP
+    request: gr.Request,
     progress=gr.Progress(track_tqdm=True)
 ):
-    # --- بخش جدید: بررسی اعتبار قبل از هر چیز ---
-    if not is_paid_user:
-        can_generate = check_and_use_credit(user_fingerprint)
-        if not can_generate:
-            return None, seed, get_error_html("اعتبار رایگان روزانه شما تمام شده است. اعتبار فردا مجدداً شارژ خواهد شد.")
-
-    if input_image is None: return None, seed, get_error_html("لط��اً ابتدا یک تصویر بارگذاری کنید.")
-    if is_image_nsfw(input_image): return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است.")
-    
+    # --- بررسی سهمیه (Quota) ---
+    allowed, message = check_and_update_quota(request, is_vip_state)
+    if not allowed:
+        return None, seed, get_error_html(message)
+
+    if input_image is None:
+        return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
+
+    if is_image_nsfw(input_image):
+        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
+
     english_prompt = translate_prompt(prompt)
-    if not check_text_safety(english_prompt): return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز است.")
+    if not check_text_safety(english_prompt):
+        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است.")
 
     adapter_internal_name = LORA_MAPPING.get(lora_adapter_persian)
-    if adapter_internal_name: pipe.set_adapters([adapter_internal_name], adapter_weights=[1.0])
+    if adapter_internal_name:
+        pipe.set_adapters([adapter_internal_name], adapter_weights=[1.0])
     
-    if randomize_seed: seed = random.randint(0, MAX_SEED)
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+
     generator = torch.Generator(device=device).manual_seed(seed)
     
-    safety_negative = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, areola, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, open chest, revealing clothes, cleavage, see-through, sheer, gore, violence, blood, navel, midriff, exposed skin, erotic, ecchi, hentai, uncensored, stripping"
-    final_negative_prompt = f"{safety_negative}, worst quality, low quality"
+    safety_negative = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood"
+    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}"
 
     original_image = input_image.convert("RGB")
+    
     selection_value = ASPECT_RATIOS_MAP.get(aspect_ratio_selection)
     
     if selection_value == "Custom":
-        width, height = (int(custom_width) // 8) * 8, (int(custom_height) // 8) * 8
-    elif selection_value == "Auto":
+        width = (int(custom_width) // 8) * 8
+        height = (int(custom_height) // 8) * 8
+    elif selection_value == "Auto" or selection_value is None:
         width, height = update_dimensions_on_upload(original_image)
     else:
         width, height = selection_value
 
     try:
-        result = pipe(image=original_image, prompt=english_prompt, negative_prompt=final_negative_prompt, height=height, width=width, num_inference_steps=steps, generator=generator, true_cfg_scale=guidance_scale).images[0]
-        if is_image_nsfw(result): return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+        result = pipe(
+            image=original_image,
+            prompt=english_prompt,
+            negative_prompt=final_negative_prompt,
+            height=height,
+            width=width,
+            num_inference_steps=steps,
+            generator=generator,
+            true_cfg_scale=guidance_scale,
+        ).images[0]
+
+        if is_image_nsfw(result):
+            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+        
+        # افزایش شمارنده مصرف فقط در صورت موفقیت
+        increment_quota(request, is_vip_state)
+        
         return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
+
     except Exception as e:
         error_str = str(e)
-        if "quota" in error_str.lower() or "exceeded" in error_str.lower(): raise e 
+        if "quota" in error_str.lower() or "exceeded" in error_str.lower():
+             raise e 
         return None, seed, get_error_html(f"خطا در پردازش: {error_str}")
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    # مثال‌ها اعتبار کم نمی‌کنند
-    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, True, "example_user")
+    # برای مثال‌ها محدودیت VIP بررسی نمی‌شود و IP لوکال فرض می‌شود
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, True, None)
     return res, s, status
 
-# --- جاوااسکریپت برای دا��لود (بدون تغییر) ---
-js_download_func = "async(e)=>{if(!e){return void alert('لطفاً ابتدا تصویر را تولید کنید.')};let o=e.url;o&&!o.startsWith('http')?o=window.location.origin+o:!o&&e.path&&(o=window.location.origin+'/file='+e.path),window.parent.postMessage({type:'DOWNLOAD_REQUEST',url:o},'*')}"
+# --- جاوااسکریپت برای دکمه دانلود ---
+js_download_func = """
+async (image) => {
+    if (!image) {
+        alert("لطفاً ابتدا تصویر را تولید کنید.");
+        return;
+    }
+    let fileUrl = image.url;
+    if (fileUrl && !fileUrl.startsWith('http')) {
+       fileUrl = window.location.origin + fileUrl;
+    } else if (!fileUrl && image.path) {
+       fileUrl = window.location.origin + "/file=" + image.path;
+    }
+    window.parent.postMessage({
+        type: 'DOWNLOAD_REQUEST',
+        url: fileUrl
+    }, '*');
+}
+"""
 
-# --- جاوااسکریپت و CSS اصلی (با تغییرات زیاد) ---
-js_and_css_code = """
+# --- جاوااسکریپت سراسری (استراتژی موبایل + بستن خودکار ۱۰ ثانیه + تشخیص VIP) ---
+js_global_content = """
 <script>
-// --- START OF JS LOGIC FOR USER AUTH & CREDITS ---
-document.addEventListener('DOMContentLoaded', () => {
-    
-    // --- بخش جدید: مدیریت وضعیت کاربر و اعتبار ---
-    const PREMIUM_PAGE_ID = '1149636'; // شناسه صفحه کاربران ویژه
-    const PREMIUM_URL = '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991';
-
-    let userFingerprint = null;
-    let isPaidUser = false;
-    let countdownInterval = null;
-
-    // توابع UI برای نمایش وضعیت
-    const statusContainer = document.getElementById('user-status-section');
-    const badgeEl = document.getElementById('subscription-badge');
-    const creditTextEl = document.getElementById('credit-status-text');
-    const countdownEl = document.getElementById('countdown-timer');
-    const upgradeBtn = document.getElementById('upgrade-btn');
-    const runBtn = document.getElementById('run-btn');
-
-    // تابع اصلی برای آپدیت UI بر اساس وضعیت کاربر
-    function updateUIForUserStatus(statusData) {
-        statusContainer.style.display = 'block';
-        if (countdownInterval) clearInterval(countdownInterval);
-
-        if (isPaidUser) {
-            badgeEl.textContent = 'کاربر ویژه';
-            badgeEl.className = 'paid-badge';
-            creditTextEl.innerHTML = 'دسترسی شما <strong>نامحدود</strong> است.';
-            countdownEl.style.display = 'none';
-            upgradeBtn.style.display = 'none';
-            runBtn.disabled = false;
-        } else {
-            badgeEl.textContent = 'کاربر رایگان';
-            badgeEl.className = 'free-badge';
-            const credits = statusData.credits || 0;
-            
-            if (credits > 0) {
-                creditTextEl.innerHTML = `شما <strong>${credits} اعتبار</strong> ساخت تصویر برای امروز دارید.`;
-                runBtn.disabled = false;
-                upgradeBtn.style.display = 'none';
-            } else {
-                creditTextEl.innerHTML = 'اعتبار رایگان امروز شما تمام شده است.';
-                runBtn.disabled = true;
-                upgradeBtn.style.display = 'block';
+// Global variable to store VIP status
+window.userIsVip = false;
+const PREMIUM_PAGE_ID = '1149635'; // شناسه صفحه پرداخت برای بررسی دسترسی
+
+// Request User Status on Load
+setTimeout(() => {
+    window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
+}, 1000);
+
+// Listener for User Status
+window.addEventListener('message', (event) => {
+    if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
+        try {
+            const user = JSON.parse(event.data.payload);
+            // بررسی دسترسی کاربر به صفحه پرمیوم
+            if (user && user.isLogin && user.accessible_pages &&
+               (user.accessible_pages.includes(PREMIUM_PAGE_ID) || 
+                user.accessible_pages.includes(parseInt(PREMIUM_PAGE_ID)))) {
+                
+                window.userIsVip = true;
+                
+                // نمایش نشان VIP
+                const badge = document.createElement('div');
+                badge.innerHTML = '👑 نسخه ویژه (نامحدود)';
+                badge.style.cssText = 'position:fixed;top:10px;left:10px;background:linear-gradient(45deg, #FFD700, #FFC107);color:#333;padding:6px 12px;border-radius:20px;z-index:9999;font-weight:bold;box-shadow:0 2px 5px rgba(0,0,0,0.2);font-family:"Vazirmatn",sans-serif;font-size:0.8rem;';
+                document.body.appendChild(badge);
+                
+                // حذف مدال محدودیت اگر باز باشد
+                window.closeErrorModal();
             }
-            
-            // شروع شمارش معکوس
-            const resetTimestamp = statusData.next_reset_timestamp;
-            const updateTimer = () => {
-                const now = Date.now() / 1000;
-                const timeLeft = Math.max(0, resetTimestamp - now);
-                if (timeLeft === 0) {
-                    clearInterval(countdownInterval);
-                    initializeApp(); // برای گرفتن اعتبار جدید
-                    return;
-                }
-                const h = Math.floor(timeLeft / 3600);
-                const m = Math.floor((timeLeft % 3600) / 60);
-                const s = Math.floor(timeLeft % 60);
-                countdownEl.querySelector('span').textContent = `${h.toString().padStart(2, '0')}:${m.toString().padStart(2, '0')}:${s.toString().padStart(2, '0')}`;
-            };
-            updateTimer();
-            countdownInterval = setInterval(updateTimer, 1000);
-            countdownEl.style.display = 'block';
-        }
+        } catch (e) { console.log('Error parsing user status:', e); }
     }
+});
 
-    // ارتباط با برنامه اصلی (Parent)
-    window.addEventListener('message', (event) => {
-        if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
-            try {
-                const userObject = JSON.parse(event.data.payload || '{}');
-                isPaidUser = userObject && userObject.isLogin && userObject.accessible_pages &&
-                    (userObject.accessible_pages.includes(PREMIUM_PAGE_ID) || userObject.accessible_pages.includes(parseInt(PREMIUM_PAGE_ID)));
-                
-                // آپدیت کردن کامپوننت‌های مخفی گرادیو
-                const fingerprintInput = document.querySelector('#fingerprint-input-for-backend textarea');
-                const isPaidInput = document.querySelector('#is-paid-input-for-backend textarea');
-                if(fingerprintInput) fingerprintInput.value = userFingerprint;
-                if(isPaidInput) isPaidInput.value = isPaidUser ? 'true' : 'false';
-
-                // ایجاد یک Event برای اینکه گرادیو تغییر را تشخیص دهد
-                const eventToDispatch = new Event('input', { bubbles: true });
-                if(fingerprintInput) fingerprintInput.dispatchEvent(eventToDispatch);
-                if(isPaidInput) isPaidInput.dispatchEvent(eventToDispatch);
-
-                // حالا که اطلاعات کاربر را داریم، از بک‌اند گرادیو وضعیت اعتبار را می‌پرسیم
-                // با کلیک کردن روی دکمه مخفی، تابع پایتون را فراخوانی می‌کنیم
-                const statusCheckBtn = document.getElementById('status-check-btn-hidden');
-                if(statusCheckBtn) statusCheckBtn.click();
-
-            } catch (e) {
-                console.error("Error processing user status:", e);
-                isPaidUser = false; // در صورت خطا، کاربر رایگان در نظر گرفته می‌شود
-            }
-        }
-        
-        if (event.data.type === 'DOWNLOAD_REQUEST' && event.data.url) {
-             // این بخش برای دکمه دانلود است که در HTML دیگر تعریف شده
-        }
-        
-        if (event.data.type === 'NAVIGATE_TO_URL' && event.data.url) {
-            window.top.location.href = event.data.url;
-        }
-    });
-
-    // تابع برای ساخت اثرانگشت مرورگر
-    async function getBrowserFingerprint() {
-        const components = [navigator.userAgent, navigator.language, screen.width+'x'+screen.height, new Date().getTimezoneOffset()];
-        const fingerprintString = components.join('~~~');
-        let hash = 0;
-        for (let i = 0; i < fingerprintString.length; i++) {
-            const char = fingerprintString.charCodeAt(i);
-            hash = ((hash << 5) - hash) + char;
-            hash |= 0;
+document.addEventListener('DOMContentLoaded', () => {
+    // 1. Force Light Mode
+    const forceLight = () => {
+        const body = document.querySelector('body');
+        if (body) {
+            body.classList.remove('dark');
+            body.style.backgroundColor = '#f5f7fa';
+            body.style.color = '#333333';
         }
-        return 'fp_' + Math.abs(hash).toString(16);
-    }
-    
-    // تابع اصلی برای شروع
-    async function initializeApp() {
-        userFingerprint = await getBrowserFingerprint();
-        // درخواست اطلاعات کاربر از برنامه اصلی
-        parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
-    }
+        document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
+    };
+    forceLight();
+    setInterval(forceLight, 1000);
+
+    // 2. RETRY FUNCTION
+    window.retryGeneration = function() {
+        const modal = document.getElementById('custom-quota-modal');
+        if (modal) modal.remove();
+
+        const runBtn = document.getElementById('run-btn');
+        if(runBtn) runBtn.click();
+    };
     
-    // اتصال دکمه ارتقا
-    if(upgradeBtn) {
-        upgradeBtn.addEventListener('click', () => {
-            parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
-        });
-    }
+    // Close function
+    window.closeErrorModal = function() {
+         const modal = document.getElementById('custom-quota-modal');
+         if (modal) modal.remove();
+    };
 
-    // --- مدیریت خطای Quota (بدون تغییر) ---
+    // 3. SHOW MODAL FUNCTION
     const showQuotaModal = () => {
         if (document.getElementById('custom-quota-modal')) return;
-        const modalHtml = `<div id="custom-quota-modal" style="position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.6); backdrop-filter: blur(5px); z-index: 99999; display: flex; align-items: center; justify-content: center; font-family: 'Vazirmatn', sans-serif;"><div class="ip-reset-guide-container"><div class="guide-header"><h2>یک قدم تا ساخت تصاویر جدید</h2><p>نیازمند تغییر نقطه دستیابی</p></div><div class="guide-content"><p>برای استفاده نامحدود از هوش مصنوعی، لطفاً نقطه اتصال اینترنت (IP) خود را تغییر دهید (مثلاً با خاموش/روشن کردن حالت پرواز) و دوباره تلاش کنید.</p><button onclick="parent.postMessage({ type: 'NAVIGATE_TO_URL', url: '#/nav/online/news/getSingle/1149635/...'}, '*')">دیدن ویدیو آموزشی</button></div><div class="guide-actions"><button class="action-button back-button" onclick="this.closest('#custom-quota-modal').remove()">بازگشت</button><button class="action-button retry-button" onclick="document.getElementById('run-btn').click(); this.closest('#custom-quota-modal').remove();">تلاش مجدد</button></div></div></div>`;
+
+        const modalHtml = `
+            <div id="custom-quota-modal" style="position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.6); backdrop-filter: blur(5px); z-index: 99999; display: flex; align-items: center; justify-content: center; font-family: 'Vazirmatn', sans-serif;">
+                <div class="ip-reset-guide-container">
+                    <div class="guide-header">
+                        <svg class="guide-header-icon" viewbox="0 0 100 100" xmlns="http://www.w3.org/2000/svg"> 
+                            <defs><lineargradient id="grad1" x1="0%" y1="0%" x2="100%" y2="100%"><stop offset="0%" style="stop-color: #667eea; stop-opacity: 1;"></stop><stop offset="100%" style="stop-color: #764ba2; stop-opacity: 1;"></stop></lineargradient></defs> 
+                            <circle cx="50" cy="50" r="45" fill="url(#grad1)" opacity="0.1"></circle>
+                            <circle cx="50" cy="50" r="35" fill="none" stroke="url(#grad1)" stroke-width="2" opacity="0.3"></circle>
+                            <path d="M35 50 L45 60 L65 40" stroke="url(#grad1)" stroke-width="4" fill="none" stroke-linecap="round" stroke-linejoin="round"></path>
+                            <circle cx="65" cy="35" r="8" fill="#fee140"></circle>
+                            <path d="M62 35 L68 35 M65 32 L65 38" stroke="white" stroke-width="2" stroke-linecap="round"></path> 
+                        </svg>
+                        <div>
+                            <h2>محدودیت سرور گرافیکی</h2>
+                            <p>نیازمند تغییر نقطه دستیابی</p>
+                        </div>
+                    </div>
+                    
+                    <div class="guide-content">
+                        <div class="info-card">
+                            <div class="info-card-header">
+                                <svg class="info-card-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z" fill="#667eea" opacity="0.2"></path><path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z" stroke="#667eea" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
+                                <span class="info-card-title">راه حل سریع</span>
+                            </div>
+                            <p>طبق ویدیو آموزشی پایین بین نقطه دستیابی جابجا شوید تلاش مجدد بزنید تا تصاویر مجدداً تولید بشه.</p>
+                        </div>
+                        
+                        <div class="summary-section">
+                            <div class="summary-header">
+                                <svg class="summary-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><circle cx="12" cy="12" r="10" fill="#56ab2f" opacity="0.2"></circle><path d="M12 2C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2z" stroke="#56ab2f" stroke-width="2"></path><path d="M9 12l2 2 4-4" stroke="#56ab2f" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
+                                <span class="summary-title">خلاصه راهنما</span>
+                            </div>
+                            <div class="summary-text">این پیام مربوط به سرور جهانی است. برای رفع آن: اینترنت سیم‌کارت استفاده کنید، VPN را خاموش کرده و طبق ویدیو آموزشی نقطه دستیابی رو تغییر دهید.</div>
+                        </div>
+                        
+                        <div class="video-button-container">
+                            <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_URL', url: '#/nav/online/news/getSingle/1149635/eyJpdiI6IjhHVGhPQWJwb3E0cjRXbnFWTW5BaUE9PSIsInZhbHVlIjoiS1V0dTdvT21wbXAwSXZaK1RCTG1pVXZqdlFJa1hXV1RKa2FLem9zU3pXMjd5MmlVOGc2YWY0NVdNR3h3Smp1aSIsIm1hYyI6IjY1NTA5ZDYzMjAzMTJhMGQyMWQ4NjA4ZDgyNGZjZDVlY2MyNjdiMjA2NWYzOWRjY2M4ZmVjYWRlMWNlMWQ3ODEiLCJ0YWciOiIifQ==/21135210' }, '*')" class="elegant-video-button"> 
+                                <svg class="elegant-video-button-icon" xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M8 5v14l11-7z"></path></svg> 
+                                <span>دیدن ویدیو آموزشی استفاده نامحدود</span> 
+                            </button>
+                        </div>
+                    </div>
+                    
+                    <div class="guide-actions">
+                        <button class="action-button back-button" onclick="window.closeErrorModal()"> 
+                            <svg class="action-button-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M19 12H5M12 19l-7-7 7-7" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
+                            <span>بازگشت</span> 
+                        </button>
+                        <button class="action-button retry-button" onclick="window.retryGeneration()"> 
+                            <svg class="action-button-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M23 4v6h-6M1 20v-6h6" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path><path d="M20.49 9A9 9 0 0 0 5.64 5.64L1 10m22 4l-4.64 4.36A9 9 0 0 1 3.51 15" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
+                            <span>تلاش مجدد</span> 
+                        </button>
+                    </div>
+                </div>
+            </div>
+        `;
+        
         document.body.insertAdjacentHTML('beforeend', modalHtml);
+
+        // Auto close after 10 seconds
+        setTimeout(() => {
+            window.closeErrorModal();
+        }, 10000);
     };
+
+    // 4. SCANNER (For HuggingFace GPU Quota)
     setInterval(() => {
-        document.querySelectorAll('.toast-body, .error').forEach(el => {
-            if ((el.innerText || "").toLowerCase().includes('quota')) {
-                showQuotaModal();
-                el.style.display = 'none';
+        const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar, div[class*="error"]');
+        
+        potentialErrors.forEach(el => {
+            const text = el.innerText || "";
+            if (text.toLowerCase().includes('quota') || text.toLowerCase().includes('exceeded')) {
+                 
+                 showQuotaModal();
+                 
+                 // Immediately hide the Gradio error
+                 el.style.display = 'none';
+                 el.style.opacity = '0';
+                 el.innerText = '';
+                 
+                 const parentWrap = el.closest('.toast-wrap');
+                 if(parentWrap) parentWrap.style.display = 'none';
             }
         });
     }, 100);
-
-    initializeApp();
-
-    // این تابع مخصوص گرادیو است تا وقتی خروجی JSON آپدیت شد، UI را آپدیت کند
-    window.updateUIFromGradio = function(jsonData) {
-        updateUIForUserStatus(jsonData);
-    }
 });
 </script>
+"""
+
+# --- CSS Updated (Larger & Auto Scroll) ---
+css_code = """
 <style>
-/* --- استایل‌های اصلی (مشابه قبل با اضافات جدید) --- */
 @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
-body, .gradio-container { font-family: 'Vazirmatn', sans-serif !important; background-color: #f5f7fa !important; color-scheme: light !important; }
-#col-container { direction: rtl; text-align: right; max-width: 980px; margin: 0 auto; padding: 30px; background: #ffffff; border-radius: 24px; box-shadow: 0 10px 40px -10px rgba(0,0,0,0.08); }
-#main-title h1 { text-align: center; font-size: 2.4em !important; font-weight: 800; background: -webkit-linear-gradient(45deg, #2563eb, #1e40af); -webkit-background-clip: text; -webkit-text-fill-color: transparent; }
-#main-description { text-align: center; font-size: 1.15em; color: #4b5563 !important; margin-bottom: 25px; line-height: 1.6; }
-footer, .flagging { display: none !important; }
-
-/* --- استایل‌های جدید برای بخش وضعیت کاربر --- */
-#user-status-section {
-    display: none; /* Initially hidden */
+
+:root, .dark, body, .gradio-container {
+    --body-background-fill: #f5f7fa !important;
+    --body-text-color: #1f2937 !important;
+    --background-fill-primary: #ffffff !important;
+    --background-fill-secondary: #f3f4f6 !important;
+    --border-color-primary: #e5e7eb !important;
+    --block-background-fill: #ffffff !important;
+    --block-label-text-color: #374151 !important;
+    --block-title-text-color: #111827 !important;
+    --input-background-fill: #ffffff !important;
+    color-scheme: light !important;
+}
+
+/* --- IP Reset Guide CSS --- */
+:root {
+    --guide-bg: rgba(255, 255, 255, 0.98);
+    --guide-border: rgba(102, 126, 234, 0.2);
+    --guide-text-title: #2d3748;
+    --guide-text-body: #4a5568;
+    --guide-accent: #667eea;
+    --primary-gradient-guide: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    --success-gradient-guide: linear-gradient(135deg, #56ab2f 0%, #a8e063 100%);
+    --radius-md-guide: 12px;
+    --radius-lg-guide: 16px;
+    --shadow-sm: 0 1px 2px 0 rgba(26, 32, 44, 0.03);
+    --shadow-md: 0 4px 6px -1px rgba(26, 32, 44, 0.05), 0 2px 4px -2px rgba(26, 32, 44, 0.04);
+    --shadow-xl: 0 20px 25px -5px rgba(26, 32, 44, 0.07), 0 8px 10px -6px rgba(26, 32, 44, 0.05);
+}
+
+@keyframes float {
+    0%, 100% { transform: translateY(0px); }
+    50% { transform: translateY(-10px); }
+}
+@keyframes slideInUp {
+    from { opacity: 0; transform: translateY(30px); }
+    to { opacity: 1; transform: translateY(0); }
+}
+
+.ip-reset-guide-container {
+    text-align: right; 
+    direction: rtl; 
+    background: var(--guide-bg); 
+    backdrop-filter: blur(10px);
+    padding: 20px; /* Increased Padding */
+    border-radius: var(--radius-lg-guide);
+    box-shadow: var(--shadow-xl); 
+    border: 1px solid var(--guide-border); 
+    animation: slideInUp 0.6s cubic-bezier(0.4, 0, 0.2, 1) both;
+    width: 90%; 
+    max-width: 420px; /* Increased width slightly */
+    max-height: 90vh; /* Prevent overflow on small screens */
+    overflow-y: auto; /* Enable scroll if needed */
+    position: relative; 
+    box-sizing: border-box;
+    font-family: 'Vazirmatn', sans-serif !important;
+}
+.ip-reset-guide-container::before {
+    content: ''; position: absolute; top: 0; left: 0; right: 0; height: 4px; background: var(--primary-gradient-guide);
+}
+.guide-header { display: flex; align-items: center; margin-bottom: 15px; }
+.guide-header-icon { width: 45px; height: 45px; margin-left: 15px; animation: float 3s ease-in-out infinite; flex-shrink: 0; }
+.guide-header h2 { font-size: 1.2rem; color: var(--guide-text-title); font-weight: 700; margin: 0; }
+.guide-header p { color: var(--guide-text-body); font-size: 0.8rem; margin-top: 3px; margin-bottom: 0; }
+
+.guide-content { font-size: 0.9rem; color: var(--guide-text-body); line-height: 1.6; }
+
+.info-card { background: linear-gradient(135deg, #667eea15 0%, #764ba215 100%); border: 1px solid rgba(102, 126, 234, 0.2); border-radius: var(--radius-md-guide); padding: 12px; margin: 12px 0; position: relative; overflow: hidden; }
+.info-card p { font-size: 0.85rem; line-height: 1.6; margin: 0; }
+.info-card::before { content: ''; position: absolute; top: 0; right: 0; width: 3px; height: 100%; background: var(--primary-gradient-guide); }
+.info-card-header { display: flex; align-items: center; margin-bottom: 8px; }
+.info-card-icon { width: 18px; height: 18px; margin-left: 8px; }
+.info-card-title { font-weight: 600; color: var(--guide-text-title); font-size: 0.95rem; }
+
+.summary-section { margin-top: 12px; padding: 12px; border-radius: var(--radius-md-guide); background: linear-gradient(135deg, #56ab2f15 0%, #a8e06315 100%); border: 1px solid rgba(86, 171, 47, 0.2); position: relative; overflow: hidden; }
+.summary-section::before { content: ''; position: absolute; top: 0; right: 0; width: 3px; height: 100%; background: var(--success-gradient-guide); }
+.summary-header { display: flex; align-items: center; margin-bottom: 8px; }
+.summary-icon { width: 18px; height: 18px; margin-left: 8px; }
+.summary-title { font-weight: 600; color: #2f5a33; font-size: 0.95rem; }
+.summary-text { color: #2f5a33; font-size: 0.85rem; line-height: 1.6; }
+
+/* Tutorial Button */
+.video-button-container { text-align: center; margin: 20px 0 15px 0; width: 100%; }
+.elegant-video-button { 
+    display: inline-flex !important; 
+    align-items: center; 
+    justify-content: center; 
+    padding: 10px 20px !important; 
+    background-color: #fff !important; 
+    color: var(--guide-accent) !important; 
+    border: 1px solid #e2e8f0 !important; 
+    text-decoration: none; 
+    border-radius: 50px !important; 
+    font-weight: 600 !important; 
+    font-size: 0.9rem !important; 
+    cursor: pointer !important; 
+    font-family: inherit; 
+    transition: all 0.3s ease !important; 
+    box-shadow: 0 2px 10px rgba(0,0,0,0.05) !important; 
+    width: auto !important;
+}
+.elegant-video-button:hover { 
+    background: var(--primary-gradient-guide) !important; 
+    color: white !important; 
+    border-color: transparent !important; 
+    transform: translateY(-2px); 
+    box-shadow: 0 6px 16px rgba(102, 126, 234, 0.3) !important; 
+}
+.elegant-video-button-icon { width: 18px; height: 18px; margin-left: 8px; fill: currentColor; }
+
+/* Action Buttons */
+.guide-actions { 
+    display: flex !important; 
+    gap: 12px !important; 
+    margin-top: 20px; 
+    padding-top: 20px; 
+    border-top: 1px solid #e2e8f0; 
+    width: 100% !important;
+}
+.action-button { 
+    padding: 12px 15px !important; 
+    border: none !important; 
+    border-radius: 12px !important; 
+    font-size: 0.95rem !important; 
+    font-weight: 600 !important; 
+    cursor: pointer !important; 
+    flex: 1 !important; 
+    transition: all 0.3s ease !important; 
+    display: flex !important; 
+    align-items: center; 
+    justify-content: center; 
+    font-family: inherit; 
+    height: 48px !important;
+}
+.action-button-icon { width: 20px; height: 20px; margin-right: 0; margin-left: 8px; }
+
+.back-button { 
+    background: white !important; 
+    color: var(--guide-text-body) !important; 
+    border: 2px solid #e2e8f0 !important; 
+}
+.back-button:hover { 
+    background: #f7fafc !important; 
+    border-color: var(--guide-accent) !important; 
+    transform: translateY(-2px); 
+    box-shadow: var(--shadow-md) !important; 
+}
+
+.retry-button { 
+    background: var(--primary-gradient-guide) !important; 
+    color: white !important; 
+    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.3) !important; 
+}
+.retry-button:hover { 
+    transform: translateY(-2px); 
+    box-shadow: 0 6px 20px rgba(102, 126, 234, 0.4) !important; 
+}
+
+/* --- Main App CSS --- */
+body {
+    font-family: 'Vazirmatn', sans-serif !important;
+    background-color: #f5f7fa !important;
+    margin: 0;
+    padding: 10px;
+}
+
+#col-container {
+    margin: 0 auto;
+    max-width: 980px;
+    direction: rtl;
+    text-align: right;
+    padding: 30px;
+    background: #ffffff !important;
+    border-radius: 24px;
+    box-shadow: 0 10px 40px -10px rgba(0,0,0,0.08); 
+    border: 1px solid rgba(255,255,255,0.8);
+}
+
+#main-title h1 {
+    font-size: 2.4em !important;
+    text-align: center;
+    color: #1a202c !important;
+    margin-bottom: 15px;
+    font-weight: 800;
+    background: -webkit-linear-gradient(45deg, #2563eb, #1e40af);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+}
+
+#main-description {
     text-align: center;
-    margin-bottom: 25px;
-    padding: 15px;
-    background-color: #f8f9fa;
-    border-radius: 14px;
-    border: 1px solid #e9ecef;
-}
-#subscription-badge {
-    display: inline-block;
-    padding: 6px 16px;
-    border-radius: 20px;
-    font-size: 0.9em;
-    font-weight: 700;
-    margin-bottom: 10px;
-}
-#subscription-badge.free-badge { background: linear-gradient(45deg, #6c757d, #495057); color: white; }
-#subscription-badge.paid-badge { background: linear-gradient(45deg, #198754, #157347); color: white; }
-#credit-status-text {
-    font-size: 1em;
-    color: #495057;
-    margin-bottom: 10px;
-}
-#credit-status-text strong { color: #0d6efd; }
-#countdown-timer {
-    font-size: 0.9em;
-    color: #6c757d;
-}
-#countdown-timer span { font-weight: bold; font-variant-numeric: tabular-nums; }
-#upgrade-btn {
-    display: none; /* Initially hidden */
+    font-size: 1.15em;
+    color: #4b5563 !important;
+    margin-bottom: 40px;
+    line-height: 1.6;
+}
+
+.gr-input-label, span.label-wrap, label span {
+    font-weight: 700 !important;
+    color: #374151 !important;
+    font-size: 0.95em !important;
+    margin-bottom: 8px !important;
+}
+
+textarea, input[type="text"] {
+    border: 2px solid #e2e8f0 !important;
+    border-radius: 12px !important;
+    background-color: #ffffff !important;
+    color: #111827 !important;
+    padding: 12px !important;
+    transition: all 0.3s ease;
+    font-family: 'Vazirmatn', sans-serif !important;
+}
+
+textarea:focus, input[type="text"]:focus {
+    border-color: #3b82f6 !important;
+    box-shadow: 0 0 0 4px rgba(59, 130, 246, 0.1) !important;
+    outline: none;
+}
+
+.gr-dropdown {
+    background: #ffffff !important;
+    border-radius: 12px !important;
+}
+
+.primary-btn, button.primary {
+    background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important;
+    border: none !important;
+    color: white !important;
+    font-weight: 700 !important;
+    font-size: 1.1em !important;
+    padding: 14px 28px !important;
+    border-radius: 14px !important;
+    box-shadow: 0 4px 15px rgba(16, 185, 129, 0.3) !important;
+    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
+    cursor: pointer !important;
+    width: 100%;
     margin-top: 15px;
-    padding: 10px 20px;
-    font-family: 'Vazirmatn', sans-serif;
-    font-weight: 700;
-    font-size: 1em;
-    color: white;
-    background: linear-gradient(135deg, #ffc107, #ff9800);
-    border: none;
-    border-radius: 12px;
-    cursor: pointer;
-    transition: all 0.2s ease;
-    box-shadow: 0 4px 10px rgba(255, 193, 7, 0.3);
-}
-#upgrade-btn:hover { transform: translateY(-2px); box-shadow: 0 6px 15px rgba(255, 193, 7, 0.4); }
-
-.primary-btn:disabled {
-    background: #adb5bd !important;
-    cursor: not-allowed;
+}
+
+.primary-btn:hover, button.primary:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 8px 25px rgba(16, 185, 129, 0.45) !important;
+}
+
+.primary-btn:active, button.primary:active {
+    transform: translateY(1px);
+}
+
+#download-btn {
+    background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
+    box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3) !important;
+}
+#download-btn:hover {
+    box-shadow: 0 8px 25px rgba(59, 130, 246, 0.45) !important;
+}
+
+.gradio-container .prose table, 
+.gradio-container table {
+    background-color: #ffffff !important;
+    color: #111827 !important;
+    border: 1px solid #e5e7eb !important;
+    border-radius: 12px !important;
+    overflow: hidden !important;
+    width: 100% !important;
+    margin-top: 20px !important;
+}
+
+.gradio-container thead th {
+    background-color: #f3f4f6 !important;
+    color: #374151 !important;
+    font-weight: 700 !important;
+    border-bottom: 2px solid #e5e7eb !important;
+    padding: 12px !important;
+    text-align: right !important;
+}
+
+.gradio-container tbody tr {
+    background-color: #ffffff !important;
+    border-bottom: 1px solid #f3f4f6 !important;
+}
+
+.gradio-container tbody tr:hover {
+    background-color: #f9fafb !important;
+}
+
+.gradio-container tbody td {
+    background-color: #ffffff !important;
+    color: #374151 !important;
+    padding: 10px !important;
+}
+
+.gradio-container tbody td span, 
+.gradio-container tbody td p {
+    color: #374151 !important;
+}
+
+footer { display: none !important; }
+.flagging { display: none !important; }
+
+/* Force toast transparency */
+.toast-body {
+    direction: rtl !important;
+    text-align: right !important;
+    background: transparent !important;
+    box-shadow: none !important;
+    border: none !important;
+    padding: 0 !important;
+    max-width: 100% !important;
+    width: auto !important;
+}
+.toast-wrap {
+    background: transparent !important;
+    border: none !important;
     box-shadow: none !important;
-    transform: none !important;
 }
 
-/* --- بقیه استایل‌ها (بدون تغییر) --- */
-.primary-btn { background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important; border:none !important; color:white !important; font-weight:700 !important; font-size:1.1em !important; padding:14px 28px !important; border-radius:14px !important; box-shadow:0 4px 15px rgba(16,185,129,0.3) !important; transition:all 0.3s ease !important; cursor:pointer !important; width:100%; margin-top:15px; }
-.primary-btn:hover { transform: translateY(-2px); box-shadow: 0 8px 25px rgba(16,185,129,0.45) !important; }
-.ip-reset-guide-container { direction: rtl; ... } /* Styles for quota modal */
+@media (prefers-color-scheme: dark) {
+    body, .gradio-container, .prose, table, tr, td, th {
+        background-color: #ffffff !important;
+        color: #333333 !important;
+    }
+}
 </style>
 """
 
+# ادغام CSS و JS
+combined_html = css_code + js_global_content
+
 # استفاده از gr.Blocks
-with gr.Blocks(theme=gr.themes.Soft(primary_hue=colors.steel_blue)) as demo:
-    gr.HTML(js_and_css_code)
+with gr.Blocks() as demo:
+    # تزریق کدها به عنوان HTML
+    gr.HTML(combined_html)
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
-        gr.Markdown("با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", elem_id="main-description")
-
-        # --- بخش جدید: نمایش وضعیت کاربر ---
-        gr.HTML("""
-            <div id="user-status-section">
-                <div id="subscription-badge">...</div>
-                <p id="credit-status-text">در حال بررسی وضعیت حساب...</p>
-                <div id="countdown-timer" style="display: none;">
-                    زمان تا شارژ مجدد اعتبار: <span></span>
-                </div>
-                <button id="upgrade-btn">⭐️ ارتقا به نسخه کامل و نامحدود</button>
-            </div>
-        """)
-
-        # --- بخش جدید: کامپوننت‌های مخفی برای ارتباط JS و Python ---
-        with gr.Row(visible=False):
-            fingerprint_input = gr.Textbox(elem_id="fingerprint-input-for-backend", label="FP")
-            is_paid_input = gr.Textbox(elem_id="is-paid-input-for-backend", label="Paid")
-            status_check_btn = gr.Button(elem_id="status-check-btn-hidden")
-            status_json_output = gr.JSON(label="Status JSON")
+        gr.Markdown(
+            "با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", 
+            elem_id="main-description"
+        )
 
         with gr.Row(equal_height=True):
             with gr.Column():
                 input_image = gr.Image(label="بارگذاری تصویر", type="pil", height=320)
-                prompt = gr.Text(label="دستور ویرایش (به فارسی)", placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...", rtl=True, lines=3)
+                
+                prompt = gr.Text(
+                    label="دستور ویرایش (به فارسی)",
+                    show_label=True,
+                    placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...",
+                    rtl=True,
+                    lines=3
+                )
+                
                 status_box = gr.HTML(label="وضعیت")
+                
+                # Checkbox مخفی برای انتقال وضعیت VIP از جاوااسکریپت به پایتون
+                is_vip_state = gr.Checkbox(visible=False, value=False, label="VIP State")
+
+                # IMPORTANT: Added elem_id="run-btn" here for JS targeting
                 run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn", elem_id="run-btn")
 
             with gr.Column():
                 output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
+                
                 download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
-                lora_adapter = gr.Dropdown(label="انتخاب سبک ویرایش (LoRA)", choices=list(LORA_MAPPING.keys()), value="تبدیل عکس به انیمه")
                 
-                with gr.Accordion("تنظیمات پیشرفته", open=False):
-                    aspect_ratio_selection = gr.Dropdown(label="ابعاد تصویر خروجی", choices=ASPECT_RATIOS_LIST, value="خودکار (پیش‌فرض)")
+                with gr.Row():
+                    lora_adapter = gr.Dropdown(
+                        label="انتخاب سبک ویرایش (LoRA)",
+                        choices=list(LORA_MAPPING.keys()),
+                        value="تبدیل عکس به انیمه"
+                    )
+                
+                with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
+                    aspect_ratio_selection = gr.Dropdown(
+                        label="ابعاد تصویر خروجی",
+                        choices=ASPECT_RATIOS_LIST,
+                        value="خودکار (پیش‌فرض)",
+                        interactive=True
+                    )
+                    
                     with gr.Row(visible=False) as custom_dims_row:
-                        custom_width = gr.Slider(label="عرض", minimum=256, maximum=2048, step=8, value=1024)
-                        custom_height = gr.Slider(label="ارتفاع", minimum=256, maximum=2048, step=8, value=1024)
+                        custom_width = gr.Slider(
+                            label="عرض دلخواه (Width)",
+                            minimum=256, maximum=2048, step=8, value=1024
+                        )
+                        custom_height = gr.Slider(
+                            label="ارتفاع دلخواه (Height)",
+                            minimum=256, maximum=2048, step=8, value=1024
+                        )
+                    
                     seed = gr.Slider(label="دانه تصادفی (Seed)", minimum=0, maximum=MAX_SEED, step=1, value=0)
                     randomize_seed = gr.Checkbox(label="استفاده از Seed تصادفی", value=True)
-                    guidance_scale = gr.Slider(label="وفاداری به متن", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
-                    steps = gr.Slider(label="مراحل پردازش", minimum=1, maximum=50, step=1, value=4)
+                    guidance_scale = gr.Slider(label="میزان وفاداری به متن (Guidance Scale)", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
+                    steps = gr.Slider(label="تعداد مراحل پردازش (Steps)", minimum=1, maximum=50, step=1, value=4)
         
+        # اصلاح تابع نمایش ردیف اسلایدرها
         def toggle_row(choice):
-            return gr.update(visible=choice == "شخصی‌سازی (Custom)")
-        aspect_ratio_selection.change(fn=toggle_row, inputs=aspect_ratio_selection, outputs=custom_dims_row)
+            if choice == "شخصی‌سازی (Custom)":
+                return gr.update(visible=True)
+            return gr.update(visible=False)
+
+        aspect_ratio_selection.change(
+            fn=toggle_row,
+            inputs=aspect_ratio_selection,
+            outputs=custom_dims_row
+        )
 
         gr.Examples(
-            examples=[["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"], ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم بده.", "اصلاح نور و سایه"]],
-            inputs=[input_image, prompt, lora_adapter], outputs=[output_image, seed, status_box], fn=infer_example, cache_examples=False, label="نمونه‌ها"
+            examples=[
+                ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
+                ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم به تصویر بده.", "اصلاح نور و سایه"],
+                ["examples/4.jpg", "از فیلتر ساعت طلایی با پخش نور ملایم استفاده کن.", "نورپردازی مجدد (Relight)"],
+                ["examples/2.jpeg", "دوربین را ۴۵ درجه به سمت چپ بچرخان.", "تغییر زاویه دید"],
+                ["examples/7.jpg", "منبع نور را از سمت راست عقب قرار بده.", "نورپردازی چند زاویه‌ای"],
+                ["examples/10.jpeg", "کیفیت تصویر را افزایش بده (Upscale).", "افزایش کیفیت (Upscale)"],
+                ["examples/7.jpg", "منبع نور را از پایین بتابان.", "نورپردازی چند زاویه‌ای"],
+                ["examples/2.jpeg", "زاویه دوربین را به نمای بالا گوشه راست تغییر بده.", "تغییر زاویه دید"],
+                ["examples/9.jpg", "دوربین کمی به جلو حرکت می‌کند در حالی که نور خورشید از میان ابرها می‌تابد و درخششی نرم اطراف شبح شخصیت در مه ایجاد می‌کند. سبک سینمایی واقعی.", "صحنه بعدی (سینمایی)"],
+                ["examples/8.jpg", "جزئیات پوست سوژه را برجسته‌تر و طبیعی‌تر کن.", "روتوش پوست"],
+                ["examples/6.jpg", "دوربین را به نمای پایین به بالا تغییر بده.", "تغییر زاویه دید"],
+            ],
+            inputs=[input_image, prompt, lora_adapter],
+            outputs=[output_image, seed, status_box],
+            fn=infer_example,
+            cache_examples=False,
+            label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-    # --- اتصال رویدادها ---
+    # جاوااسکریپت برای تزریق متغیر سراسری JS به ورودی Checkbox پایتون
+    # این تابع مقدار window.userIsVip را می‌خواند و به عنوان آخرین آرگومان به تابع پایتون می‌فرستد
+    inject_vip_js = """
+    (img, p, lora, seed, rand, scale, steps, aspect, w, h, vip_dummy) => {
+        return [img, p, lora, seed, rand, scale, steps, aspect, w, h, window.userIsVip];
+    }
+    """
+
     run_button.click(
         fn=infer,
-        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, is_paid_input, fingerprint_input],
+        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, is_vip_state],
         outputs=[output_image, seed, status_box],
+        js=inject_vip_js,
         api_name="predict"
     )
-    
-    # رویداد دکمه مخفی برای گرفتن وضعیت کاربر
-    status_check_btn.click(
-        fn=get_user_status_api,
-        inputs=[fingerprint_input],
-        outputs=[status_json_output]
-    )
 
-    # وقتی خروجی JSON آپدیت شد، تابع جاوااسکریپت را صدا بزن
-    status_json_output.change(
+    download_button.click(
         fn=None,
-        inputs=[status_json_output],
-        js="""
-        (jsonData) => {
-            // Check if the function exists on window before calling
-            if (window.updateUIFromGradio) {
-                window.updateUIFromGradio(jsonData);
-            } else {
-                console.error("Gradio UI update function not found!");
-            }
-        }
-        """
+        inputs=[output_image],
+        outputs=None,
+        js=js_download_func
     )
-    
-    download_button.click(fn=None, inputs=[output_image], outputs=None, js=js_download_func)
-    
-    # --- تعریف API برای ارتباط مستقیم (در صورت نیاز) ---
-    demo.api(name="get_status")(get_user_status_api)
 
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(show_error=True)