Update app.py

app.py

@@ -13,7 +13,7 @@ from gradio.themes.utils import colors, fonts, sizes
 from deep_translator import GoogleTranslator
 from transformers import pipeline
 
-# --- تعریف تم ---
+# --- تنظیمات تم و رنگ ---
 colors.steel_blue = colors.Color(
     name="steel_blue",
     c50="#EBF3F8",
@@ -31,13 +31,13 @@ colors.steel_blue = colors.Color(
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# --- مدیریت کاربران و اعتبار (در حافظه) ---
+# --- دیتابیس موقت حافظه برای کاربران رایگان ---
 # ساختار: { 'fingerprint_hash': {'date': '2023-10-27', 'count': 0} }
 USER_CREDITS_DB = {}
 MAX_FREE_DAILY_CREDITS = 5
-PREMIUM_PAGE_ID = '1149636' # شناسه صفحه پرمیوم برای تشخیص
+PREMIUM_PAGE_ID = '1149636'
 
-# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
+# --- مدل تشخیص محتوای نامناسب (NSFW) ---
 print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
@@ -53,7 +53,7 @@ def is_image_nsfw(image):
         print(f"Safety check error: {e}")
         return False
 
-# --- لیست کلمات ممنوعه بسیار سختگیرانه (انگلیسی) ---
+# --- لیست کلمات ممنوعه (انگلیسی) ---
 BANNED_WORDS = [
     "nsfw", "nude", "naked", "sex", "porn", "erotic", "xxx", "18+", "uncensored",
     "breast", "nipple", "genital", "vagina", "penis", "ass", "butt", "buttocks",
@@ -74,18 +74,14 @@ BANNED_WORDS = [
 def check_text_safety(text):
     if not text: return True
     text_lower = text.lower()
-    # بررسی کلمات دقیق به همراه فاصله برای جلوگیری از اشتباه (مثلا ass در class)
-    # و همچنین بررسی کلمات بدون فاصله برای اطمینان
     for word in BANNED_WORDS:
-        # حالت اول: کلمه مستقل
         if f" {word} " in f" {text_lower} ": 
             return False
-        # حالت دوم: کلمات خاص که حتی چسبیده هم خطرناکند
         if word in ["porn", "hentai", "xxx", "nude", "sex"] and word in text_lower:
             return False
     return True
 
-# --- بارگذاری مدل اصلی ---
+# --- بارگذاری مدل Qwen ---
 from diffusers import FlowMatchEulerDiscreteScheduler
 from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
@@ -106,7 +102,7 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
     torch_dtype=dtype
 ).to(device)
 
-# بارگذاری LoRA ها
+# LoRA Loading
 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")
@@ -130,14 +126,6 @@ LORA_MAPPING = {
     "افزایش کیفیت (Upscale)": "upscale-image"
 }
 
-ASPECT_RATIOS_LIST = [
-    "خودکار (پیش‌فرض)",
-    "۱:۱ (مربع - 1024x1024)",
-    "۱۶:۹ (افقی - 1344x768)",
-    "۹:۱۶ (عمودی - 768x1344)",
-    "شخصی‌سازی (Custom)"
-]
-
 ASPECT_RATIOS_MAP = {
     "خودکار (پیش‌فرض)": "Auto",
     "۱:۱ (مربع - 1024x1024)": (1024, 1024),
@@ -146,6 +134,8 @@ ASPECT_RATIOS_MAP = {
     "شخصی‌سازی (Custom)": "Custom"
 }
 
+ASPECT_RATIOS_LIST = list(ASPECT_RATIOS_MAP.keys())
+
 def translate_prompt(text):
     if not text:
         return ""
@@ -168,16 +158,12 @@ def update_dimensions_on_upload(image):
         new_height = 1024
         aspect_ratio = original_width / original_height
         new_width = int(new_height * aspect_ratio)
+    
+    # Ensure divisible by 8
     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; direction: rtl;">
@@ -194,40 +180,40 @@ def get_success_html(message):
     </div>
     """
 
-# --- منطق بررسی اعتبار کاربر ---
-def check_and_update_credits(fingerprint, is_premium_user):
-    # اگر کاربر پولی است، محدودیت ندارد
-    if is_premium_user:
+# --- مدیریت اعتبار کاربر ---
+def check_credits(fingerprint, is_premium):
+    if is_premium:
         return True, "Unlimited"
     
     if not fingerprint:
-        # اگر اثر انگشت دریافت نشد (مثلاً مرورگر خاص)، موقتا اجازه می‌دهیم اما هشدار لاگ می‌کنیم
-        print("Warning: No fingerprint received.")
-        return True, "Unknown"
+        return True, "Unknown" # Fallback if fingerprint missing
 
     today_str = datetime.now().strftime("%Y-%m-%d")
     
-    # ایجاد رکورد برای کاربر جدید
     if fingerprint not in USER_CREDITS_DB:
         USER_CREDITS_DB[fingerprint] = {'date': today_str, 'count': 0}
     
     user_data = USER_CREDITS_DB[fingerprint]
     
-    # ریست کردن اعتبار اگر روز تغییر کرده باشد
     if user_data['date'] != today_str:
         user_data['date'] = today_str
         user_data['count'] = 0
         
-    # بررسی تعداد استفاده
     if user_data['count'] >= MAX_FREE_DAILY_CREDITS:
         return False, 0
     
-    # کسر اعتبار (افزایش شمارنده)
-    user_data['count'] += 1
+    # محاسبه اعتبار باقی‌مانده بدون کسر (کسر بعد از موفقیت انجام میشه)
     remaining = MAX_FREE_DAILY_CREDITS - user_data['count']
     return True, remaining
 
-@spaces.GPU(duration=30)
+def deduct_credit(fingerprint, is_premium):
+    if is_premium or not fingerprint:
+        return
+    
+    if fingerprint in USER_CREDITS_DB:
+        USER_CREDITS_DB[fingerprint]['count'] += 1
+
+@spaces.GPU(duration=45)
 def infer(
     input_image,
     prompt,
@@ -239,890 +225,388 @@ def infer(
     aspect_ratio_selection,
     custom_width,
     custom_height,
-    user_fingerprint,  # ورودی مخفی
-    user_is_premium,   # ورودی مخفی (رشته "true" یا "false")
+    user_fingerprint,
+    user_is_premium,
     progress=gr.Progress(track_tqdm=True)
 ):
-    # تبدیل وضعیت پرمیوم به بولین
+    # --- رفع خطای NoneType vs Int ---
+    # اگر مقادیر عددی نال باشند، مقدار پیش‌فرض ست می‌کنیم
+    seed = seed if seed is not None else 0
+    steps = steps if steps is not None else 4
+    guidance_scale = guidance_scale if guidance_scale is not None else 1.0
+    custom_width = custom_width if custom_width is not None else 1024
+    custom_height = custom_height if custom_height is not None else 1024
+    
+    # تبدیل وضعیت پرمیوم
     is_premium = str(user_is_premium).lower() == 'true'
 
     if input_image is None:
         return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
 
-    # 1. بررسی سهمیه کاربر (قبل از هر پردازش سنگین)
-    allowed, remaining_credits = check_and_update_credits(user_fingerprint, is_premium)
+    # 1. بررسی اعتبار
+    allowed, credit_info = check_credits(user_fingerprint, is_premium)
     if not allowed:
-        msg = "اعتبار رایگان امروز شما (۵ تصویر) به پایان رسیده است. برای استفاده نامحدود حساب خود را ارتقا دهید."
-        # تریگر کردن نمایش دکمه ارتقا در فرانت‌اند با یک کلاس خاص یا پیام
+        msg = "اعتبار رایگان امروز شما (۵ تصویر) به پایان رسیده است."
         return None, seed, get_error_html(msg + " <span class='quota-limit-reached' style='display:none'></span>")
 
-    # 2. بررسی تصویر ورودی برای NSFW
+    # 2. بررسی تصویر ورودی (NSFW)
     if is_image_nsfw(input_image):
-        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
+        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است.")
 
-    # 3. ترجمه و بررسی متن
+    # 3. بررسی متن
     english_prompt = translate_prompt(prompt)
     if not check_text_safety(english_prompt):
-        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز، غیراخلاقی یا جنسی است.")
+        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])
+    # تنظیمات مدل
+    adapter_name = LORA_MAPPING.get(lora_adapter_persian)
+    if adapter_name:
+        pipe.set_adapters([adapter_name], adapter_weights=[1.0])
     
     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, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood, child abuse"
-    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 = (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)
+    # ابعاد
+    dims = ASPECT_RATIOS_MAP.get(aspect_ratio_selection)
+    if dims == "Custom":
+        w = (int(custom_width) // 8) * 8
+        h = (int(custom_height) // 8) * 8
+    elif dims == "Auto" or dims is None:
+        w, h = update_dimensions_on_upload(input_image)
     else:
-        width, height = selection_value
+        w, h = dims
 
     try:
+        final_negative = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood, bad anatomy, bad hands, missing fingers, blurry"
+        
         result = pipe(
-            image=original_image,
+            image=input_image.convert("RGB"),
             prompt=english_prompt,
-            negative_prompt=final_negative_prompt,
-            height=height,
-            width=width,
-            num_inference_steps=steps,
+            negative_prompt=final_negative,
+            height=h,
+            width=w,
+            num_inference_steps=int(steps), # اطمینان از int بودن
             generator=generator,
-            true_cfg_scale=guidance_scale,
+            true_cfg_scale=float(guidance_scale), # اطمینان از float بودن
         ).images[0]
 
-        # 4. بررسی تصویر خروجی
+        # 4. بررسی خروجی (NSFW)
         if is_image_nsfw(result):
-            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+            return None, seed, get_error_html("تصویر خروجی حاوی محتوای نامناسب بود و حذف شد.")
         
-        # پیام موفقیت با نمایش اعتبار باقی‌مانده برای کاربران رایگان
-        success_msg = "تصویر با موفقیت ویرایش شد."
+        # کسر اعتبار پس از موفقیت
+        deduct_credit(user_fingerprint, is_premium)
+        
+        # پیام موفقیت
+        success_msg = "تصویر با موفقیت ساخته شد."
         if not is_premium:
-            success_msg += f"<br><small>اعتبار باقی‌مانده امروز: {remaining_credits} تصویر</small>"
+            # محاسبه مجدد باقیمانده
+            _, new_rem = check_credits(user_fingerprint, is_premium)
+            success_msg += f"<br><small>اعتبار باقی‌مانده امروز: {new_rem}</small>"
             
         return result, seed, get_success_html(success_msg)
 
     except Exception as e:
         error_str = str(e)
-        if "quota" in error_str.lower() or "exceeded" in error_str.lower():
-             raise e 
-        return None, seed, get_error_html(f"خطا در پردازش: {error_str}")
+        if "quota" 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, "example_user", "true")
-    return res, s, status
-
-# --- جاوااسکریپت برای دکمه دانلود ---
-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
-    }, '*');
-}
-"""
+    # کاربر پرمیوم فیک برای اجرای نمونه‌ها
+    return infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "test_fp", "true")
 
-# --- جاوااسکریپت سراسری (شناسایی کاربر، اثر انگشت، UI) ---
-js_global_content = """
+# --- جاوااسکریپت و HTML ---
+js_code = """
 <script>
-// --- پیکربندی ---
+// --- تنظیمات ---
 const PREMIUM_PAGE_ID = '1149636';
 const PREMIUM_URL = '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991';
 
-// --- تولید اثر انگشت مرورگر ---
+// --- تولید اثر انگشت ---
 function getBrowserFingerprint() {
     const components = [navigator.userAgent, navigator.language, screen.width + 'x' + screen.height, new Date().getTimezoneOffset()];
     try {
         const canvas = document.createElement('canvas');
         const ctx = canvas.getContext('2d');
-        ctx.textBaseline = "top";
-        ctx.font = "14px 'Arial'";
-        ctx.textBaseline = "alphabetic";
-        ctx.fillStyle = "#f60";
-        ctx.fillRect(125, 1, 62, 20);
-        ctx.fillStyle = "#069";
-        ctx.fillText("a1b2c3d4e5f6g7h8i9j0_!@#$%^&*()", 2, 15);
+        ctx.textBaseline = "top"; ctx.font = "14px 'Arial'"; ctx.textBaseline = "alphabetic";
+        ctx.fillStyle = "#f60"; ctx.fillRect(125, 1, 62, 20);
+        ctx.fillStyle = "#069"; ctx.fillText("val", 2, 15);
         components.push(canvas.toDataURL());
-    } catch (e) {
-        components.push("canvas-error");
-    }
-    const fingerprintString = components.join('~~~');
+    } catch(e) { components.push("err"); }
+    const str = components.join('~~~');
     let hash = 0;
-    for (let i = 0; i < fingerprintString.length; i++) {
-        const char = fingerprintString.charCodeAt(i);
-        hash = ((hash << 5) - hash) + char;
+    for (let i = 0; i < str.length; i++) {
+        hash = ((hash << 5) - hash) + str.charCodeAt(i);
         hash |= 0;
     }
     return 'fp_' + Math.abs(hash).toString(16);
 }
 
-// --- مدیریت وضعیت اشتراک ---
 let userFingerprint = getBrowserFingerprint();
 let isUserPremium = false;
 
-function isUserPaid(userObject) {
-    return userObject && userObject.isLogin && userObject.accessible_pages && (userObject.accessible_pages.includes(PREMIUM_PAGE_ID) || userObject.accessible_pages.includes(parseInt(PREMIUM_PAGE_ID)));
-}
-
-function updateUIForSubscription(status) {
-    const badge = document.getElementById('user-badge');
-    const descText = document.getElementById('user-desc-text');
+// --- بروزرسانی UI وضعیت کاربر ---
+function updateStatusUI(status) {
+    const badge = document.getElementById('status-badge-text');
+    const desc = document.getElementById('status-desc');
+    const upgradeBox = document.getElementById('upgrade-box');
     
     if (status === 'paid') {
         isUserPremium = true;
         if(badge) {
-            badge.textContent = 'نسخه نامحدود 💎';
-            badge.className = 'status-badge paid-badge';
+            badge.innerText = 'نسخه نامحدود 💎';
+            badge.className = 'badge paid';
         }
-        if(descText) descText.textContent = 'برای کاربران پولی: استفاده نامحدود';
-        document.getElementById('upgrade-btn-container').style.display = 'none';
+        if(desc) desc.innerText = 'وضعیت: کاربر ویژه (بدون محدودیت)';
+        if(upgradeBox) upgradeBox.style.display = 'none';
     } else {
         isUserPremium = false;
         if(badge) {
-            badge.textContent = 'نسخه رایگان';
-            badge.className = 'status-badge free-badge';
+            badge.innerText = 'نسخه رایگان';
+            badge.className = 'badge free';
         }
-        if(descText) descText.textContent = 'برای کاربران رایگان: ۵ تصویر در روز';
+        if(desc) desc.innerText = '۵ تصویر رایگان در روز دارید';
     }
-    
-    // آپدیت کردن اینپوت‌های مخفی گرادیو
-    updateGradioHiddenInputs();
+    syncHiddenInputs();
 }
 
-function updateGradioHiddenInputs() {
-    // پیدا کردن تکست‌باکس‌های مخفی و مقداردهی
-    // ما از کلاس‌های خاص استفاده می‌کنیم که در پایتون تعریف شده‌اند
-    const fpInput = document.querySelector('.hidden-fingerprint textarea');
-    const premiumInput = document.querySelector('.hidden-premium textarea');
+function syncHiddenInputs() {
+    // پیدا کردن تکست‌باکس‌های مخفی گرادیو و پر کردن آن‌ها
+    const fpArea = document.querySelector('.hidden-fingerprint textarea');
+    const premArea = document.querySelector('.hidden-premium textarea');
     
-    if (fpInput) {
-        fpInput.value = userFingerprint;
-        fpInput.dispatchEvent(new Event('input', { bubbles: true }));
+    if (fpArea) {
+        fpArea.value = userFingerprint;
+        fpArea.dispatchEvent(new Event('input', { bubbles: true }));
     }
-    if (premiumInput) {
-        premiumInput.value = isUserPremium ? "true" : "false";
-        premiumInput.dispatchEvent(new Event('input', { bubbles: true }));
+    if (premArea) {
+        premArea.value = isUserPremium ? "true" : "false";
+        premArea.dispatchEvent(new Event('input', { bubbles: true }));
     }
 }
 
-// گوش دادن به پیام از اپلیکیشن مادر
+function isUserPaid(userObject) {
+    return userObject && userObject.isLogin && userObject.accessible_pages && 
+           (userObject.accessible_pages.includes(PREMIUM_PAGE_ID) || userObject.accessible_pages.includes(parseInt(PREMIUM_PAGE_ID)));
+}
+
+// --- شنود پیام از اپلیکیشن مادر ---
 window.addEventListener('message', (event) => {
     if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
-        if (event.data.error || !event.data.payload) {
-            updateUIForSubscription('free');
-            return;
-        }
         try {
-            const userObject = JSON.parse(event.data.payload);
-            const status = isUserPaid(userObject) ? 'paid' : 'free';
-            updateUIForSubscription(status);
-        } catch (e) {
-            updateUIForSubscription('free');
-        }
+            if (event.data.error || !event.data.payload) {
+                updateStatusUI('free');
+            } else {
+                const userObj = JSON.parse(event.data.payload);
+                updateStatusUI(isUserPaid(userObj) ? 'paid' : 'free');
+            }
+        } catch (e) { updateStatusUI('free'); }
     }
 });
 
-// درخواست وضعیت کاربر هنگام لود
+// --- شروع کار ---
 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';
-        }
-        document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
-    };
-    forceLight();
-    setInterval(forceLight, 1000);
-
-    // درخواست وضعیت از پرنت
+    // درخواست وضعیت کاربر به محض لود
     parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
     
-    // اگر پاسخی نیامد پیش‌فرض رایگان
+    // Fallback: اگر پاسخی نیامد بعد از 2 ثانیه رایگان فرض کن
     setTimeout(() => {
-        if (!document.getElementById('user-badge').textContent) {
-            updateUIForSubscription('free');
+        const badge = document.getElementById('status-badge-text');
+        if (!badge || badge.innerText === '') {
+            updateStatusUI('free');
         }
-    }, 2000);
-
-    // 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();
-    };
-    
-    // Close function
-    window.closeErrorModal = function() {
-         const modal = document.getElementById('custom-quota-modal');
-         if (modal) modal.remove();
-    };
-
-    // هدایت به صفحه خرید
-    window.navigateToPremium = function() {
-        parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
-    };
-
-    // 3. SHOW MODAL FUNCTION (IP LIMIT)
-    const showQuotaModal = () => {
-        if (document.getElementById('custom-quota-modal')) return;
+    }, 2500);
 
-        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 (Monitor for errors)
+    // اسکن برای پیام‌های خطا و لیمیت
     setInterval(() => {
-        // الف) چک کردن خطای Quota
-        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();
-                 el.style.display = 'none';
-                 el.style.opacity = '0';
-                 const parentWrap = el.closest('.toast-wrap');
-                 if(parentWrap) parentWrap.style.display = 'none';
+        // Quota Limit from Python
+        if (document.querySelector('.quota-limit-reached')) {
+            document.getElementById('upgrade-box').style.display = 'block';
+            document.querySelector('.quota-limit-reached').remove();
+        }
+        
+        // Hide default Gradio toasts if needed
+        document.querySelectorAll('.toast-wrap').forEach(el => {
+            if (el.innerText.includes('quota') || el.innerText.includes('exceeded')) {
+                el.style.display = 'none';
+                showQuotaModal();
             }
         });
-        
-        // ب) چک کردن اتمام اعتبار روزانه (از طریق کلاس HTML که در پایتون برگرداندیم)
-        const limitReached = document.querySelector('.quota-limit-reached');
-        if (limitReached) {
-            document.getElementById('upgrade-btn-container').style.display = 'block';
-            limitReached.remove(); // جلوگیری از اجرای مکرر
-        }
-
-    }, 200);
+    }, 500);
 });
 
-// تابع کمکی برای اطمینان از ارسال مقادیر قبل از کلیک
-function prepareSubmission() {
-    updateGradioHiddenInputs();
+// توابع کمکی
+window.openPremium = function() {
+    parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
+};
+
+window.prepareSubmission = function() {
+    syncHiddenInputs();
     return [];
 }
+
+// دانلود
+window.downloadImage = async (image) => {
+    if (!image) return alert("تصویری وجود ندارد!");
+    let url = image.url;
+    if (!url.startsWith('http')) url = window.location.origin + url;
+    window.parent.postMessage({ type: 'DOWNLOAD_REQUEST', url: url }, '*');
+};
+
+// مودال خطای IP
+function showQuotaModal() {
+    if (document.getElementById('custom-quota-modal')) return;
+    const html = `<div id="custom-quota-modal" style="position:fixed;top:0;left:0;width:100%;height:100%;background:rgba(0,0,0,0.8);z-index:99999;display:flex;justify-content:center;align-items:center;color:#fff;font-family:'Vazirmatn';">
+        <div style="background:#2d3748;padding:2rem;border-radius:1rem;text-align:center;max-width:90%;">
+            <h3>خطای محدودیت سرور</h3>
+            <p>لطفاً IP خود را تغییر دهید (VPN را خاموش/روشن کنید) و دوباره تلاش کنید.</p>
+            <button onclick="document.getElementById('custom-quota-modal').remove();" style="padding:10px 20px;margin-top:10px;border-radius:8px;border:none;background:#4a5568;color:white;cursor:pointer;">باشه</button>
+        </div>
+    </div>`;
+    document.body.insertAdjacentHTML('beforeend', html);
+}
 </script>
-"""
 
-# --- CSS Updated ---
-css_code = """
 <style>
-@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700;900&display=swap');
+@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;700&display=swap');
 
-: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;
-}
+body { font-family: 'Vazirmatn', sans-serif !important; }
 
-/* User Status Badge */
-#user-status-container {
+/* استایل بج و هدر */
+.user-info-header {
     display: flex;
     flex-direction: column;
     align-items: center;
     justify-content: center;
+    background: #ffffff;
+    padding: 15px;
+    border-radius: 16px;
     margin-bottom: 20px;
-    animation: fadeIn 0.8s ease-out;
-}
-.status-badge {
-    padding: 6px 16px;
-    border-radius: 20px;
-    font-size: 0.9em;
-    font-weight: 700;
-    margin-bottom: 8px;
-    letter-spacing: 0.5px;
-    text-shadow: 0 1px 2px rgba(0,0,0,0.1);
-}
-.free-badge {
-    background: linear-gradient(45deg, #6c757d, #495057);
-    color: white;
-    box-shadow: 0 4px 10px rgba(108, 117, 125, 0.3);
-}
-.paid-badge {
-    background: linear-gradient(45deg, #FFD700, #FFA500);
-    color: #333;
-    box-shadow: 0 4px 10px rgba(255, 193, 7, 0.4);
-}
-#user-desc-text {
-    font-size: 0.9em;
-    color: #4b5563;
-}
-
-/* Upgrade Button */
-@keyframes pulse-gold {
-    0% { transform: scale(1); box-shadow: 0 0 0 0 rgba(255, 193, 7, 0.7); }
-    70% { transform: scale(1.02); box-shadow: 0 0 0 10px rgba(255, 193, 7, 0); }
-    100% { transform: scale(1); box-shadow: 0 0 0 0 rgba(255, 193, 7, 0); }
-}
-#upgrade-premium-btn {
-    width: 100%;
-    margin-top: 10px;
-    padding: 12px;
-    font-family: 'Vazirmatn', sans-serif;
-    font-size: 1.1em;
-    font-weight: 800;
-    color: #212529;
-    background: linear-gradient(95deg, #FFD54F, #FFC107 100%);
-    border: none;
-    border-radius: 14px;
-    cursor: pointer;
-    box-shadow: 0 8px 20px -5px rgba(255, 193, 7, 0.5);
-    animation: pulse-gold 2s infinite;
-    display: block;
-}
-#upgrade-premium-btn:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 12px 25px -5px rgba(255, 193, 7, 0.6);
-}
-
-/* --- 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);
+    box-shadow: 0 4px 15px rgba(0,0,0,0.05);
+    border: 1px solid #e2e8f0;
 }
 
-@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;
+.badge {
+    padding: 6px 18px;
+    border-radius: 50px;
+    font-size: 0.95rem;
     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: 20px;
-    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;
+    margin-bottom: 8px;
+    display: inline-block;
 }
 
-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;
+.badge.free {
+    background: #e2e8f0;
+    color: #4a5568;
+    border: 1px solid #cbd5e0;
 }
 
-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;
+.badge.paid {
+    background: linear-gradient(135deg, #FFD700 0%, #FFA500 100%);
+    color: #744210;
+    box-shadow: 0 4px 12px rgba(255, 165, 0, 0.3);
 }
 
-.gr-dropdown {
-    background: #ffffff !important;
-    border-radius: 12px !important;
+#status-desc {
+    font-size: 0.9rem;
+    color: #718096;
 }
 
-.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;
+/* دکمه ارتقا */
+#upgrade-box {
     width: 100%;
-    margin-top: 15px;
-}
-
-.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;
+    margin-top: 12px;
+    display: none; /* پیش‌فرض مخفی */
 }
 
-.gradio-container tbody tr {
-    background-color: #ffffff !important;
-    border-bottom: 1px solid #f3f4f6 !important;
-}
-
-.gradio-container tbody tr:hover {
-    background-color: #f9fafb !important;
+.upgrade-btn {
+    width: 100%;
+    background: linear-gradient(90deg, #F59E0B, #D97706);
+    color: white;
+    border: none;
+    padding: 12px;
+    border-radius: 12px;
+    font-weight: 700;
+    cursor: pointer;
+    font-family: inherit;
+    animation: pulse 2s infinite;
 }
-
-.gradio-container tbody td {
-    background-color: #ffffff !important;
-    color: #374151 !important;
-    padding: 10px !important;
+@keyframes pulse {
+    0% { transform: scale(1); }
+    50% { transform: scale(1.02); }
+    100% { transform: scale(1); }
 }
 
-.gradio-container tbody td span, 
-.gradio-container tbody td p {
-    color: #374151 !important;
-}
+.hidden-fingerprint, .hidden-premium { display: none !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;
-}
-
-/* Hidden inputs logic */
-.hidden-fingerprint, .hidden-premium {
-    display: none !important;
-}
-
-@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() as demo:
-    # تزریق کدها به عنوان HTML
-    gr.HTML(combined_html)
+with gr.Blocks(theme=Soft()) as demo:
+    gr.HTML(js_code)
     
     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-container">
-            <span id="user-badge" class="status-badge"></span>
-            <span id="user-desc-text"></span>
-            <div id="upgrade-btn-container" style="display:none; width: 100%;">
-                <button id="upgrade-premium-btn" onclick="navigateToPremium()">
-                    ⭐️ ارتقا به نسخه کامل و نامحدود
-                </button>
+        <div class="user-info-header">
+            <span id="status-badge-text" class="badge"></span>
+            <span id="status-desc">در حال بررسی اشتراک...</span>
+            <div id="upgrade-box">
+                <button class="upgrade-btn" onclick="openPremium()">⭐️ ارتقا به نسخه نامحدود</button>
             </div>
         </div>
         """)
+        
+        # ورودی‌های مخفی برای ارتباط JS -> Python
+        user_fingerprint = gr.Textbox(elem_classes="hidden-fingerprint", visible=False)
+        user_is_premium = gr.Textbox(elem_classes="hidden-premium", visible=False)
 
-        # --- ورودی‌های مخفی برای انتقال اطلاعات از JS به Python ---
-        hidden_fingerprint = gr.Textbox(elem_classes="hidden-fingerprint", visible=False)
-        hidden_premium = gr.Textbox(elem_classes="hidden-premium", visible=False)
-
-        with gr.Row(equal_height=True):
+        with gr.Row():
             with gr.Column():
-                input_image = gr.Image(label="بارگذاری تصویر", type="pil", height=320)
-                
-                prompt = gr.Text(
-                    label="دستور ویرایش (به فارسی)",
-                    show_label=True,
-                    placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...",
-                    rtl=True,
-                    lines=3
-                )
-                
-                status_box = gr.HTML(label="وضعیت")
+                input_image = gr.Image(label="تصویر اصلی", type="pil", height=320)
+                prompt = gr.Text(label="دستور ویرایش (فارسی)", placeholder="مثلاً: تصویر را کارتونی کن", rtl=True, lines=3)
+                run_btn = gr.Button("✨ شروع پردازش", variant="primary")
                 
-                # 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")
-                
-                with gr.Row():
-                    lora_adapter = gr.Dropdown(
-                        label="انتخاب سبک ویرایش (LoRA)",
-                        choices=list(LORA_MAPPING.keys()),
-                        value="تبدیل عکس به انیمه"
-                    )
+                output_image = gr.Image(label="نتیجه", interactive=False)
+                dl_btn = gr.Button("📥 دانلود تصویر")
                 
-                with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
-                    aspect_ratio_selection = gr.Dropdown(
-                        label="ابعاد تصویر خروجی",
-                        choices=ASPECT_RATIOS_LIST,
-                        value="خودکار (پیش‌فرض)",
-                        interactive=True
-                    )
+                with gr.Accordion("تنظیمات پیشرفته", open=False):
+                    lora_adapter = gr.Dropdown(list(LORA_MAPPING.keys()), label="سبک", value="تبدیل عکس به انیمه")
+                    ratio = gr.Dropdown(ASPECT_RATIOS_LIST, label="ابعاد", value="خودکار (پیش‌فرض)")
                     
-                    with gr.Row(visible=False) as custom_dims_row:
-                        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
-                        )
+                    with gr.Row(visible=False) as custom_row:
+                        w_slider = gr.Slider(256, 2048, value=1024, step=8, label="عرض")
+                        h_slider = gr.Slider(256, 2048, value=1024, step=8, label="ارتفاع")
+                        
+                    ratio.change(lambda x: gr.update(visible=(x == "شخصی‌سازی (Custom)")), ratio, custom_row)
                     
-                    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="میزان وفاداری به متن (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)
+                    seed = gr.Slider(0, MAX_SEED, label="Seed", value=0)
+                    rand_seed = gr.Checkbox(True, label="Seed تصادفی")
+                    cfg = gr.Slider(1.0, 10.0, value=5.0, label="Guidance Scale") # مقدار پیشفرض مطمئن
+                    steps = gr.Slider(1, 50, value=30, label="Steps") # مقدار پیشفرض مطمئن
         
-        # اصلاح تابع نمایش ردیف اسلایدرها
-        def toggle_row(choice):
-            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
-        )
+        status_box = gr.HTML()
 
-        gr.Examples(
-            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="نمونه‌ها (برای تست کلیک کنید)"
-        )
-
-    # اجرای پروسه با انتقال اطلاعات کاربر
-    run_button.click(
+    # اتصال دکمه‌ها
+    run_btn.click(
         fn=infer,
         inputs=[
-            input_image, prompt, lora_adapter, seed, randomize_seed, 
-            guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height,
-            hidden_fingerprint, hidden_premium # ورودی‌های جدید
+            input_image, prompt, lora_adapter, seed, rand_seed, cfg, steps,
+            ratio, w_slider, h_slider, user_fingerprint, user_is_premium
         ],
         outputs=[output_image, seed, status_box],
-        api_name="predict",
-        js="prepareSubmission" # فراخوانی تابع JS قبل از ارسال به پایتون
-    )
-
-    download_button.click(
-        fn=None,
-        inputs=[output_image],
-        outputs=None,
-        js=js_download_func
+        js="window.prepareSubmission"  # فراخوانی JS قبل از اجرا برای سینک کردن وضعیت
     )
+    
+    dl_btn.click(fn=None, inputs=[output_image], js="window.downloadImage")
 
 if __name__ == "__main__":
-    demo.queue(max_size=30).launch(show_error=True)
+    demo.queue().launch(show_error=True)