Update app.py

app.py

@@ -4,80 +4,14 @@ import numpy as np
 import spaces
 import torch
 import random
-import sqlite3
-import json
-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 deep_translator import GoogleTranslator
 from transformers import pipeline
-
-# --- تنظیمات دیتابیس برای مدیریت کردیت کاربران ---
-DB_NAME = "user_limits.db"
-
-def init_db():
-    """ایجاد جدول برای ذخیره تعداد استفاده کاربران"""
-    conn = sqlite3.connect(DB_NAME)
-    c = conn.cursor()
-    c.execute('''
-        CREATE TABLE IF NOT EXISTS usage_logs (
-            user_id TEXT,
-            date TEXT,
-            count INTEGER,
-            PRIMARY KEY (user_id, date)
-        )
-    ''')
-    conn.commit()
-    conn.close()
-
-init_db()
-
-def check_and_update_quota(user_id, is_premium):
-    """
-    بررسی مجاز بودن کاربر برای تولید تصویر
-    کاربران پریمیوم: همیشه مجاز
-    کاربران رایگان: حداکثر 5 تصویر در روز
-    """
-    if not user_id:
-        return True, "Guest" # اگر یوزر آیدی نبود، سختگیری نمی‌کنیم (یا می‌توان محدود کرد)
-
-    if is_premium:
-        return True, "نامحدود"
-
-    today = datetime.now().strftime("%Y-%m-%d")
-    
-    conn = sqlite3.connect(DB_NAME)
-    c = conn.cursor()
-    
-    c.execute("SELECT count FROM usage_logs WHERE user_id = ? AND date = ?", (user_id, today))
-    result = c.fetchone()
-    
-    current_count = result[0] if result else 0
-    limit = 5
-    
-    conn.close()
-    
-    if current_count >= limit:
-        return False, f"0 از {limit}"
-    
-    return True, f"{limit - current_count} از {limit}"
-
-def increment_usage(user_id, is_premium):
-    """افزایش شمارنده استفاده کاربر پس از تولید موفق"""
-    if not user_id or is_premium:
-        return
-
-    today = datetime.now().strftime("%Y-%m-%d")
-    conn = sqlite3.connect(DB_NAME)
-    c = conn.cursor()
-    
-    c.execute("INSERT OR IGNORE INTO usage_logs (user_id, date, count) VALUES (?, ?, 0)", (user_id, today))
-    c.execute("UPDATE usage_logs SET count = count + 1 WHERE user_id = ? AND date = ?", (user_id, today))
-    
-    conn.commit()
-    conn.close()
+import json
+from datetime import datetime
 
 # --- تعریف تم ---
 colors.steel_blue = colors.Color(
@@ -97,74 +31,21 @@ colors.steel_blue = colors.Color(
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# --- بارگذاری سیستم امنیتی دوگانه (Dual Safety System) ---
-print("Loading Safety Checkers...")
-
-# مدل اول: سریع و استاندارد
-safety_classifier_1 = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
-
-# مدل دوم: بسیار دقیق و سخت‌گیر (ViT)
-safety_classifier_2 = pipeline("image-classification", model="AdamCodd/vit-base-nsfw-detector", device=-1)
+# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
+print("Loading Safety Checker...")
+safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
 def is_image_nsfw(image):
     if image is None: return False
     try:
-        # بررسی با مدل اول
-        results1 = safety_classifier_1(image)
-        for result in results1:
-            if result['label'] == 'nsfw' and result['score'] > 0.5:
-                print(f"Safety Check 1 Failed: {result['score']}")
-                return True
-
-        # بررسی با مدل دوم
-        results2 = safety_classifier_2(image)
-        for result in results2:
-            label = result['label'].lower()
-            score = result['score']
-            
-            if label == 'nsfw' and score > 0.3:
-                print(f"Safety Check 2 (NSFW) Failed: {score}")
-                return True
-            
-            if label in ['sexy', 'porn', 'hentai'] and score > 0.4:
-                print(f"Safety Check 2 (Partial) Failed: {label} - {score}")
+        results = safety_classifier(image)
+        for result in results:
+            if result['label'] == 'nsfw' and result['score'] > 0.75:
                 return True
-
         return False
     except Exception as e:
         print(f"Safety check error: {e}")
-        return True
-
-# --- لیست کلمات ممنوعه ---
-BANNED_WORDS = [
-    "nsfw", "nude", "naked", "sex", "porn", "erotic", "xxx", "18+", "uncensored",
-    "breast", "nipple", "areola", "cleavage", "topless", "open chest",
-    "genital", "vagina", "penis", "dick", "cock", "pussy", "ass", "butt", "anus",
-    "lingerie", "bikini", "swimwear", "underwear", "panties", "bra",
-    "fetish", "bdsm", "bondage", "exhibitionism", "voyeur",
-    "hentai", "ecchi", "ahegao", "paizuri",
-    "undressed", "stripping", "naked body", "exposed skin", "sheer", "see-through",
-    "rape", "violence", "blood", "gore", "sexual"
-]
-
-def check_text_safety(text):
-    if not text: return True
-    text_lower = text.lower()
-    for word in BANNED_WORDS:
-        if word in text_lower: 
-            print(f"Banned word found: {word}")
-            return False
-    return True
-
-def translate_prompt(text):
-    if not text:
-        return ""
-    try:
-        translated = GoogleTranslator(source='auto', target='en').translate(text)
-        return translated
-    except Exception as e:
-        print(f"Translation Error: {e}")
-        return text
+        return False
 
 # --- بارگذاری مدل اصلی ---
 from diffusers import FlowMatchEulerDiscreteScheduler
@@ -175,7 +56,7 @@ from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-print("Loading Generation Pipeline...")
+print("Loading pipeline...")
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "Qwen/Qwen-Image-Edit-2509",
     transformer=QwenImageTransformer2DModel.from_pretrained(
@@ -227,6 +108,30 @@ ASPECT_RATIOS_MAP = {
     "شخصی‌سازی (Custom)": "Custom"
 }
 
+BANNED_WORDS = [
+    "nude", "naked", "sex", "porn", "undressed", "nsfw", "erotic", "xxx", 
+    "breast", "nipple", "genital", "vagina", "penis", "ass", "butt", "sexual",
+    "lingerie", "bikini", "swimwear", "underwear", "fetish", "topless",
+    "exhibitionism", "hentai", "ecchi", "18+"
+]
+
+def check_text_safety(text):
+    text_lower = text.lower()
+    for word in BANNED_WORDS:
+        if f" {word} " in f" {text_lower} ": 
+            return False
+    return True
+
+def translate_prompt(text):
+    if not text:
+        return ""
+    try:
+        translated = GoogleTranslator(source='auto', target='en').translate(text)
+        return translated
+    except Exception as e:
+        print(f"Translation Error: {e}")
+        return text
+
 def update_dimensions_on_upload(image):
     if image is None:
         return 1024, 1024
@@ -243,6 +148,12 @@ 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;">
@@ -259,16 +170,6 @@ def get_success_html(message):
     </div>
     """
 
-def get_limit_error_html():
-    return f"""
-    <div style="background-color: #fff3cd; border: 1px solid #ffc107; color: #856404; padding: 16px; border-radius: 12px; text-align: center; margin-bottom: 15px; direction: rtl;">
-        <h3 style="margin: 0 0 10px 0;">🚫 محدودیت اعتبار روزانه</h3>
-        <p style="margin: 0;">شما از سهمیه ۵ تصویر رایگان امروز خود استفاده کرده‌اید.</p>
-        <p style="margin: 8px 0 0 0; font-size: 0.9em;">برای استفاده نامحدود، حساب خود را به نسخه ویژه ارتقا دهید.</p>
-        <button onclick="window.parent.postMessage({{type: 'NAVIGATE_TO_PREMIUM'}}, '*')" style="margin-top: 12px; background: linear-gradient(90deg, #FFD700, #FFA500); border: none; padding: 8px 16px; border-radius: 20px; color: black; font-weight: bold; cursor: pointer;">ارتقا به نسخه ویژه ⭐️</button>
-    </div>
-    """
-
 @spaces.GPU(duration=30)
 def infer(
     input_image,
@@ -281,26 +182,14 @@ def infer(
     aspect_ratio_selection,
     custom_width,
     custom_height,
-    user_id_hidden, # دریافت شناسه کاربر از فرانت
-    is_premium_hidden, # دریافت وضعیت پریمیوم از فرانت
     progress=gr.Progress(track_tqdm=True)
 ):
-    # --- بررسی اعتبار کاربر ---
-    is_premium = str(is_premium_hidden).lower() == "true"
-    allowed, limit_msg = check_and_update_quota(user_id_hidden, is_premium)
-    
-    if not allowed:
-        return None, seed, get_limit_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("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
 
-    # --- ترجمه و بررسی متن ---
     english_prompt = translate_prompt(prompt)
     if not check_text_safety(english_prompt):
         return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است.")
@@ -314,13 +203,7 @@ def infer(
 
     generator = torch.Generator(device=device).manual_seed(seed)
     
-    safety_negative = (
-        "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, areola, genital, "
-        "vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, "
-        "topless, open chest, revealing clothes, cleavage, see-through, sheer, "
-        "gore, violence, blood, navel, midriff, exposed skin, erotic, ecchi, hentai, "
-        "uncensored, stripping"
-    )
+    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}"
 
@@ -348,13 +231,9 @@ def infer(
             true_cfg_scale=guidance_scale,
         ).images[0]
 
-        # --- بررسی امنیتی خروجی ---
         if is_image_nsfw(result):
             return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
         
-        # --- کسر اعتبار (فقط برای کاربران رایگان) ---
-        increment_usage(user_id_hidden, is_premium)
-        
         return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
 
     except Exception as e:
@@ -365,8 +244,7 @@ def infer(
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    # برای مثال‌ها، آیدی کاربر فرضی و پریمیوم در نظر گرفته می‌شود تا محدودیت اعمال نشود
-    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "example_user", "true")
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024)
     return res, s, status
 
 # --- جاوااسکریپت برای دکمه دانلود ---
@@ -389,85 +267,362 @@ async (image) => {
 }
 """
 
-# --- جاوااسکریپت سراسری (شناسایی کاربر، پیام رفع محدودیت، دانلود) ---
-js_global_content = """
+# --- جاوااسکریپت مدیریت کاربر (رایگان/پولی) و محدودیت‌ها ---
+js_user_auth_logic = """
 <script>
-document.addEventListener('DOMContentLoaded', () => {
-    // --------------------------------------------------------
-    // 1. User Identification & Credit Logic (PostMessage)
-    // --------------------------------------------------------
-    const PREMIUM_PAGE_ID = '1149636'; 
-    let userId = "guest_" + Math.random().toString(36).substr(2, 9);
-    let isPremium = false;
-
-    // Listen for response from Parent (Android/Iframe)
-    window.addEventListener('message', (event) => {
-        if (event.data && event.data.type === 'USER_DATA_RESPONSE') {
-            const payload = event.data.payload;
-            
-            if (payload) {
-                try {
-                    const userObj = JSON.parse(payload);
-                    if (userObj.id) userId = userObj.id;
-                    
-                    // Check Premium Status
-                    if (userObj.isLogin && userObj.accessible_pages) {
-                        const pages = userObj.accessible_pages;
-                        // Check if premium page ID exists in allowed pages
-                        if (pages.includes(PREMIUM_PAGE_ID) || pages.includes(parseInt(PREMIUM_PAGE_ID))) {
-                            isPremium = true;
-                        }
-                    }
-                } catch (e) {
-                    console.error("Error parsing user data:", e);
-                }
-            }
-            updateUserStatusUI();
-            updateHiddenInputs();
-        }
-    });
+// --- تنظیمات پریمیوم ---
+const PREMIUM_PAGE_ID = '1149636';
+const PREMIUM_URL = '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991';
+const DAILY_LIMIT = 5;
+
+// --- متغیرهای وضعیت ---
+window.userSubscriptionStatus = 'free'; // پیش‌فرض رایگان
+window.userFingerprint = null;
+
+// --- ایجاد فینگرپرینت ---
+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");
+    }
+    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);
+}
 
-    // Request User Data on Load
-    window.parent.postMessage({ type: 'REQUEST_USER_DATA' }, '*');
+// --- بررسی اینکه آیا کاربر پولی است ---
+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 updateUserStatusUI() {
-        const badge = document.getElementById('user-status-badge');
-        if (!badge) return;
+// --- به‌روزرسانی UI بر اساس وضعیت کاربر ---
+function updateUIForSubscriptionStatus(status) {
+    window.userSubscriptionStatus = status;
+    
+    const badge = document.getElementById('subscription-status-badge');
+    const creditInfo = document.getElementById('credit-info');
+    const runBtn = document.getElementById('run-btn');
+    const upgradeBtn = document.getElementById('upgrade-premium-btn');
+    
+    if (status === 'paid') {
+        if(badge) {
+            badge.textContent = '⭐️ نسخه نامحدود';
+            badge.className = 'paid-badge';
+            badge.style.display = 'inline-block';
+        }
+        if(creditInfo) creditInfo.style.display = 'none';
+        
+        // کاربر پولی همیشه دکمه فعال دارد
+        if(runBtn) runBtn.disabled = false;
+        if(upgradeBtn) upgradeBtn.style.display = 'none';
+        
+    } else {
+        if(badge) {
+            badge.textContent = 'نسخه رایگان';
+            badge.className = 'free-badge';
+            badge.style.display = 'inline-block';
+        }
+        // بررسی اعتبار برای کاربر رایگان
+        checkFreeUserCredit();
+    }
+}
 
-        if (isPremium) {
-            badge.className = 'status-badge premium';
-            badge.innerHTML = '⭐️ نسخه نامحدود ویژه';
+// --- مدیریت اعتبار کاربر رایگان (ذخیره در LocalStorage) ---
+function checkFreeUserCredit() {
+    if (window.userSubscriptionStatus === 'paid') return;
+    
+    const today = new Date().toLocaleDateString();
+    const storageKey = 'alpha_edit_usage';
+    let data = JSON.parse(localStorage.getItem(storageKey) || '{}');
+    
+    // اگر تاریخ عوض شده، ریست کن
+    if (data.date !== today) {
+        data = { date: today, count: 0 };
+        localStorage.setItem(storageKey, JSON.stringify(data));
+    }
+    
+    const used = data.count;
+    const remaining = DAILY_LIMIT - used;
+    
+    const creditInfo = document.getElementById('credit-info');
+    if(creditInfo) {
+        creditInfo.style.display = 'block';
+        if (remaining > 0) {
+            creditInfo.innerHTML = `اعتبار امروز: <b>${remaining}</b> تصویر دیگر`;
+            creditInfo.style.color = '#4b5563';
         } else {
-            badge.className = 'status-badge free';
-            badge.innerHTML = '👤 نسخه رایگان (۵ تصویر در روز)';
+             creditInfo.innerHTML = `<span style="color:#ef4444">اعتبار امروز شما تمام شده است.</span>`;
         }
-        badge.style.display = 'inline-flex';
     }
-
-    function updateHiddenInputs() {
-        // Find hidden textboxes in Gradio (hacky but standard for Gradio)
-        // We look for elements with specific IDs assigned in Python
-        const idInput = document.querySelector('#hidden_user_id textarea');
-        const premiumInput = document.querySelector('#hidden_user_premium textarea');
-        
-        if (idInput) {
-            idInput.value = userId;
-            idInput.dispatchEvent(new Event('input', { bubbles: true }));
+    
+    const runBtn = document.getElementById('run-btn');
+    const upgradeBtn = document.getElementById('upgrade-premium-btn');
+    
+    if (used >= DAILY_LIMIT) {
+        if(runBtn) {
+            runBtn.disabled = true;
+            runBtn.innerText = "⛔ اعتبار روزانه تمام شد";
+            runBtn.style.opacity = "0.6";
+            runBtn.style.cursor = "not-allowed";
+        }
+        if(upgradeBtn) {
+            upgradeBtn.style.display = 'block'; // نمایش دکمه ارتقا
         }
-        if (premiumInput) {
-            premiumInput.value = isPremium ? "true" : "false";
-            premiumInput.dispatchEvent(new Event('input', { bubbles: true }));
+    } else {
+        if(runBtn) {
+            runBtn.disabled = false;
+            runBtn.innerText = "✨ شروع پردازش و ساخت تصویر";
+            runBtn.style.opacity = "1";
+            runBtn.style.cursor = "pointer";
         }
+        if(upgradeBtn) upgradeBtn.style.display = 'none';
     }
+}
+
+// --- تابع کسر اعتبار (قبل از اجرای پایتون صدا زده می‌شود) ---
+async function deductCreditAndCheck() {
+    if (window.userSubscriptionStatus === 'paid') return true;
+    
+    const today = new Date().toLocaleDateString();
+    const storageKey = 'alpha_edit_usage';
+    let data = JSON.parse(localStorage.getItem(storageKey) || '{}');
+    
+    if (data.date !== today) {
+        data = { date: today, count: 0 };
+    }
+    
+    if (data.count >= DAILY_LIMIT) {
+        alert("اعتبار روزانه شما تمام شده است. لطفاً نسخه نامحدود را تهیه کنید.");
+        throw new Error("Credit limit reached"); // توقف اجرا
+    }
+    
+    // افزایش کنتور
+    data.count++;
+    localStorage.setItem(storageKey, JSON.stringify(data));
+    checkFreeUserCredit(); // بروزرسانی UI
+    return true;
+}
+
+// --- هندلر ارتقا ---
+window.navigateToPremium = function() {
+    parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
+}
+
+// --- شروع و لیسنرها ---
+document.addEventListener('DOMContentLoaded', async () => {
+    window.userFingerprint = await getBrowserFingerprint();
+    
+    // درخواست وضعیت از اندروید
+    parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
+    
+    window.addEventListener('message', (event) => {
+        if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
+            if (event.data.error || !event.data.payload) {
+                updateUIForSubscriptionStatus('free');
+                return;
+            }
+            try {
+                const userObject = JSON.parse(event.data.payload);
+                const status = isUserPaid(userObject) ? 'paid' : 'free';
+                updateUIForSubscriptionStatus(status);
+            } catch (e) {
+                updateUIForSubscriptionStatus('free');
+            }
+        }
+    });
     
-    // Retry finding elements if Gradio hasn't fully loaded them
-    setTimeout(updateHiddenInputs, 2000);
-    setTimeout(updateHiddenInputs, 5000);
+    // بررسی اولیه
+    setTimeout(() => {
+        if(window.userSubscriptionStatus === 'free') checkFreeUserCredit();
+    }, 1000);
+});
+</script>
+"""
+
+# --- CSS Updated (With Badges & Upgrade Button) ---
+css_code = """
+<style>
+@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700;800&display=swap');
+
+:root, .dark, body, .gradio-container {
+    --body-background-fill: #f5f7fa !important;
+    --body-text-color: #1f2937 !important;
+    font-family: 'Vazirmatn', sans-serif !important;
+}
+
+/* Badges */
+#subscription-status-badge {
+    display: none; /* توسط JS کنترل می‌شود */
+    padding: 6px 16px;
+    border-radius: 20px;
+    font-size: 0.9em;
+    font-weight: 700;
+    margin-top: 10px;
+    letter-spacing: 0.5px;
+    text-shadow: 0 1px 2px rgba(0,0,0,0.1);
+    animation: fadeIn 0.5s ease;
+}
+
+.free-badge {
+    background: linear-gradient(45deg, #6c757d, #495057);
+    color: white;
+    box-shadow: 0 4px 10px rgba(108, 117, 125, 0.3);
+}
+
+.paid-badge {
+    background: linear-gradient(45deg, #FFC107, #ffca2c);
+    color: #333;
+    box-shadow: 0 4px 10px rgba(255, 193, 7, 0.4);
+}
+
+@keyframes fadeIn { from { opacity: 0; transform: translateY(5px); } to { opacity: 1; transform: translateY(0); } }
+
+/* Upgrade Button */
+@keyframes upgrade-pulse {
+    0% { transform: scale(1); box-shadow: 0 0 0 0 rgba(255, 193, 7, 0.7); }
+    70% { transform: scale(1.02); box-shadow: 0 0 0 10px rgba(255, 193, 7, 0); }
+    100% { transform: scale(1); box-shadow: 0 0 0 0 rgba(255, 193, 7, 0); }
+}
+
+#upgrade-premium-btn {
+    background: linear-gradient(95deg, #FFD54F, #FFC107 100%) !important;
+    color: #212529 !important;
+    font-weight: 800 !important;
+    border: none !important;
+    font-size: 1.1em !important;
+    padding: 12px !important;
+    border-radius: 12px !important;
+    margin-top: 15px !important;
+    width: 100%;
+    animation: upgrade-pulse 2s infinite;
+    cursor: pointer !important;
+    display: none; /* پیش‌فرض مخفی */
+}
+#upgrade-premium-btn:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 10px 20px -5px rgba(255, 193, 7, 0.5);
+    animation: none;
+}
+
+/* Other Styles */
+#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: 5px;
+    font-weight: 800;
+    background: -webkit-linear-gradient(45deg, #2563eb, #1e40af);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+}
+
+#main-description {
+    text-align: center;
+    font-size: 1.15em;
+    color: #4b5563 !important;
+    margin-bottom: 10px;
+    line-height: 1.6;
+}
+
+/* Center badge container */
+.badge-container {
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    justify-content: center;
+    margin-bottom: 30px;
+}
+
+textarea, input[type="text"] {
+    border: 2px solid #e2e8f0 !important;
+    border-radius: 12px !important;
+    background-color: #ffffff !important;
+    color: #111827 !important;
+    padding: 12px !important;
+    font-family: 'Vazirmatn', sans-serif !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;
+    width: 100%;
+}
+
+.primary-btn:disabled {
+    background: #9ca3af !important;
+    box-shadow: none !important;
+    cursor: not-allowed !important;
+}
+
+#download-btn {
+    background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
+}
 
+footer { display: none !important; }
 
-    // --------------------------------------------------------
-    // 2. Force Light Mode
-    // --------------------------------------------------------
+/* IP Guide Styles (Keep existing) */
+.ip-reset-guide-container {
+    text-align: right; direction: rtl; background: rgba(255, 255, 255, 0.98); 
+    backdrop-filter: blur(10px); padding: 20px; border-radius: 16px;
+    box-shadow: 0 20px 25px -5px rgba(0,0,0,0.1); border: 1px solid rgba(100,100,200,0.2);
+    width: 90%; max-width: 420px; font-family: 'Vazirmatn', sans-serif;
+}
+.guide-header { display: flex; align-items: center; margin-bottom: 15px; }
+.guide-header h2 { font-size: 1.2rem; margin: 0; color: #2d3748; }
+.elegant-video-button { 
+    display: inline-flex; align-items: center; justify-content: center; padding: 10px 20px; 
+    background: white; color: #667eea; border: 1px solid #e2e8f0; border-radius: 50px; cursor: pointer;
+    font-family: inherit; font-weight: 600; margin-top: 10px;
+}
+.action-button { 
+    padding: 10px; border-radius: 12px; border: none; cursor: pointer; flex: 1; font-family: inherit; font-weight: 600;
+}
+.retry-button { background: linear-gradient(135deg, #667eea, #764ba2); color: white; }
+.back-button { background: white; border: 1px solid #e2e8f0; }
+</style>
+"""
+
+# جاوااسکریپت سراسری (شامل راهنمای IP و استایل‌ها)
+# توجه: ما کدهای جدید Auth را هم به این بخش اضافه می‌کنیم تا یکجا اینجکت شود
+js_global_combined = js_user_auth_logic + """
+<script>
+document.addEventListener('DOMContentLoaded', () => {
+    // 1. Force Light Mode
     const forceLight = () => {
         const body = document.querySelector('body');
         if (body) {
@@ -475,14 +630,11 @@ document.addEventListener('DOMContentLoaded', () => {
             body.style.backgroundColor = '#f5f7fa';
             body.style.color = '#333333';
         }
-        document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
     };
     forceLight();
     setInterval(forceLight, 1000);
 
-    // --------------------------------------------------------
-    // 3. Retry Functionality
-    // --------------------------------------------------------
+    // 2. IP Reset Guide Functions
     window.retryGeneration = function() {
         const modal = document.getElementById('custom-quota-modal');
         if (modal) modal.remove();
@@ -495,82 +647,39 @@ document.addEventListener('DOMContentLoaded', () => {
          if (modal) modal.remove();
     };
 
-    // --------------------------------------------------------
-    // 4. Quota Modal (IP Reset Guide)
-    // --------------------------------------------------------
     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>
+                        <p>برای استفاده نامحدود، اینترنت را خاموش/روشن کنید یا IP را تغییر دهید.</p>
+                        <div style="display:flex; gap:10px; margin-top:20px;">
+                            <button class="action-button back-button" onclick="window.closeErrorModal()">بازگشت</button>
+                            <button class="action-button retry-button" onclick="window.retryGeneration()">تلاش مجدد</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);
-        setTimeout(() => { window.closeErrorModal(); }, 10000);
+        setTimeout(() => window.closeErrorModal(), 10000);
     };
 
-    // --------------------------------------------------------
-    // 5. Error Scanner
-    // --------------------------------------------------------
+    // 3. Scanner for Quota Errors
     setInterval(() => {
-        const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar, div[class*="error"]');
+        const potentialErrors = document.querySelectorAll('.toast-body, .error, 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);
@@ -578,146 +687,100 @@ document.addEventListener('DOMContentLoaded', () => {
 </script>
 """
 
-# --- CSS Updated ---
-css_code = """
-<style>
-@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
-
-:root, .dark, body, .gradio-container {
-    --body-background-fill: #f5f7fa !important;
-    --body-text-color: #1f2937 !important;
-    font-family: 'Vazirmatn', sans-serif !important;
-}
-
-/* User Status Badge CSS */
-.status-badge {
-    display: none; /* Initially hidden until JS detects user */
-    align-items: center;
-    justify-content: center;
-    padding: 8px 16px;
-    border-radius: 50px;
-    font-size: 0.95em;
-    font-weight: 700;
-    margin: 10px auto 20px auto;
-    width: fit-content;
-    box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);
-    animation: fadeInBadge 0.5s ease-out;
-}
-
-.status-badge.premium {
-    background: linear-gradient(135deg, #FFD700 0%, #FDB931 100%);
-    color: #333;
-    border: 2px solid #fff;
-    box-shadow: 0 0 15px rgba(255, 215, 0, 0.4);
-}
-
-.status-badge.free {
-    background: #e2e8f0;
-    color: #475569;
-    border: 1px solid #cbd5e1;
-}
-
-@keyframes fadeInBadge {
-    from { opacity: 0; transform: translateY(-10px); }
-    to { opacity: 1; transform: translateY(0); }
-}
-
-/* Existing CSS for Guide, Buttons, etc. */
-.ip-reset-guide-container {
-    text-align: right; direction: rtl; background: rgba(255, 255, 255, 0.98); 
-    backdrop-filter: blur(10px); padding: 20px; border-radius: 16px;
-    box-shadow: 0 20px 25px -5px rgba(0,0,0,0.1); border: 1px solid rgba(102, 126, 234, 0.2); 
-    width: 90%; max-width: 420px; max-height: 90vh; overflow-y: auto; 
-    font-family: 'Vazirmatn', sans-serif !important;
-}
-.guide-header { display: flex; align-items: center; margin-bottom: 15px; }
-.guide-header-icon { width: 45px; height: 45px; margin-left: 15px; }
-.guide-header h2 { font-size: 1.2rem; margin: 0; }
-.guide-content { font-size: 0.9rem; line-height: 1.6; }
-.info-card, .summary-section { margin-top: 12px; padding: 12px; border-radius: 12px; }
-.info-card { background: #f0f4ff; border: 1px solid #c3dafe; }
-.summary-section { background: #f0fff4; border: 1px solid #c6f6d5; }
-.video-button-container { text-align: center; margin-top: 15px; }
-.elegant-video-button { padding: 8px 16px; border-radius: 20px; border: 1px solid #ddd; background: white; cursor: pointer; display: inline-flex; align-items: center; }
-.guide-actions { display: flex; gap: 10px; margin-top: 20px; }
-.action-button { flex: 1; padding: 10px; border-radius: 10px; border: none; cursor: pointer; display: flex; align-items: center; justify-content: center; font-weight: bold; }
-.back-button { background: white; border: 1px solid #ddd; }
-.retry-button { background: #667eea; color: white; }
-
-/* Main UI Tweaks */
-#col-container { max-width: 980px; margin: 0 auto; direction: rtl; text-align: right; background: white; padding: 20px; border-radius: 20px; }
-#main-title h1 { text-align: center; color: #2563eb; }
-#main-description { text-align: center; color: #4b5563; }
-.primary-btn { background: linear-gradient(135deg, #10b981, #059669); color: white; border: none; }
-footer { display: none !important; }
-</style>
-"""
-
-combined_html = css_code + js_global_content
+# ادغام نهایی
+combined_html = css_code + js_global_combined
 
+# استفاده از 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.Markdown(
+            "با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", 
+            elem_id="main-description"
+        )
         
-        # بج وضعیت کاربر (HTML خالی که توسط JS پر می‌شود)
-        gr.HTML('<div id="user-status-badge" class="status-badge"></div>')
-
-        # فیلدهای مخفی برای انتقال اطلاعات کاربر از JS به پایتون
-        user_id_storage = gr.Textbox(visible=False, elem_id="hidden_user_id")
-        premium_status_storage = gr.Textbox(visible=False, elem_id="hidden_user_premium")
+        # --- بخش جدید: بج و اطلاعات اعتبار ---
+        gr.HTML("""
+        <div class="badge-container">
+            <div id="subscription-status-badge"></div>
+            <div id="credit-info" style="font-size: 0.9em; margin-top: 8px; color: #666; display:none;"></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="وضعیت")
-                run_button = gr.Button("✨ شروع پردازش", variant="primary", elem_classes="primary-btn", elem_id="run-btn")
+                
+                # دکمه اجرا (با شناسه برای JS)
+                run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn", elem_id="run-btn")
+                
+                # دکمه ارتقا (مخفی، توسط JS کنترل می‌شود)
+                upgrade_button = gr.Button("⭐️ ارتقا به نسخه کامل و نامحدود", visible=True, elem_id="upgrade-premium-btn")
 
             with gr.Column():
                 output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
+                
                 download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
                 
                 with gr.Row():
-                    lora_adapter = gr.Dropdown(label="انتخاب سبک", choices=list(LORA_MAPPING.keys()), value="تبدیل عکس به انیمه")
+                    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)
-                    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)
+                        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)
         
-        # لاجیک نمایش تنظیمات ابعاد
         def toggle_row(choice):
-            return gr.update(visible=True) if choice == "شخصی‌سازی (Custom)" else gr.update(visible=False)
-
-        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,
-                user_id_storage, premium_status_storage  # ارسال اطلاعات کاربر به تابع infer
-            ],
-            outputs=[output_image, seed, status_box],
-            api_name="predict"
+            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
         )
 
-        download_button.click(fn=None, inputs=[output_image], outputs=None, js=js_download_func)
-        
-        # نمونه‌ها
         gr.Examples(
             examples=[
                 ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
-                ["examples/5.jpg", "سایه‌ها را حذف کن.", "اصلاح نور و سایه"],
+                ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم به تصویر بده.", "اصلاح نور و سایه"],
+                ["examples/4.jpg", "از فیلتر ساعت طلایی با پخش نور ملایم استفاده کن.", "نورپردازی مجدد (Relight)"],
             ],
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_image, seed, status_box],
@@ -726,5 +789,31 @@ with gr.Blocks() as demo:
             label="نمونه‌ها"
         )
 
+    # --- اتصال رویدادها ---
+    
+    # دکمه اجرا: اول JS اجرا می‌شود (کسر اعتبار)، اگر موفق بود پایتون اجرا می‌شود
+    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],
+        api_name="predict",
+        js="deductCreditAndCheck" # این تابع JS قبل از پایتون اجرا می‌شود
+    )
+
+    # دکمه ارتقا: فقط JS اجرا می‌شود (پیام به اندروید)
+    upgrade_button.click(
+        fn=None,
+        inputs=None,
+        outputs=None,
+        js="navigateToPremium"
+    )
+
+    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)