Update app.py

app.py

@@ -4,44 +4,14 @@ import numpy as np
 import spaces
 import torch
 import random
+import json
+import hashlib
+from datetime import datetime
 from PIL import Image, ImageFilter
-from typing import Iterable
 from gradio.themes import Soft
-from gradio.themes.utils import colors, fonts, sizes
+from gradio.themes.utils import colors
 from deep_translator import GoogleTranslator
 from transformers import pipeline
-from datetime import datetime
-
-# --- متغیرهای سراسری برای مدیریت محدودیت ---
-USAGE_TRACKER = {}  # ذخیره: {ip: {'count': 0, 'date': 'YYYY-MM-DD'}}
-DAILY_LIMIT = 5     # محدودیت تعداد عکس در روز برای کاربران رایگان
-
-# --- تعریف لیست کلمات ممنوعه (سخت‌گیرانه) ---
-# این کلمات در متن ترجمه شده (انگلیسی) جستجو می‌شوند
-BANNED_WORDS = [
-    # Nudity & Sexual content
-    "nsfw", "nude", "naked", "sex", "sexy", "sexual", "porn", "porno", "pornography",
-    "erotic", "erotica", "xxx", "18+", "uncensored", "hentai", "ecchi", "r34", "rule34",
-    "fetish", "bondage", "bdsm", "kink", "sadism", "masochism",
-    "lingerie", "bikini", "swimwear", "underwear", "panties", "bra", "topless", "bottomless",
-    "upskirt", "exhibitionism", "voyeur", "orgasm", "seductive", "sensual",
-    
-    # Body parts (Specific context)
-    "breast", "boobs", "tits", "nipple", "areola", "cleavage",
-    "genital", "vagina", "pussy", "cunt", "clitoris", "vulva",
-    "penis", "dick", "cock", "phallus", "boner", "erection",
-    "ass", "butt", "booty", "anal", "anus", "rectum", "glute",
-    "groin", "crotch", "pubic",
-    
-    # Sexual Acts
-    "blowjob", "bj", "handjob", "rimming", "fingering", "fucking", "screwing",
-    "masturbation", "masturbating", "sucking", "licking", "penetration", "intercourse",
-    "cum", "sperm", "ejaculation", "bukkake", "creampie", "gangbang", "orgy", "threesome",
-    
-    # Violence & Gore (Optional but recommended for strict safety)
-    "gore", "blood", "bloody", "kill", "murder", "death", "torture", "suicide",
-    "dead", "corpse", "mutilation", "decapitation", "amputee"
-]
 
 # --- تعریف تم ---
 colors.steel_blue = colors.Color(
@@ -61,7 +31,7 @@ colors.steel_blue = colors.Color(
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) برای تصاویر ---
+# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
 print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
@@ -77,15 +47,13 @@ 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
 from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 
 dtype = torch.bfloat16
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
 print("Loading pipeline...")
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "Qwen/Qwen-Image-Edit-2509",
@@ -122,14 +90,6 @@ LORA_MAPPING = {
     "افزایش کیفیت (Upscale)": "upscale-image"
 }
 
-ASPECT_RATIOS_LIST = [
-    "خودکار (پیش‌فرض)",
-    "۱:۱ (مربع - 1024x1024)",
-    "۱۶:۹ (افقی - 1344x768)",
-    "۹:۱۶ (عمودی - 768x1344)",
-    "شخصی‌سازی (Custom)"
-]
-
 ASPECT_RATIOS_MAP = {
     "خودکار (پیش‌فرض)": "Auto",
     "۱:۱ (مربع - 1024x1024)": (1024, 1024),
@@ -138,32 +98,84 @@ ASPECT_RATIOS_MAP = {
     "شخصی‌سازی (Custom)": "Custom"
 }
 
-def check_text_safety(text):
+# --- لیست کلمات ممنوعه (سختگیرانه و کامل) ---
+# این لیست شامل کلمات جنسی، خشونت آمیز و نامناسب است
+BANNED_WORDS = [
+    "nsfw", "nude", "naked", "sex", "sexy", "porn", "erotic", "xxx", 
+    "breast", "nipple", "genital", "vagina", "penis", "ass", "butt", "sexual",
+    "lingerie", "bikini", "swimwear", "underwear", "fetish", "topless",
+    "exhibitionism", "hentai", "ecchi", "18+", "hot girl", "sensual",
+    "intercourse", "bdsm", "bondage", "anus", "pussy", "dick", "cock", 
+    "cum", "orgasm", "masturbation", "slut", "whore", "milf", "orgy", 
+    "penetration", "striptease", "gore", "blood", "violence", "suicide",
+    "kill", "death", "murder", "torture", "abuse", "rapist", "rape",
+    "cleavage", "areola", "pubic", "upskirt", "intimate"
+]
+
+# --- سیستم مدیریت اعتبار (حافظه موقت) ---
+# ساختار: { "ip_fingerprint_hash": { "date": "YYYY-MM-DD", "count": 0 } }
+USAGE_DB = {}
+MAX_FREE_DAILY_LIMIT = 5
+
+def get_quota_status(fingerprint_hash, is_paid, request: gr.Request):
     """
-    بررسی سخت‌گیرانه متن برای کل��ات ممنوعه
+    بررسی وضعیت اعتبار کاربر.
+    اگر کاربر پولی باشد، همیشه مجاز است.
+    اگر رایگان باشد، بر اساس ترکیب IP و Fingerprint بررسی می‌شود.
     """
-    if not text: return True
-    text_lower = text.lower()
+    if is_paid:
+        return True, "نامحدود", 0
+
+    # ترکیب IP و Fingerprint برای جلوگیری از تغییر IP ساده
+    client_ip = request.client.host if request else "unknown"
+    # ساخت کلید یکتا برای کاربر
+    user_key = f"{client_ip}_{fingerprint_hash}"
     
-    # حذف علائم نگارشی برای بررسی دقیق‌تر
-    import string
-    translator = str.maketrans('', '', string.punctuation)
-    clean_text = text_lower.translate(translator)
+    current_date = datetime.now().strftime("%Y-%m-%d")
     
-    words = clean_text.split()
+    # دریافت یا ایجاد رکورد کاربر
+    user_data = USAGE_DB.get(user_key, {"date": current_date, "count": 0})
     
-    for banned in BANNED_WORDS:
-        # بررسی دقیق کلمه (Exact Match)
-        if banned in words:
+    # ریست کردن در روز جدید
+    if user_data["date"] != current_date:
+        user_data = {"date": current_date, "count": 0}
+    
+    # بررسی محدودیت
+    if user_data["count"] >= MAX_FREE_DAILY_LIMIT:
+        return False, "اتمام اعتبار", 0
+    
+    remaining = MAX_FREE_DAILY_LIMIT - user_data["count"]
+    return True, "مجاز", remaining
+
+def increment_usage(fingerprint_hash, is_paid, request: gr.Request):
+    if is_paid: return
+    
+    client_ip = request.client.host if request else "unknown"
+    user_key = f"{client_ip}_{fingerprint_hash}"
+    current_date = datetime.now().strftime("%Y-%m-%d")
+    
+    user_data = USAGE_DB.get(user_key, {"date": current_date, "count": 0})
+    if user_data["date"] != current_date:
+        user_data = {"date": current_date, "count": 0}
+        
+    user_data["count"] += 1
+    USAGE_DB[user_key] = user_data
+
+# --- توابع کمکی ---
+
+def check_text_safety(text):
+    text_lower = text.lower()
+    # بررسی دقیق کلمات (با فاصله برای جلوگیری از تطابق اشتباه)
+    for word in BANNED_WORDS:
+        # چک کردن کلمه به صورت جداگانه
+        if f" {word} " in f" {text_lower} " or \
+           text_lower.startswith(f"{word} ") or \
+           text_lower.endswith(f" {word}") or \
+           text_lower == word:
             return False
-        # بررسی وجود کلمه در داخل کلمات دیگر (Substring Match) برای کلمات خاص و خطرناک
-        # مثلاً اگر کاربر بنویسد "supersexy" باید شناسایی شود
-        if len(banned) > 3 and banned in text_lower:
-             # برای جلوگیری از خطای مثبت (مثلاً "ass" در "class")
-             # اینجا یک بررسی ساده انجام می‌دهیم، اما برای کلمات خیلی رکیک سخت‌گیر هستیم
-             if banned in ["sex", "porn", "fuck", "nude", "naked", "xxx"]:
-                 return False
-                 
+        # چک کردن کلمات ترکیبی خطرناک
+        if word in text_lower and len(word) > 4: # برای کلمات طولانی‌تر سختگیرانه‌تر
+             return False
     return True
 
 def translate_prompt(text):
@@ -192,15 +204,9 @@ def update_dimensions_on_upload(image):
     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;">
+    <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;">
         <span style="font-size: 1.2em;">⛔</span>
         {message}
     </div>
@@ -208,58 +214,13 @@ def get_error_html(message):
 
 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;">
+    <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; direction: rtl;">
         <span style="font-size: 1.2em;">✅</span>
         {message}
     </div>
     """
 
-def check_and_update_quota(request: gr.Request, is_vip: bool):
-    """
-    بررسی محدودیت کاربر و بروزرسانی شمارنده.
-    اگر VIP باشد، محدودیت ندارد.
-    اگر رایگان باشد، بر اساس IP محدودیت ۵ بار در روز اعمال می‌شود.
-    """
-    if is_vip:
-        return True, ""
-        
-    try:
-        client_ip = request.client.host
-    except:
-        client_ip = "unknown"
-        
-    today = datetime.now().strftime("%Y-%m-%d")
-    
-    # مقداردهی اولیه برای IP جدید
-    if client_ip not in USAGE_TRACKER:
-        USAGE_TRACKER[client_ip] = {'count': 0, 'date': today}
-    
-    user_data = USAGE_TRACKER[client_ip]
-    
-    # ریست کردن اگر روز تغییر کرده باشد
-    if user_data['date'] != today:
-        user_data['count'] = 0
-        user_data['date'] = today
-        
-    # بررسی سقف مجاز
-    if user_data['count'] >= DAILY_LIMIT:
-        return False, f"شما به سقف مجاز روزانه ({DAILY_LIMIT} تصویر) رسیده‌اید. فردا مجدداً تلاش کنید یا حساب خود را ارتقا دهید."
-        
-    return True, ""
-
-def increment_quota(request: gr.Request, is_vip: bool):
-    """
-    افزایش شمارنده پس از موفقیت آمیز بودن عملیات
-    """
-    if is_vip: return
-    try:
-        client_ip = request.client.host
-        if client_ip in USAGE_TRACKER:
-            USAGE_TRACKER[client_ip]['count'] += 1
-    except:
-        pass
-
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=45)
 def infer(
     input_image,
     prompt,
@@ -271,26 +232,31 @@ def infer(
     aspect_ratio_selection,
     custom_width,
     custom_height,
-    is_vip_state, # ورودی جدید برای وضعیت VIP
+    fingerprint_hash,
+    is_paid_user_str,
     request: gr.Request,
     progress=gr.Progress(track_tqdm=True)
 ):
-    # --- مرحله ۱: بررسی سهمیه (Quota) ---
-    allowed, message = check_and_update_quota(request, is_vip_state)
+    # تبدیل وضعیت کاربر از رشته به بولین
+    is_paid = is_paid_user_str == "true"
+    
+    # 1. بررسی اعتبار قبل از هر کاری
+    allowed, status_msg, remaining = get_quota_status(fingerprint_hash, is_paid, request)
+    
     if not allowed:
-        return None, seed, get_error_html(message)
+        return None, seed, get_error_html("اعتبار ساخت تصویر شما برای امروز تمام شده است. لطفاً برای استفاده نامحدود حساب خود را ارتقا دهید."), f"0 اعتبار"
 
     if input_image is None:
-        return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
+        return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید."), f"{remaining} اعتبار"
 
-    # --- مرحله ۲: بررسی تصویر ورودی (Safety Check) ---
+    # 2. بررسی ایمنی تصویر ورودی
     if is_image_nsfw(input_image):
-        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
+        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود."), f"{remaining} اعتبار"
 
-    # --- مرحله ۳: ترجمه و بررسی متن (Safety Check) ---
+    # 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("متن درخواست شامل کلمات غیرمجاز، غیراخلاقی یا جنسی است."), f"{remaining} اعتبار"
 
     adapter_internal_name = LORA_MAPPING.get(lora_adapter_persian)
     if adapter_internal_name:
@@ -318,7 +284,6 @@ def infer(
         width, height = selection_value
 
     try:
-        # --- تولید تصویر ---
         result = pipe(
             image=original_image,
             prompt=english_prompt,
@@ -330,85 +295,109 @@ def infer(
             true_cfg_scale=guidance_scale,
         ).images[0]
 
-        # --- مرحله ۴: بررسی تصویر خروجی (Safety Check) ---
+        # 4. بررسی ایمنی تصویر خروجی
         if is_image_nsfw(result):
-            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد."), f"{remaining} اعتبار"
+        
+        # 5. کسر اعتبار (فقط اگر همه چیز موفق بود)
+        increment_usage(fingerprint_hash, is_paid, request)
         
-        # افزایش شمارنده مصرف فقط در صورت موفقیت
-        increment_quota(request, is_vip_state)
+        # محاسبه مجدد باقی مانده برای نمایش دقیق
+        _, _, new_remaining = get_quota_status(fingerprint_hash, is_paid, request)
+        credit_text = "نامحدود" if is_paid else f"{new_remaining} اعتبار برای امروز"
         
-        return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
+        return result, seed, get_success_html("تصویر با موفقیت ویرایش شد."), credit_text
 
     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}")
-
-@spaces.GPU(duration=30)
-def infer_example(input_image, prompt, lora_adapter):
-    # برای مثال‌ها محدودیت VIP بررسی نمی‌شود و IP لوکال فرض می‌شود
-    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, True, None)
-    return res, s, status
-
-# --- جاوااسکریپت برای دکمه دانلود ---
-js_download_func = """
-async (image) => {
-    if (!image) {
-        alert("لطفاً ابتدا تصویر را تولید کنید.");
-        return;
+        return None, seed, get_error_html(f"خطا در پردازش: {error_str}"), f"{remaining} اعتبار"
+
+# --- جاوااسکریپت برای دانلود و ارتباطات ---
+js_func = """
+<script>
+// --- پیکربندی پرداخت ---
+const PREMIUM_PAGE_ID = '1149636'; 
+const PREMIUM_URL = '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991';
+
+// --- تولید Fingerprint مرورگر ---
+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("alpha_img_edit_fp_v1", 2, 15);
+        components.push(canvas.toDataURL());
+    } catch (e) {
+        components.push("canvas-error");
     }
-    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;
+    const fingerprintString = components.join('~~~');
+    
+    // Simple hash function
+    let hash = 0;
+    for (let i = 0; i < fingerprintString.length; i++) {
+        const char = fingerprintString.charCodeAt(i);
+        hash = ((hash << 5) - hash) + char;
+        hash |= 0;
     }
-    window.parent.postMessage({
-        type: 'DOWNLOAD_REQUEST',
-        url: fileUrl
-    }, '*');
+    return 'fp_' + Math.abs(hash).toString(16);
 }
-"""
 
-# --- جاوااسکریپت سراسری (استراتژی موبایل + بستن خودکار ۱۰ ثانیه + تشخیص VIP) ---
-js_global_content = """
-<script>
-// Global variable to store VIP status
-window.userIsVip = false;
-const PREMIUM_PAGE_ID = '1149635'; // شناسه صفحه پرداخت برای بررسی دسترسی
+// --- مدیریت ارتباط با والد (Android/Web) ---
+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)));
+}
 
-// Request User Status on Load
-setTimeout(() => {
-    window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
-}, 1000);
-
-// Listener for User Status
-window.addEventListener('message', (event) => {
-    if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
-        try {
-            const user = JSON.parse(event.data.payload);
-            // بررسی دسترسی کاربر به صفحه پرمیوم
-            if (user && user.isLogin && user.accessible_pages &&
-               (user.accessible_pages.includes(PREMIUM_PAGE_ID) || 
-                user.accessible_pages.includes(parseInt(PREMIUM_PAGE_ID)))) {
-                
-                window.userIsVip = true;
-                
-                // نمایش نشان VIP
-                const badge = document.createElement('div');
-                badge.innerHTML = '👑 نسخه ویژه (نامحدود)';
-                badge.style.cssText = 'position:fixed;top:10px;left:10px;background:linear-gradient(45deg, #FFD700, #FFC107);color:#333;padding:6px 12px;border-radius:20px;z-index:9999;font-weight:bold;box-shadow:0 2px 5px rgba(0,0,0,0.2);font-family:"Vazirmatn",sans-serif;font-size:0.8rem;';
-                document.body.appendChild(badge);
-                
-                // حذف مدال محدودیت اگر باز باشد
-                window.closeErrorModal();
-            }
-        } catch (e) { console.log('Error parsing user status:', e); }
+function updateUI(isPaid) {
+    const badge = document.getElementById('subscription-status-badge');
+    const upgradeBtn = document.getElementById('upgrade-premium-btn');
+    const hiddenStatus = document.querySelector('#is-paid-user textarea');
+    
+    // بروزرسانی مقدار مخفی برای پایتون
+    if(hiddenStatus) {
+        hiddenStatus.value = isPaid ? "true" : "false";
+        hiddenStatus.dispatchEvent(new Event('input'));
     }
-});
 
-document.addEventListener('DOMContentLoaded', () => {
+    if (isPaid) {
+        if(badge) {
+            badge.textContent = 'نسخه نامحدود';
+            badge.className = 'paid-badge';
+            badge.style.display = 'inline-block';
+        }
+        if(upgradeBtn) upgradeBtn.style.display = 'none';
+        
+        // Hide credits text if visible
+        const creditDisplay = document.getElementById('credit-display-text');
+        if(creditDisplay) creditDisplay.textContent = 'نامحدود';
+        
+    } else {
+        if(badge) {
+            badge.textContent = 'نسخه رایگان';
+            badge.className = 'free-badge';
+            badge.style.display = 'inline-block';
+        }
+        if(upgradeBtn) upgradeBtn.style.display = 'block';
+    }
+}
+
+// --- مقداردهی اولیه ---
+document.addEventListener('DOMContentLoaded', async () => {
+    
     // 1. Force Light Mode
     const forceLight = () => {
         const body = document.querySelector('body');
@@ -422,115 +411,65 @@ document.addEventListener('DOMContentLoaded', () => {
     forceLight();
     setInterval(forceLight, 1000);
 
-    // 2. RETRY FUNCTION
-    window.retryGeneration = function() {
-        const modal = document.getElementById('custom-quota-modal');
-        if (modal) modal.remove();
+    // 2. Fingerprint Setup
+    const fp = await getBrowserFingerprint();
+    const fpInput = document.querySelector('#fingerprint-input textarea');
+    if(fpInput) {
+        fpInput.value = fp;
+        fpInput.dispatchEvent(new Event('input'));
+    }
 
-        const runBtn = document.getElementById('run-btn');
-        if(runBtn) runBtn.click();
-    };
+    // 3. Request User Status
+    window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
     
-    // Close function
-    window.closeErrorModal = function() {
-         const modal = document.getElementById('custom-quota-modal');
-         if (modal) modal.remove();
+    // 4. Listen for Response
+    window.addEventListener('message', (event) => {
+        if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
+            try {
+                const userObject = JSON.parse(event.data.payload);
+                const isPaid = isUserPaid(userObject);
+                updateUI(isPaid);
+            } catch (e) {
+                console.error("Error parsing user status", e);
+                updateUI(false);
+            }
+        }
+    });
+
+    // 5. Upgrade Button Logic
+    window.triggerUpgrade = function() {
+        window.parent.postMessage({ 
+            type: 'NAVIGATE_TO_PREMIUM', 
+            payload: { url: PREMIUM_URL } 
+        }, '*');
     };
-
-    // 3. SHOW MODAL FUNCTION
+    
+    // 6. Download Logic
+    window.downloadImage = 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
+        }, '*');
+    }
+    
+    // 7. Error Modal Logic (Quota Exceeded from Gradio Backend)
     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);
+        // ساخت مودال مشابه پادکست نیست، بلکه مودال IP Reset قبلی است
+        // اما اینجا ما سیستم کردیت داریم، پس ارور کردیت را مدیریت می‌کنیم
+        // اگر ارور سمت سرور "Quota Exceeded" بود یعنی کاربر رایگان محدودیتش تمام شده
     };
-
-    // 4. SCANNER (For HuggingFace GPU Quota)
-    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();
-                 
-                 // Immediately hide the Gradio error
-                 el.style.display = 'none';
-                 el.style.opacity = '0';
-                 el.innerText = '';
-                 
-                 const parentWrap = el.closest('.toast-wrap');
-                 if(parentWrap) parentWrap.style.display = 'none';
-            }
-        });
-    }, 100);
 });
 </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');
@@ -538,168 +477,72 @@ css_code = """
 :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;
+    --app-font: 'Vazirmatn', sans-serif !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);
-}
+body { font-family: var(--app-font); direction: rtl; }
 
-@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); }
-}
+/* --- Badges & Buttons Styles (From Podcast Code) --- */
+@keyframes badge-fade-in { from { opacity: 0; transform: translateY(-10px) scale(0.95); } to { opacity: 1; transform: translateY(0) scale(1); } }
 
-.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;
+#subscription-status-badge {
+    display: none; /* Controlled by JS */
+    padding: 6px 16px;
+    border-radius: 20px;
+    font-size: 0.9em;
+    font-weight: 700;
+    margin-top: 1rem;
+    margin-bottom: 1rem;
+    letter-spacing: 0.5px;
+    text-shadow: 0 1px 2px rgba(0,0,0,0.1);
+    animation: badge-fade-in 0.6s 0.5s ease-out backwards;
 }
-.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; 
+.free-badge {
+    background: linear-gradient(45deg, #6c757d, #495057);
+    color: white;
+    box-shadow: 0 4px 10px rgba(108, 117, 125, 0.3);
 }
-.back-button:hover { 
-    background: #f7fafc !important; 
-    border-color: var(--guide-accent) !important; 
-    transform: translateY(-2px); 
-    box-shadow: var(--shadow-md) !important; 
+
+.paid-badge {
+    background: linear-gradient(45deg, #FFC107, #ffca2c);
+    color: #333;
+    box-shadow: 0 4px 10px rgba(255, 193, 7, 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; 
+@keyframes upgrade-btn-pulse {
+    0% { transform: scale(1); box-shadow: 0 8px 20px -5px rgba(255, 193, 7, 0.3); }
+    50% { transform: scale(1.02); box-shadow: 0 12px 25px -5px rgba(255, 193, 7, 0.3); }
+    100% { transform: scale(1); box-shadow: 0 8px 20px -5px rgba(255, 193, 7, 0.3); }
 }
-.retry-button:hover { 
-    transform: translateY(-2px); 
-    box-shadow: 0 6px 20px rgba(102, 126, 234, 0.4) !important; 
+
+#upgrade-premium-btn {
+    display: none; /* Controlled by JS */
+    width: 100%;
+    margin-top: 1.5rem;
+    padding: 1rem;
+    font-family: var(--app-font);
+    font-size: 1.1em;
+    font-weight: 800;
+    color: #212529;
+    background: linear-gradient(95deg, #FFD54F, #FFC107 100%);
+    border: none;
+    border-radius: 14px;
+    cursor: pointer;
+    transition: transform 0.2s ease, box-shadow 0.2s ease;
+    box-shadow: 0 8px 20px -5px rgba(255, 193, 7, 0.3);
+    animation: upgrade-btn-pulse 2.5s infinite;
+    text-align: center;
 }
 
-/* --- Main App CSS --- */
-body {
-    font-family: 'Vazirmatn', sans-serif !important;
-    background-color: #f5f7fa !important;
-    margin: 0;
-    padding: 10px;
+#upgrade-premium-btn:hover {
+    transform: translateY(-3px);
+    box-shadow: 0 12px 25px -5px rgba(255, 193, 7, 0.4);
+    animation-play-state: paused;
 }
 
+/* --- Layout --- */
 #col-container {
     margin: 0 auto;
     max-width: 980px;
@@ -716,7 +559,7 @@ body {
     font-size: 2.4em !important;
     text-align: center;
     color: #1a202c !important;
-    margin-bottom: 15px;
+    margin-bottom: 5px;
     font-weight: 800;
     background: -webkit-linear-gradient(45deg, #2563eb, #1e40af);
     -webkit-background-clip: text;
@@ -727,39 +570,11 @@ body {
     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;
-}
-
-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;
+    margin-bottom: 10px;
 }
 
-.gr-dropdown {
-    background: #ffffff !important;
-    border-radius: 12px !important;
-}
-
-.primary-btn, button.primary {
+/* --- Gradio Overrides --- */
+.primary-btn {
     background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important;
     border: none !important;
     color: white !important;
@@ -768,106 +583,43 @@ textarea:focus, input[type="text"]:focus {
     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);
 }
+.primary-btn:hover { transform: translateY(-2px); }
 
 #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;
+textarea, input[type="text"] {
+    font-family: var(--app-font) !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;
+    border: 2px solid #e2e8f0 !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;
-    }
+/* Toast/Error hiding */
+.toast-body, .toast-wrap { 
+    display: none !important; 
+    opacity: 0 !important; 
 }
 </style>
 """
 
 # ادغام CSS و JS
-combined_html = css_code + js_global_content
+combined_html = css_code + js_func
 
-# استفاده از gr.Blocks
-with gr.Blocks() as demo:
-    # تزریق کدها به عنوان HTML
+with gr.Blocks(title="Alpha Image Editor") as demo:
     gr.HTML(combined_html)
     
+    # متغیرهای مخفی برای ارتباط با JS
+    fingerprint_hash = gr.Textbox(elem_id="fingerprint-input", visible=False)
+    is_paid_user = gr.Textbox(elem_id="is-paid-user", value="false", visible=False)
+    
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
         gr.Markdown(
@@ -875,15 +627,10 @@ with gr.Blocks() as demo:
             elem_id="main-description"
         )
         
-        # --- بخش نمایش وضعیت اشتراک (مشابه پادکست) ---
+        # بخش نشان وضعیت کاربر
         gr.HTML("""
-        <div style="display: flex; gap: 10px; justify-content: center; margin-bottom: 20px; flex-wrap: wrap;">
-            <span style="background: #e2e8f0; color: #475569; padding: 6px 16px; border-radius: 20px; font-size: 0.9em; font-weight: 700; display: flex; align-items: center; gap: 6px;">
-                👤 برای کاربران رایگان: ۵ تصویر در روز
-            </span>
-            <span style="background: linear-gradient(90deg, #ffd700, #ffb700); color: #333; padding: 6px 16px; border-radius: 20px; font-size: 0.9em; font-weight: 700; box-shadow: 0 4px 10px rgba(255, 193, 7, 0.3); display: flex; align-items: center; gap: 6px;">
-                👑 نسخه نامحدود: برای کاربران پولی
-            </span>
+        <div style="text-align: center;">
+            <div id="subscription-status-badge">در حال بررسی...</div>
         </div>
         """)
 
@@ -899,13 +646,17 @@ with gr.Blocks() as demo:
                     lines=3
                 )
                 
+                credit_status = gr.Textbox(label="اعتبار باقی‌مانده", value="...", interactive=False, elem_id="credit-display-text")
                 status_box = gr.HTML(label="وضعیت")
                 
-                # Checkbox مخفی برای انتقال وضعیت VIP از جاوااسکریپت به پایتون
-                is_vip_state = gr.Checkbox(visible=False, value=False, label="VIP State")
-
-                # IMPORTANT: Added elem_id="run-btn" here for JS targeting
                 run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn", elem_id="run-btn")
+                
+                # دکمه ارتقا (توسط JS کنترل می‌شود)
+                gr.HTML("""
+                <button type="button" id="upgrade-premium-btn" onclick="window.triggerUpgrade()">
+                    ⭐️ ارتقا به نسخه کامل و نامحدود
+                </button>
+                """)
 
             with gr.Column():
                 output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
@@ -922,81 +673,37 @@ with gr.Blocks() as demo:
                 with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
                     aspect_ratio_selection = gr.Dropdown(
                         label="ابعاد تصویر خروجی",
-                        choices=ASPECT_RATIOS_LIST,
+                        choices=list(ASPECT_RATIOS_MAP.keys()),
                         value="خودکار (پیش‌فرض)",
                         interactive=True
                     )
                     
                     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
-                        )
+                        custom_width = gr.Slider(minimum=256, maximum=2048, step=8, value=1024, label="Width")
+                        custom_height = gr.Slider(minimum=256, maximum=2048, step=8, value=1024, label="Height")
                     
                     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)
+                    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):
-            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
-        )
+            return gr.update(visible=(choice == "شخصی‌سازی (Custom)"))
 
-        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="نمونه‌ها (برای تست کلیک کنید)"
-        )
-
-    # جاوااسکریپت برای تزریق متغیر سراسری JS به ورودی Checkbox پایتون
-    # این تابع مقدار window.userIsVip را می‌خواند و به عنوان آخرین آرگومان به تابع پایتون می‌فرستد
-    inject_vip_js = """
-    (img, p, lora, seed, rand, scale, steps, aspect, w, h, vip_dummy) => {
-        return [img, p, lora, seed, rand, scale, steps, aspect, w, h, window.userIsVip];
-    }
-    """
+        aspect_ratio_selection.change(fn=toggle_row, inputs=aspect_ratio_selection, outputs=custom_dims_row)
 
+    # اتصال رویداد کلیک
     run_button.click(
         fn=infer,
-        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, is_vip_state],
-        outputs=[output_image, seed, status_box],
-        js=inject_vip_js,
-        api_name="predict"
+        inputs=[
+            input_image, prompt, lora_adapter, seed, randomize_seed, 
+            guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height,
+            fingerprint_hash, is_paid_user
+        ],
+        outputs=[output_image, seed, status_box, credit_status]
     )
 
-    download_button.click(
-        fn=None,
-        inputs=[output_image],
-        outputs=None,
-        js=js_download_func
-    )
+    download_button.click(fn=None, inputs=[output_image], outputs=None, js="window.downloadImage")
 
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(show_error=True)