Update app.py

app.py

@@ -4,21 +4,16 @@ import numpy as np
 import spaces
 import torch
 import random
-import json
-import time
-from datetime import date
-from PIL import Image
-from gradio.themes.utils import colors
+from PIL import Image, ImageFilter
+from typing import Iterable
+from gradio.themes import Soft
+from gradio.themes.utils import colors, fonts, sizes
 from deep_translator import GoogleTranslator
 from transformers import pipeline
+from datetime import date
+import json
 
-# --- 1. تنظیمات سیستم اعتبار و اشتراک ---
-USAGE_LIMIT = 5  # اعتبار رایگان روزانه
-PREMIUM_PAGE_ID = '1149636' # آی‌دی صفحه محصول پریمیم
-PREMIUM_URL = '#/nav/online/news/getSingle/1149636' # لینک خرید اشتراک
-usage_data_cache = {} # حافظه موقت سرور برای نگهداری تعداد استفاده
-
-# --- 2. تنظیمات تم و مدل‌ها ---
+# --- تعریف تم ---
 colors.steel_blue = colors.Color(
     name="steel_blue",
     c50="#EBF3F8",
@@ -36,7 +31,12 @@ colors.steel_blue = colors.Color(
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# --- بارگذاری مدل Safety Checker ---
+# --- تنظیمات سیستم اعتبار ---
+USAGE_LIMIT = 5
+PREMIUM_PAGE_ID = '1149636'  # شناسه محصول برای تشخیص کاربر پولی
+user_quotas = {}  # حافظه موقت برای ذخیره تعداد درخواست‌های روزانه
+
+# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
 print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
@@ -52,7 +52,7 @@ def is_image_nsfw(image):
         print(f"Safety check error: {e}")
         return False
 
-# --- بارگذاری مدل Qwen ---
+# --- بارگذاری مدل اصلی ---
 from diffusers import FlowMatchEulerDiscreteScheduler
 from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
@@ -97,35 +97,113 @@ LORA_MAPPING = {
     "افزایش کیفیت (Upscale)": "upscale-image"
 }
 
-ASPECT_RATIOS_LIST = ["خودکار (پیش‌فرض)", "۱:۱ (مربع - 1024x1024)", "۱۶:۹ (افقی - 1344x768)", "۹:۱۶ (عمودی - 768x1344)", "شخصی‌سازی (Custom)"]
-ASPECT_RATIOS_MAP = {"خودکار (پیش‌فرض)": "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+"]
+ASPECT_RATIOS_LIST = [
+    "خودکار (پیش‌فرض)",
+    "۱:۱ (مربع - 1024x1024)",
+    "۱۶:۹ (افقی - 1344x768)",
+    "۹:۱۶ (عمودی - 768x1344)",
+    "شخصی‌سازی (Custom)"
+]
+
+ASPECT_RATIOS_MAP = {
+    "خودکار (پیش‌فرض)": "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
+        if f" {word} " in f" {text_lower} ": 
+            return False
     return True
 
 def translate_prompt(text):
-    if not text: return ""
-    try: return GoogleTranslator(source='auto', target='en').translate(text)
-    except: return 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
-    if w > h: new_w = 1024; ar = h/w; new_h = int(new_w * ar)
-    else: new_h = 1024; ar = w/h; new_w = int(new_h * ar)
-    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_user_quota(fingerprint, status):
+    # اگر کاربر پولی است، محدودیت ندارد
+    if status == 'paid':
+        return True, ""
+    
+    # اگر فینگرپرینت وجود ندارد (خطای کلاینت)، سخت‌گیری نمی‌کنیم اما لاگ می‌کنیم
+    if not fingerprint:
+        return True, "" 
+
+    today = date.today().isoformat()
+    
+    # اگر کاربر جدید است یا روز تغییر کرده، رکورد جدید بساز
+    if fingerprint not in user_quotas or user_quotas[fingerprint]['date'] != today:
+        user_quotas[fingerprint] = {'date': today, 'count': 0}
+    
+    # بررسی محدودیت
+    if user_quotas[fingerprint]['count'] >= USAGE_LIMIT:
+        return False, "اعتبار رایگان روزانه شما (۵ تصویر) تمام شده است. لطفاً برای استفاده نامحدود ارتقا دهید یا فردا تلاش کنید."
+    
+    return True, ""
+
+# --- تابع افزایش شمارنده اعتبار ---
+def increment_user_quota(fingerprint, status):
+    if status == 'paid' or not fingerprint:
+        return
+    if fingerprint in user_quotas:
+        user_quotas[fingerprint]['count'] += 1
 
-# --- 3. تابع اصلی پردازش (شامل بررسی اعتبار) ---
 @spaces.GPU(duration=30)
 def infer(
     input_image,
@@ -138,346 +216,735 @@ def infer(
     aspect_ratio_selection,
     custom_width,
     custom_height,
-    # ورودی‌های جدید برای سیستم اعتبار که از JS می‌آیند
-    user_fingerprint,
-    user_subscription_status,
+    fingerprint,  # ورودی جدید مخفی
+    user_status,  # ورودی جدید مخفی
     progress=gr.Progress(track_tqdm=True)
 ):
-    # --- بررسی اعتبار کاربر ---
-    # اگر Fingerprint نرسد یعنی کاربر مستقیماً API را صدا زده یا JS لود نشده
-    if not user_fingerprint:
-        # برای جلوگیری از خطا در تست‌های لوکال، اگر نال بود یک مقدار موقت می‌دهیم، 
-        # اما در پروداکشن واقعی می‌توانید ریترن ارور کنید.
-        user_fingerprint = "unknown_user"
-    
-    # اگر کاربر پولی نباشد، اعتبار چک می‌شود
-    if user_subscription_status != 'paid':
-        today_str = date.today().isoformat()
-        
-        # بازیابی یا ایجاد رکورد کاربر
-        user_record = usage_data_cache.get(user_fingerprint)
-        if not user_record or user_record.get("last_reset") != today_str:
-            user_record = {"count": 0, "last_reset": today_str}
-            usage_data_cache[user_fingerprint] = user_record
-        
-        # چک کردن لیمیت
-        if user_record["count"] >= USAGE_LIMIT:
-            msg = f"اعتبار رایگان روزانه شما ({USAGE_LIMIT} تصویر) به پایان رسیده است. برای استفاده نامحدود ارتقا دهید."
-            return None, seed, get_error_html(msg)
-        
-        # کسر اعتبار (افزایش شمارنده)
-        user_record["count"] += 1
-        print(f"User {user_fingerprint} usage: {user_record['count']}/{USAGE_LIMIT}")
+    # 1. بررسی اعتبار قبل از هر پردازش
+    is_allowed, msg = check_user_quota(fingerprint, user_status)
+    if not is_allowed:
+        return None, seed, get_error_html(msg)
+
+    if input_image is None:
+        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("تصویر ورودی دارای محتوای نامناسب است.")
+    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)
     
-    final_negative_prompt = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood, worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
+    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" or selection_value is None: width, height = update_dimensions_on_upload(original_image)
-    else: width, height = selection_value
+    
+    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)
+    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,
+            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("تصویر تولید شده حاوی محتوای نامناسب بود.")
+        if is_image_nsfw(result):
+            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
         
+        # اگر موفقیت‌آمیز بود، اعتبار را کم کن
+        increment_user_quota(fingerprint, user_status)
+
         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, "example_user", "paid")
+    # برای مثال‌ها، اعتبار را چک ��می‌کنیم یا فینگرپرینت خالی می‌فرستیم
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "", "free")
     return res, s, status
 
-# --- 4. کدهای جاوا اسکریپت ---
-
-# کد دانلود فایل
+# --- جاوااسکریپت برای دکمه دانلود ---
 js_download_func = """
 async (image) => {
-    if (!image) { alert("لطفاً ابتدا تصویر را تولید کنید."); return; }
+    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 }, '*');
+    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
+    }, '*');
 }
 """
 
-# جاوا اسکریپت اصلی (اعتبار، فینگرپرینت، تم، ارور)
+# --- جاوااسکریپت سراسری (شامل فینگرپرینت، استعلام وضعیت کاربر و ریست آی‌پی) ---
 js_global_content = """
 <script>
-// --- تنظیمات ثابت ---
-const PREMIUM_PAGE_ID = '1149636'; 
-const PREMIUM_URL = '#/nav/online/news/getSingle/1149636';
-
-// متغیرهای گلوبال برای نگهداری وضعیت
-window.userFingerprint = null;
-window.userSubscriptionStatus = 'free';
-
-// --- 1. تابع ساخت فینگرپرینت ---
+// --- START: Fingerprint & User Status Logic ---
 async 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); 
-        components.push(canvas.toDataURL()); 
-    } catch (e) { components.push("canvas-error"); }
+    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); components.push(canvas.toDataURL()); } catch (e) { components.push("canvas-error"); }
     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; }
     return 'fp_' + Math.abs(hash).toString(16);
 }
 
-// --- 2. توابع بررسی اشتراک ---
-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 updateUIForSubscriptionStatus(status) {
-    window.userSubscriptionStatus = status;
-    const badge = document.getElementById('subscription-badge');
-    const infoText = document.getElementById('credit-info-text');
-    const upgradeBtn = document.getElementById('upgrade-btn-custom');
-    
-    if (status === 'paid') { 
-        if(badge) { badge.textContent = '⭐️ نسخه نامحدود'; badge.className = 'badge-paid'; }
-        if(infoText) infoText.innerHTML = 'شما به صورت <b>نامحدود</b> می‌توانید تصویر بسازید.';
-        if(upgradeBtn) upgradeBtn.style.display = 'none';
-    } else { 
-        if(badge) { badge.textContent = 'نسخه رایگان'; badge.className = 'badge-free'; }
-        if(infoText) infoText.innerHTML = 'اعتبار رایگان روزانه: <b>۵ تصویر</b>. برای استفاده نامحدود ارتقا دهید.';
-        if(upgradeBtn) upgradeBtn.style.display = 'block';
-    }
-}
+// Global variables to hold status
+window.userFingerprint = null;
+window.userSubscriptionStatus = 'free';
 
-// --- 3. اجرای اولیه و لیسنرها ---
 document.addEventListener('DOMContentLoaded', async () => {
-    // تولید فینگرپرینت
+    // 1. Calculate Fingerprint
     window.userFingerprint = await getBrowserFingerprint();
-    
-    // درخواست وضعیت از والد
-    window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
+    console.log("Fingerprint:", window.userFingerprint);
 
-    // شنیدن پاسخ والد
+    // 2. Request User Status from Parent
     window.addEventListener('message', (event) => {
         if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
-            if (event.data.error || !event.data.payload) { 
-                updateUIForSubscriptionStatus('free'); 
-            } else {
-                try { 
-                    const userObject = JSON.parse(event.data.payload); 
-                    const status = isUserPaid(userObject) ? 'paid' : 'free'; 
-                    updateUIForSubscriptionStatus(status); 
-                } catch (e) { updateUIForSubscriptionStatus('free'); }
+            try {
+                const userObject = JSON.parse(event.data.payload);
+                const PREMIUM_PAGE_ID = 1149636; // شناسه محصول
+                // بررسی دسترسی کاربر
+                if (userObject && userObject.isLogin && userObject.accessible_pages && 
+                   (userObject.accessible_pages.includes(PREMIUM_PAGE_ID) || userObject.accessible_pages.includes(String(PREMIUM_PAGE_ID)))) {
+                    window.userSubscriptionStatus = 'paid';
+                } else {
+                    window.userSubscriptionStatus = 'free';
+                }
+                console.log("User Status:", window.userSubscriptionStatus);
+            } catch (e) {
+                console.error("Error parsing user status:", e);
+                window.userSubscriptionStatus = 'free';
             }
         }
-        if (event.data && event.data.type === 'NAVIGATE_TO_PREMIUM') {
-             // این رویداد توسط دکمه ارتقا تریگر میشه
-        }
     });
-
-    // فورس تم روشن
+    
+    // ارسال درخواست به پنجره والد
+    window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
+    
+    // --- Existing Logic (Theme, IP Reset, Scanner) ---
+    
+    // 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'; }
+        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);
+    forceLight();
+    setInterval(forceLight, 1000);
 
-    // اسکنر خطای Quota
+    // 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();
+    };
+
+    // 3. SHOW MODAL FUNCTION (IP Reset)
+    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">
+                        <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
     setInterval(() => {
         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';
+                 el.innerText = '';
+                 const parentWrap = el.closest('.toast-wrap');
+                 if(parentWrap) parentWrap.style.display = 'none';
             }
         });
     }, 100);
 });
-
-// --- 4. توابع کمکی ---
-window.retryGeneration = function() {
-    const modal = document.getElementById('custom-quota-modal'); if (modal) modal.remove();
-    const runBtn = document.getElementById('run-btn'); if(runBtn) runBtn.click();
-};
-window.closeErrorModal = function() {
-     const modal = document.getElementById('custom-quota-modal'); if (modal) modal.remove();
-};
-window.navigateToPremium = function() {
-    window.parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
-};
-
-const showQuotaModal = () => {
-    if (document.getElementById('custom-quota-modal')) return;
-    // (HTML مودال خطا همانند قبل است، خلاصه شده)
-    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;">
-        <div class="ip-reset-guide-container">
-            <h3>خطای محدودیت سرور</h3>
-            <p>لطفاً IP خود را تغییر دهید یا چند دقیقه صبر کنید.</p>
-            <button onclick="window.closeErrorModal()" class="action-button back-button">بازگشت</button>
-            <button onclick="window.retryGeneration()" class="action-button retry-button">تلاش مجدد</button>
-        </div>
-    </div>`;
-    document.body.insertAdjacentHTML('beforeend', modalHtml);
-    setTimeout(() => window.closeErrorModal(), 10000);
-};
 </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');
 
-/* Main Styling */
 :root, .dark, body, .gradio-container {
-    --body-background-fill: #f5f7fa !important; --body-text-color: #1f2937 !important;
+    --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; }
 
-/* Subscription Badge & Info */
-.subscription-info-container {
-    text-align: center; margin-bottom: 20px; padding: 15px;
-    background: white; border-radius: 16px; border: 1px solid #e5e7eb;
-    box-shadow: 0 4px 6px -1px rgba(0,0,0,0.05);
-    display: flex; flex-direction: column; align-items: center; gap: 10px;
+.back-button { 
+    background: white !important; 
+    color: var(--guide-text-body) !important; 
+    border: 2px solid #e2e8f0 !important; 
 }
-.badge-free {
-    background: linear-gradient(45deg, #6b7280, #4b5563); color: white;
-    padding: 6px 16px; border-radius: 20px; font-size: 0.9em; font-weight: 700;
+.back-button:hover { 
+    background: #f7fafc !important; 
+    border-color: var(--guide-accent) !important; 
+    transform: translateY(-2px); 
+    box-shadow: var(--shadow-md) !important; 
 }
-.badge-paid {
-    background: linear-gradient(45deg, #f59e0b, #d97706); color: white;
-    padding: 6px 16px; border-radius: 20px; font-size: 0.9em; font-weight: 700;
-    box-shadow: 0 4px 10px rgba(245, 158, 11, 0.3);
+
+.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;
 }
-.credit-text { color: #4b5563; font-size: 1rem; }
-.upgrade-btn {
-    background: linear-gradient(95deg, #FFD54F, #FFC107); color: #212529;
-    border: none; padding: 8px 20px; border-radius: 12px; font-weight: 800; cursor: pointer;
-    box-shadow: 0 4px 15px rgba(255, 193, 7, 0.3); transition: transform 0.2s;
-    display: none; /* Hidden by default, shown by JS */
+
+#main-description {
+    text-align: center;
+    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;
+}
+
+.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);
 }
-.upgrade-btn:hover { transform: translateY(-2px); }
 
-/* Buttons & Inputs */
-.primary-btn {
-    background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important; color: white !important;
-    border-radius: 14px !important; box-shadow: 0 4px 15px rgba(16, 185, 129, 0.3) !important;
+#download-btn {
+    background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
+    box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3) !important;
 }
-/* Modal Styles (Simplified for brevity) */
-.ip-reset-guide-container { 
-    background: white; padding: 20px; border-radius: 16px; width: 90%; max-width: 400px; text-align: right; direction: rtl;
+#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;
+}
+
+@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
 
-# --- 5. رابط کاربری Gradio ---
-with gr.Blocks(title="ویرایشگر هوشمند آلفا") as demo:
+# استفاده از gr.Blocks
+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 class="subscription-info-container">
-            <div id="subscription-badge" class="badge-free">نسخه رایگان</div>
-            <div id="credit-info-text" class="credit-text">در حال بررسی اعتبار...</div>
-            <button id="upgrade-btn-custom" class="upgrade-btn" onclick="navigateToPremium()">⭐️ ارتقا به نسخه کامل و نامحدود</button>
-            <div style="font-size: 0.85em; color: #6b7280; margin-top: 5px;">
-                نسخه رایگان: ۵ تصویر در روز | نسخه نامحدود: بدون محدودیت
+        gr.Markdown(
+            "با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", 
+            elem_id="main-description"
+        )
+        # --- متن اضافه شده برای مشخصات کاربران ---
+        gr.HTML(
+            """
+            <div style="text-align: center; margin-bottom: 20px; font-size: 0.9em; font-family: 'Vazirmatn', sans-serif;">
+                <span style="color: #666; margin: 0 10px; background: #f0f0f0; padding: 4px 10px; border-radius: 12px;">👤 نسخه رایگان: ۵ تصویر در روز</span>
+                <span style="color: #059669; margin: 0 10px; font-weight: bold; background: #d1fae5; padding: 4px 10px; border-radius: 12px;">⭐️ نسخه نامحدود: دسترسی کامل</span>
             </div>
-        </div>
-        """)
-        # ----------------------------------------------------
+            """
+        )
 
         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="وضعیت")
                 
-                # دکمه اجرا
+                # 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")
                 
-                lora_adapter = gr.Dropdown(label="انتخاب سبک ویرایش (LoRA)", choices=list(LORA_MAPPING.keys()), value="تبدیل عکس به انیمه")
+                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="تبدیل عکس به انیمه"
+                    )
                 
-                with gr.Accordion("تنظیمات پیشرفته", open=False):
-                    aspect_ratio_selection = gr.Dropdown(label="ابعاد تصویر", choices=ASPECT_RATIOS_LIST, 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="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(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)
         
-        # مدیریت نمایش اسلایدرهای ابعاد
-        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)
+        # --- کامپوننت‌های مخفی برای انتقال اطلاعات کاربر ---
+        hidden_fingerprint = gr.Textbox(visible=False)
+        hidden_status = gr.Textbox(visible=False)
+
+        # اصلاح تابع نمایش ردیف اسلایدرها
+        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
+        )
 
-        # مثال‌ها
         gr.Examples(
             examples=[
                 ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
-                ["examples/5.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,
-            label="نمونه‌ها"
-        )
-
-        # --- اتصال JS به دکمه اجرا برای ارسال Fingerprint و Status ---
-        # این تابع JS مقادیر ورودی‌ها + متغیرهای گلوبال (Fingerprint, Status) را می‌خواند و به تابع پایتون می‌فرستد
-        js_click_handler = """
-        (img, p, lora, s, rs, gs, st, ar, cw, ch) => {
-            return [img, p, lora, s, rs, gs, st, ar, cw, ch, window.userFingerprint, window.userSubscriptionStatus];
-        }
-        """
-
-        run_button.click(
-            fn=infer,
-            inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height],
-            outputs=[output_image, seed, status_box],
-            js=js_click_handler
+            cache_examples=False,
+            label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-        download_button.click(fn=None, inputs=[output_image], js=js_download_func)
+    # جاوا اسکریپت برای تزریق فینگرپرینت و وضعیت به ورودی‌های تابع پایتون
+    # آرگومان‌های ورودی تابع js باید با inputs همخوانی داشته باشند (به جز دو مورد آخر که اضافه می‌کنیم)
+    js_click_handler = """
+    (img, p, lora, s, rs, gs, steps, ar, cw, ch, fp, stat) => {
+        return [
+            img, p, lora, s, rs, gs, steps, ar, cw, ch,
+            window.userFingerprint, 
+            window.userSubscriptionStatus
+        ];
+    }
+    """
+
+    run_button.click(
+        fn=infer,
+        inputs=[input_image, prompt, lora_adapter_persian, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, hidden_fingerprint, hidden_status],
+        outputs=[output_image, seed, status_box],
+        js=js_click_handler,
+        api_name="predict"
+    )
+
+    download_button.click(
+        fn=None,
+        inputs=[output_image],
+        outputs=None,
+        js=js_download_func
+    )
 
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(show_error=True)