Update app.py

app.py

@@ -4,14 +4,21 @@ import numpy as np
 import spaces
 import torch
 import random
-from PIL import Image, ImageFilter
-from typing import Iterable
-from gradio.themes import Soft
-from gradio.themes.utils import colors, fonts, sizes
+import json
+import time
+from datetime import date
+from PIL import Image
+from gradio.themes.utils import colors
 from deep_translator import GoogleTranslator
 from transformers import pipeline
 
-# --- تعریف تم ---
+# --- 1. تنظیمات سیستم اعتبار و اشتراک ---
+USAGE_LIMIT = 5  # اعتبار رایگان روزانه
+PREMIUM_PAGE_ID = '1149636' # آی‌دی صفحه محصول پریمیم
+PREMIUM_URL = '#/nav/online/news/getSingle/1149636' # لینک خرید اشتراک
+usage_data_cache = {} # حافظه موقت سرور برای نگهداری تعداد استفاده
+
+# --- 2. تنظیمات تم و مدل‌ها ---
 colors.steel_blue = colors.Color(
     name="steel_blue",
     c50="#EBF3F8",
@@ -29,7 +36,7 @@ colors.steel_blue = colors.Color(
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
+# --- بارگذاری مدل Safety Checker ---
 print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
@@ -45,7 +52,7 @@ def is_image_nsfw(image):
         print(f"Safety check error: {e}")
         return False
 
-# --- بارگذاری مدل اصلی ---
+# --- بارگذاری مدل Qwen ---
 from diffusers import FlowMatchEulerDiscreteScheduler
 from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
@@ -90,84 +97,35 @@ LORA_MAPPING = {
     "افزایش کیفیت (Upscale)": "upscale-image"
 }
 
-ASPECT_RATIOS_LIST = [
-    "خودکار (پیش‌فرض)",
-    "۱:۱ (مربع - 1024x1024)",
-    "۱۶:۹ (افقی - 1344x768)",
-    "۹:۱۶ (عمودی - 768x1344)",
-    "شخصی‌سازی (Custom)"
-]
-
-ASPECT_RATIOS_MAP = {
-    "خودکار (پیش‌فرض)": "Auto",
-    "۱:۱ (مربع - 1024x1024)": (1024, 1024),
-    "۱۶:۹ (افقی - 1344x768)": (1344, 768),
-    "۹:۱۶ (عمودی - 768x1344)": (768, 1344),
-    "شخصی‌سازی (Custom)": "Custom"
-}
-
-BANNED_WORDS = [
-    "nude", "naked", "sex", "porn", "undressed", "nsfw", "erotic", "xxx", 
-    "breast", "nipple", "genital", "vagina", "penis", "ass", "butt", "sexual",
-    "lingerie", "bikini", "swimwear", "underwear", "fetish", "topless",
-    "exhibitionism", "hentai", "ecchi", "18+"
-]
+ASPECT_RATIOS_LIST = ["خودکار (پیش‌فرض)", "۱:۱ (مربع - 1024x1024)", "۱۶:۹ (افقی - 1344x768)", "۹:۱۶ (عمودی - 768x1344)", "شخصی‌سازی (Custom)"]
+ASPECT_RATIOS_MAP = {"خودکار (پیش‌فرض)": "Auto", "۱:۱ (مربع - 1024x1024)": (1024, 1024), "۱۶:۹ (افقی - 1344x768)": (1344, 768), "۹:۱۶ (عمودی - 768x1344)": (768, 1344), "شخصی‌سازی (Custom)": "Custom"}
+BANNED_WORDS = ["nude", "naked", "sex", "porn", "undressed", "nsfw", "erotic", "xxx", "breast", "nipple", "genital", "vagina", "penis", "ass", "butt", "sexual", "lingerie", "bikini", "swimwear", "underwear", "fetish", "topless", "exhibitionism", "hentai", "ecchi", "18+"]
 
 def check_text_safety(text):
     text_lower = text.lower()
     for word in BANNED_WORDS:
-        if f" {word} " in f" {text_lower} ": 
-            return False
+        if f" {word} " in f" {text_lower} ": return False
     return True
 
 def translate_prompt(text):
-    if not text:
-        return ""
-    try:
-        translated = GoogleTranslator(source='auto', target='en').translate(text)
-        return translated
-    except Exception as e:
-        print(f"Translation Error: {e}")
-        return text
+    if not text: return ""
+    try: return GoogleTranslator(source='auto', target='en').translate(text)
+    except: return text
 
 def update_dimensions_on_upload(image):
-    if image is None:
-        return 1024, 1024
-    original_width, original_height = image.size
-    if original_width > original_height:
-        new_width = 1024
-        aspect_ratio = original_height / original_width
-        new_height = int(new_width * aspect_ratio)
-    else:
-        new_height = 1024
-        aspect_ratio = original_width / original_height
-        new_width = int(new_height * aspect_ratio)
-    new_width = (new_width // 8) * 8
-    new_height = (new_height // 8) * 8
-    return new_width, new_height
-
-def update_sliders_visibility(choice):
-    if choice == "شخصی‌سازی (Custom)":
-        return gr.update(visible=True), gr.update(visible=True)
-    else:
-        return gr.update(visible=False), gr.update(visible=False)
+    if image is None: return 1024, 1024
+    w, h = image.size
+    if w > h: new_w = 1024; ar = h/w; new_h = int(new_w * ar)
+    else: new_h = 1024; ar = w/h; new_w = int(new_h * ar)
+    return (new_w // 8) * 8, (new_h // 8) * 8
 
 def get_error_html(message):
-    return f"""
-    <div style="background-color: #fee2e2; border: 1px solid #ef4444; color: #b91c1c; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;">
-        <span style="font-size: 1.2em;">⛔</span>
-        {message}
-    </div>
-    """
+    return f"""<div style="background-color: #fee2e2; border: 1px solid #ef4444; color: #b91c1c; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;"><span style="font-size: 1.2em;">⛔</span>{message}</div>"""
 
 def get_success_html(message):
-    return f"""
-    <div style="background-color: #dcfce7; border: 1px solid #22c55e; color: #15803d; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;">
-        <span style="font-size: 1.2em;">✅</span>
-        {message}
-    </div>
-    """
+    return f"""<div style="background-color: #dcfce7; border: 1px solid #22c55e; color: #15803d; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;"><span style="font-size: 1.2em;">✅</span>{message}</div>"""
 
+# --- 3. تابع اصلی پردازش (شامل بررسی اعتبار) ---
 @spaces.GPU(duration=30)
 def infer(
     input_image,
@@ -180,270 +138,176 @@ def infer(
     aspect_ratio_selection,
     custom_width,
     custom_height,
+    # ورودی‌های جدید برای سیستم اعتبار که از JS می‌آیند
+    user_fingerprint,
+    user_subscription_status,
     progress=gr.Progress(track_tqdm=True)
 ):
-    if input_image is None:
-        return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
+    # --- بررسی اعتبار کاربر ---
+    # اگر Fingerprint نرسد یعنی کاربر مستقیماً API را صدا زده یا JS لود نشده
+    if not user_fingerprint:
+        # برای جلوگیری از خطا در تست‌های لوکال، اگر نال بود یک مقدار موقت می‌دهیم، 
+        # اما در پروداکشن واقعی می‌توانید ریترن ارور کنید.
+        user_fingerprint = "unknown_user"
+    
+    # اگر کاربر پولی نباشد، اعتبار چک می‌شود
+    if user_subscription_status != 'paid':
+        today_str = date.today().isoformat()
+        
+        # بازیابی یا ایجاد رکورد کاربر
+        user_record = usage_data_cache.get(user_fingerprint)
+        if not user_record or user_record.get("last_reset") != today_str:
+            user_record = {"count": 0, "last_reset": today_str}
+            usage_data_cache[user_fingerprint] = user_record
+        
+        # چک کردن لیمیت
+        if user_record["count"] >= USAGE_LIMIT:
+            msg = f"اعتبار رایگان روزانه شما ({USAGE_LIMIT} تصویر) به پایان رسیده است. برای استفاده نامحدود ارتقا دهید."
+            return None, seed, get_error_html(msg)
+        
+        # کسر اعتبار (افزایش شمارنده)
+        user_record["count"] += 1
+        print(f"User {user_fingerprint} usage: {user_record['count']}/{USAGE_LIMIT}")
 
-    if is_image_nsfw(input_image):
-        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
+    # --- ادامه منطق اصلی برنامه ---
+    if input_image is None: return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
+    if is_image_nsfw(input_image): return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است.")
 
     english_prompt = translate_prompt(prompt)
-    if not check_text_safety(english_prompt):
-        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است.")
+    if not check_text_safety(english_prompt): return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز است.")
 
     adapter_internal_name = LORA_MAPPING.get(lora_adapter_persian)
-    if adapter_internal_name:
-        pipe.set_adapters([adapter_internal_name], adapter_weights=[1.0])
+    if adapter_internal_name: pipe.set_adapters([adapter_internal_name], adapter_weights=[1.0])
     
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-
+    if randomize_seed: seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
     
-    safety_negative = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, 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}"
+    final_negative_prompt = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood, worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
 
     original_image = input_image.convert("RGB")
-    
     selection_value = ASPECT_RATIOS_MAP.get(aspect_ratio_selection)
-    
-    if selection_value == "Custom":
-        width = (int(custom_width) // 8) * 8
-        height = (int(custom_height) // 8) * 8
-    elif selection_value == "Auto" or selection_value is None:
-        width, height = update_dimensions_on_upload(original_image)
-    else:
-        width, height = selection_value
+    if selection_value == "Custom": width, height = (int(custom_width)//8)*8, (int(custom_height)//8)*8
+    elif selection_value == "Auto" or selection_value is None: width, height = update_dimensions_on_upload(original_image)
+    else: width, height = selection_value
 
     try:
         result = pipe(
-            image=original_image,
-            prompt=english_prompt,
-            negative_prompt=final_negative_prompt,
-            height=height,
-            width=width,
-            num_inference_steps=steps,
-            generator=generator,
-            true_cfg_scale=guidance_scale,
+            image=original_image, prompt=english_prompt, negative_prompt=final_negative_prompt,
+            height=height, width=width, num_inference_steps=steps, generator=generator, true_cfg_scale=guidance_scale,
         ).images[0]
 
-        if is_image_nsfw(result):
-            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+        if is_image_nsfw(result): return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود.")
         
         return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
 
     except Exception as e:
         error_str = str(e)
-        if "quota" in error_str.lower() or "exceeded" in error_str.lower():
-             raise e 
+        if "quota" in error_str.lower() or "exceeded" in error_str.lower(): raise e 
         return None, seed, get_error_html(f"خطا در پردازش: {error_str}")
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024)
+    # برای مثال‌ها بررسی اعتبار را رد می‌کنیم یا فیک می‌فرستیم
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "example_user", "paid")
     return res, s, status
 
-# --- جاوااسکریپت برای دکمه دانلود ---
+# --- 4. کدهای جاوا اسکریپت ---
+
+# کد دانلود فایل
 js_download_func = """
 async (image) => {
-    if (!image) {
-        alert("لطفاً ابتدا تصویر را تولید کنید.");
-        return;
-    }
+    if (!image) { alert("لطفاً ابتدا تصویر را تولید کنید."); return; }
     let fileUrl = image.url;
-    if (fileUrl && !fileUrl.startsWith('http')) {
-       fileUrl = window.location.origin + fileUrl;
-    } else if (!fileUrl && image.path) {
-       fileUrl = window.location.origin + "/file=" + image.path;
-    }
-    window.parent.postMessage({
-        type: 'DOWNLOAD_REQUEST',
-        url: fileUrl
-    }, '*');
+    if (fileUrl && !fileUrl.startsWith('http')) { fileUrl = window.location.origin + fileUrl; } 
+    else if (!fileUrl && image.path) { fileUrl = window.location.origin + "/file=" + image.path; }
+    window.parent.postMessage({ type: 'DOWNLOAD_REQUEST', url: fileUrl }, '*');
 }
 """
 
-# --- منطق جاوااسکریپت برای مدیریت اعتبار و اشتراک ---
-js_credit_logic = """
+# جاوا اسکریپت اصلی (اعتبار، فینگرپرینت، تم، ارور)
+js_global_content = """
 <script>
-document.addEventListener('DOMContentLoaded', () => {
-    const PREMIUM_PAGE_ID = '1149636';
-    const PREMIUM_URL = '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991';
-    
-    let isUserPaid = false;
-    let dailyCredits = 5;
-    
-    // تابع فینگرپرینت ساده برای شناسایی مرورگر
-    function getFingerprint() {
-        const str = navigator.userAgent + navigator.language + screen.width + 'x' + screen.height;
-        let hash = 0;
-        for (let i = 0; i < str.length; i++) {
-            hash = ((hash << 5) - hash) + str.charCodeAt(i);
-            hash |= 0;
-        }
-        return 'fp_' + Math.abs(hash).toString(16);
-    }
+// --- تنظیمات ثابت ---
+const PREMIUM_PAGE_ID = '1149636'; 
+const PREMIUM_URL = '#/nav/online/news/getSingle/1149636';
+
+// متغیرهای گلوبال برای نگهداری وضعیت
+window.userFingerprint = null;
+window.userSubscriptionStatus = 'free';
+
+// --- 1. تابع ساخت فینگرپرینت ---
+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);
+}
 
-    const fingerprint = getFingerprint();
-    const today = new Date().toISOString().slice(0, 10); // YYYY-MM-DD
-    const storageKey = `app_credits_${fingerprint}_${today}`;
-
-    function checkCredits() {
-        const stored = localStorage.getItem(storageKey);
-        if (stored === null) {
-            dailyCredits = 5;
-            localStorage.setItem(storageKey, 5);
-        } else {
-            dailyCredits = parseInt(stored);
-        }
-        updateUI();
-    }
+// --- 2. توابع بررسی اشتراک ---
+function isUserPaid(userObject) { 
+    return userObject && userObject.isLogin && userObject.accessible_pages && 
+           (userObject.accessible_pages.includes(PREMIUM_PAGE_ID) || userObject.accessible_pages.includes(parseInt(PREMIUM_PAGE_ID))); 
+}
 
-    function updateUI() {
-        const badgeEl = document.getElementById('sub-badge');
-        const creditTextEl = document.getElementById('credit-text');
-        const runBtn = document.getElementById('run-btn');
-        const upgradeBtn = document.getElementById('upgrade-btn');
-        
-        if (!badgeEl || !runBtn) return;
-
-        if (isUserPaid) {
-            // حالت پولی
-            badgeEl.innerHTML = '<span class="paid-badge">✨ نسخه نامحدود</span>';
-            creditTextEl.style.display = 'none';
-            runBtn.style.display = 'block';
-            if(upgradeBtn) upgradeBtn.style.display = 'none';
-        } else {
-            // حالت رایگان
-            badgeEl.innerHTML = '<span class="free-badge">نسخه رایگان</span>';
-            creditTextEl.style.display = 'block';
-            creditTextEl.innerHTML = `اعتبار باقی‌مانده امروز: <b>${dailyCredits}</b>`;
-            
-            if (dailyCredits > 0) {
-                runBtn.style.display = 'block';
-                if(upgradeBtn) upgradeBtn.style.display = 'none';
-            } else {
-                runBtn.style.display = 'none';
-                if(upgradeBtn) upgradeBtn.style.display = 'flex';
-                creditTextEl.innerHTML = `<span style="color:#ef4444">اعتبار امروز شما تمام شده است!</span>`;
-            }
-        }
+function updateUIForSubscriptionStatus(status) {
+    window.userSubscriptionStatus = status;
+    const badge = document.getElementById('subscription-badge');
+    const infoText = document.getElementById('credit-info-text');
+    const upgradeBtn = document.getElementById('upgrade-btn-custom');
+    
+    if (status === 'paid') { 
+        if(badge) { badge.textContent = '⭐️ نسخه نامحدود'; badge.className = 'badge-paid'; }
+        if(infoText) infoText.innerHTML = 'شما به صورت <b>نامحدود</b> می‌توانید تصویر بسازید.';
+        if(upgradeBtn) upgradeBtn.style.display = 'none';
+    } else { 
+        if(badge) { badge.textContent = 'نسخه رایگان'; badge.className = 'badge-free'; }
+        if(infoText) infoText.innerHTML = 'اعتبار رایگان روزانه: <b>۵ تصویر</b>. برای استفاده نامحدود ارتقا دهید.';
+        if(upgradeBtn) upgradeBtn.style.display = 'block';
     }
+}
+
+// --- 3. اجرای اولیه و لیسنرها ---
+document.addEventListener('DOMContentLoaded', async () => {
+    // تولید فینگرپرینت
+    window.userFingerprint = await getBrowserFingerprint();
+    
+    // درخواست وضعیت از والد
+    window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
 
-    // بررسی وضعیت کاربر از اپلیکیشن
+    // شنیدن پاسخ والد
     window.addEventListener('message', (event) => {
         if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
-            try {
-                const userObject = JSON.parse(event.data.payload);
-                if (userObject && userObject.isLogin && userObject.accessible_pages && 
-                   (userObject.accessible_pages.includes(PREMIUM_PAGE_ID) || 
-                    userObject.accessible_pages.includes(parseInt(PREMIUM_PAGE_ID)))) {
-                    isUserPaid = true;
-                } else {
-                    isUserPaid = false;
-                }
-                updateUI();
-            } catch (e) {
-                console.error("Error parsing user status", e);
-                isUserPaid = false;
-                updateUI();
+            if (event.data.error || !event.data.payload) { 
+                updateUIForSubscriptionStatus('free'); 
+            } else {
+                try { 
+                    const userObject = JSON.parse(event.data.payload); 
+                    const status = isUserPaid(userObject) ? 'paid' : 'free'; 
+                    updateUIForSubscriptionStatus(status); 
+                } catch (e) { updateUIForSubscriptionStatus('free'); }
             }
         }
-    });
-
-    // درخواست وضعیت کاربر هنگام لود
-    parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
-    
-    // لود اولیه کردیت
-    checkCredits();
-
-    // تابع سراسری برای کسر اعتبار (قبل از اجرای پایتون)
-    window.checkAndDeductCredit = async function() {
-        if (isUserPaid) return []; // اگر پولی بود، اجازه بده
-
-        checkCredits(); // رفرش کردن مقدار
-        if (dailyCredits > 0) {
-            dailyCredits--;
-            localStorage.setItem(storageKey, dailyCredits);
-            updateUI();
-            return []; // اجازه اجرا
-        } else {
-            alert("اعتبار امروز شما به پایان رسیده است. لطفاً برای استفاده نامحدود ارتقا دهید.");
-            throw new Error("Credit Exceeded"); // جلوگیری از اجرا
+        if (event.data && event.data.type === 'NAVIGATE_TO_PREMIUM') {
+             // این رویداد توسط دکمه ارتقا تریگر میشه
         }
-    };
-    
-    // تابع برای دکمه ارتقا
-    window.navigateToPremium = function() {
-        parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
-    };
-
-    // Force UI Refresh periodically
-    setInterval(updateUI, 2000);
-});
-</script>
-"""
+    });
 
-# --- جاوااسکریپت برای مدیریت خطا و تم (کد قبلی شما با کمی اصلاح) ---
-js_global_content = """
-<script>
-document.addEventListener('DOMContentLoaded', () => {
-    // 1. Force Light Mode
+    // فورس تم روشن
     const forceLight = () => {
         const body = document.querySelector('body');
-        if (body) {
-            body.classList.remove('dark');
-            body.style.backgroundColor = '#f5f7fa';
-            body.style.color = '#333333';
-        }
+        if (body) { body.classList.remove('dark'); body.style.backgroundColor = '#f5f7fa'; body.style.color = '#333333'; }
         document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
     };
-    forceLight();
-    setInterval(forceLight, 1000);
-
-    // 2. RETRY FUNCTION
-    window.retryGeneration = function() {
-        const modal = document.getElementById('custom-quota-modal');
-        if (modal) modal.remove();
-        // تلاش مجدد نباید اعتبار کسر کند یا باید مدیریت شود، اما اینجا دکمه اصلی را کلیک می‌کنیم
-        // چون تابع checkAndDeductCredit روی دکمه است، دوباره اعتبار چک می‌شود.
-        const runBtn = document.getElementById('run-btn');
-        if(runBtn) runBtn.click();
-    };
-    
-    // Close function
-    window.closeErrorModal = function() {
-         const modal = document.getElementById('custom-quota-modal');
-         if (modal) modal.remove();
-    };
-
-    // 3. SHOW MODAL FUNCTION (Quota Exceeded)
-    const showQuotaModal = () => {
-        if (document.getElementById('custom-quota-modal')) return;
-
-        const modalHtml = `
-            <div id="custom-quota-modal" style="position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.6); backdrop-filter: blur(5px); z-index: 99999; display: flex; align-items: center; justify-content: center; font-family: 'Vazirmatn', sans-serif;">
-                <div class="ip-reset-guide-container">
-                    <div class="guide-header">
-                        <h2>یک قدم تا ساخت تصاویر جدید</h2>
-                        <p>نیازمند تغییر نقطه دستیابی</p>
-                    </div>
-                    <div class="guide-content">
-                        <p>لطفاً IP خود را تغییر دهید (خاموش/روشن کردن حالت پرواز) و دوباره تلاش کنید.</p>
-                        <div class="guide-actions">
-                            <button class="action-button back-button" onclick="window.closeErrorModal()">بازگشت</button>
-                            <button class="action-button retry-button" onclick="window.retryGeneration()">تلاش مجدد</button>
-                        </div>
-                    </div>
-                </div>
-            </div>
-        `;
-        document.body.insertAdjacentHTML('beforeend', modalHtml);
-    };
+    forceLight(); setInterval(forceLight, 1000);
 
-    // 4. SCANNER
+    // اسکنر خطای Quota
     setInterval(() => {
         const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar, div[class*="error"]');
         potentialErrors.forEach(el => {
@@ -453,220 +317,167 @@ document.addEventListener('DOMContentLoaded', () => {
                  el.style.display = 'none';
             }
         });
-    }, 1000);
+    }, 100);
 });
+
+// --- 4. توابع کمکی ---
+window.retryGeneration = function() {
+    const modal = document.getElementById('custom-quota-modal'); if (modal) modal.remove();
+    const runBtn = document.getElementById('run-btn'); if(runBtn) runBtn.click();
+};
+window.closeErrorModal = function() {
+     const modal = document.getElementById('custom-quota-modal'); if (modal) modal.remove();
+};
+window.navigateToPremium = function() {
+    window.parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
+};
+
+const showQuotaModal = () => {
+    if (document.getElementById('custom-quota-modal')) return;
+    // (HTML مودال خطا همانند قبل است، خلاصه شده)
+    const modalHtml = `<div id="custom-quota-modal" style="position:fixed;top:0;left:0;width:100%;height:100%;background:rgba(0,0,0,0.6);backdrop-filter:blur(5px);z-index:99999;display:flex;align-items:center;justify-content:center;">
+        <div class="ip-reset-guide-container">
+            <h3>خطای محدودیت سرور</h3>
+            <p>لطفاً IP خود را تغییر دهید یا چند دقیقه صبر کنید.</p>
+            <button onclick="window.closeErrorModal()" class="action-button back-button">بازگشت</button>
+            <button onclick="window.retryGeneration()" class="action-button retry-button">تلاش مجدد</button>
+        </div>
+    </div>`;
+    document.body.insertAdjacentHTML('beforeend', modalHtml);
+    setTimeout(() => window.closeErrorModal(), 10000);
+};
 </script>
 """
 
-# --- CSS Updated ---
 css_code = """
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
 
+/* Main Styling */
 :root, .dark, body, .gradio-container {
-    --body-background-fill: #f5f7fa !important;
-    --body-text-color: #1f2937 !important;
+    --body-background-fill: #f5f7fa !important; --body-text-color: #1f2937 !important;
     font-family: 'Vazirmatn', sans-serif !important;
 }
 
-/* Badge Styles */
-.paid-badge {
-    background: linear-gradient(45deg, #FFC107, #ffca2c);
-    color: #333;
-    padding: 6px 12px;
-    border-radius: 20px;
-    font-size: 0.9em;
-    font-weight: 800;
-    box-shadow: 0 4px 10px rgba(255, 193, 7, 0.3);
-    display: inline-block;
-}
-
-.free-badge {
-    background: linear-gradient(45deg, #6c757d, #495057);
-    color: white;
-    padding: 6px 12px;
-    border-radius: 20px;
-    font-size: 0.9em;
-    font-weight: 700;
-    box-shadow: 0 4px 10px rgba(108, 117, 125, 0.3);
-    display: inline-block;
+/* Subscription Badge & Info */
+.subscription-info-container {
+    text-align: center; margin-bottom: 20px; padding: 15px;
+    background: white; border-radius: 16px; border: 1px solid #e5e7eb;
+    box-shadow: 0 4px 6px -1px rgba(0,0,0,0.05);
+    display: flex; flex-direction: column; align-items: center; gap: 10px;
 }
-
-#subscription-status-container {
-    text-align: center;
-    margin-bottom: 20px;
-    padding: 10px;
-    background: #fff;
-    border-radius: 12px;
-    border: 1px solid #e5e7eb;
+.badge-free {
+    background: linear-gradient(45deg, #6b7280, #4b5563); color: white;
+    padding: 6px 16px; border-radius: 20px; font-size: 0.9em; font-weight: 700;
 }
-
-#credit-text {
-    margin-top: 8px;
-    color: #4b5563;
-    font-size: 0.95em;
+.badge-paid {
+    background: linear-gradient(45deg, #f59e0b, #d97706); color: white;
+    padding: 6px 16px; border-radius: 20px; font-size: 0.9em; font-weight: 700;
+    box-shadow: 0 4px 10px rgba(245, 158, 11, 0.3);
 }
-
-#upgrade-btn {
-    display: none; /* Controlled by JS */
-    width: 100%;
-    margin-top: 10px;
-    padding: 12px;
-    font-family: 'Vazirmatn', sans-serif;
-    font-size: 1.1em;
-    font-weight: 800;
-    color: #212529;
-    background: linear-gradient(95deg, #FFD54F, #FFC107 100%);
-    border: none;
-    border-radius: 14px;
-    cursor: pointer;
-    box-shadow: 0 8px 20px -5px rgba(255, 193, 7, 0.4);
-    animation: upgrade-pulse 2.5s infinite;
-    justify-content: center;
-    align-items: center;
+.credit-text { color: #4b5563; font-size: 1rem; }
+.upgrade-btn {
+    background: linear-gradient(95deg, #FFD54F, #FFC107); color: #212529;
+    border: none; padding: 8px 20px; border-radius: 12px; font-weight: 800; cursor: pointer;
+    box-shadow: 0 4px 15px rgba(255, 193, 7, 0.3); transition: transform 0.2s;
+    display: none; /* Hidden by default, shown by JS */
 }
+.upgrade-btn:hover { transform: translateY(-2px); }
 
-@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); }
+/* Buttons & Inputs */
+.primary-btn {
+    background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important; color: white !important;
+    border-radius: 14px !important; box-shadow: 0 4px 15px rgba(16, 185, 129, 0.3) !important;
 }
-
-/* Modal and other existing styles... */
-.ip-reset-guide-container {
+/* Modal Styles (Simplified for brevity) */
+.ip-reset-guide-container { 
     background: white; padding: 20px; border-radius: 16px; width: 90%; max-width: 400px; text-align: right; direction: rtl;
 }
-.action-button { margin: 5px; padding: 10px 20px; border-radius: 8px; cursor: pointer; border: none; }
-.retry-button { background: #667eea; color: white; }
-.back-button { background: #f0f0f0; }
-
-#main-title h1 {
-    font-size: 2.2em !important;
-    text-align: center;
-    color: #1a202c !important;
-    background: -webkit-linear-gradient(45deg, #2563eb, #1e40af);
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-}
 </style>
 """
 
-# ادغام CSS و JS ها
-combined_html = css_code + js_global_content + js_credit_logic
+# ترکیب کدها
+combined_html = css_code + js_global_content
 
-# استفاده از gr.Blocks
-with gr.Blocks() as demo:
-    # تزریق کدها به عنوان HTML
+# --- 5. رابط کاربری Gradio ---
+with gr.Blocks(title="ویرایشگر هوشمند آلفا") as demo:
     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")
+        
+        # --- بخش جدید: نمایش وضعیت اشتراک و توضیحات ---
         gr.HTML("""
-        <div id="subscription-status-container">
-            <div id="sub-badge"></div>
-            <div id="credit-text">در حال بررسی اعتبار...</div>
-            <button id="upgrade-btn" onclick="window.navigateToPremium()">⭐️ ارتقا به نسخه کامل و نامحدود</button>
+        <div class="subscription-info-container">
+            <div id="subscription-badge" class="badge-free">نسخه رایگان</div>
+            <div id="credit-info-text" class="credit-text">در حال بررسی اعتبار...</div>
+            <button id="upgrade-btn-custom" class="upgrade-btn" onclick="navigateToPremium()">⭐️ ارتقا به نسخه کامل و نامحدود</button>
+            <div style="font-size: 0.85em; color: #6b7280; margin-top: 5px;">
+                نسخه رایگان: ۵ تصویر در روز | نسخه نامحدود: بدون محدودیت
+            </div>
         </div>
         """)
-        # ------------------------------------
+        # ----------------------------------------------------
 
         with gr.Row(equal_height=True):
             with gr.Column():
                 input_image = gr.Image(label="بارگذاری تصویر", type="pil", height=320)
-                
-                prompt = gr.Text(
-                    label="دستور ویرایش (به فارسی)",
-                    show_label=True,
-                    placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...",
-                    rtl=True,
-                    lines=3
-                )
-                
+                prompt = gr.Text(label="دستور ویرایش (به فارسی)", placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...", rtl=True, lines=3)
                 status_box = gr.HTML(label="وضعیت")
                 
-                # دکمه اصلی پردازش (elem_id="run-btn" مهم است)
+                # دکمه اجرا
                 run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn", elem_id="run-btn")
 
             with gr.Column():
                 output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
+                download_button = gr.Button("📥 دانلود تصویر", variant="secondary", elem_id="download-btn")
                 
-                download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
-                
-                with gr.Row():
-                    lora_adapter = gr.Dropdown(
-                        label="انتخاب سبک ویرایش (LoRA)",
-                        choices=list(LORA_MAPPING.keys()),
-                        value="تبدیل عکس به انیمه"
-                    )
+                lora_adapter = gr.Dropdown(label="انتخاب سبک ویرایش (LoRA)", choices=list(LORA_MAPPING.keys()), value="تبدیل عکس به انیمه")
                 
-                with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
-                    aspect_ratio_selection = gr.Dropdown(
-                        label="ابعاد تصویر خروجی",
-                        choices=ASPECT_RATIOS_LIST,
-                        value="خودکار (پیش‌فرض)",
-                        interactive=True
-                    )
-                    
+                with gr.Accordion("تنظیمات پیشرفته", open=False):
+                    aspect_ratio_selection = gr.Dropdown(label="ابعاد تصویر", choices=ASPECT_RATIOS_LIST, value="خودکار (پیش‌فرض)")
                     with gr.Row(visible=False) as custom_dims_row:
-                        custom_width = gr.Slider(
-                            label="عرض دلخواه (Width)",
-                            minimum=256, maximum=2048, step=8, value=1024
-                        )
-                        custom_height = gr.Slider(
-                            label="ارتفاع دلخواه (Height)",
-                            minimum=256, maximum=2048, step=8, value=1024
-                        )
+                        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)
+                    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):
-            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
-        )
+        # مدیریت نمایش اسلایدرهای ابعاد
+        def toggle_row(choice): return gr.update(visible=(choice == "شخصی‌سازی (Custom)"))
+        aspect_ratio_selection.change(fn=toggle_row, inputs=aspect_ratio_selection, outputs=custom_dims_row)
 
+        # مثال‌ها
         gr.Examples(
             examples=[
                 ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
-                ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم به تصویر بده.", "اصلاح نور و سایه"],
-                ["examples/4.jpg", "از فیلتر ساعت طلایی با پخش نور ملایم استفاده کن.", "نورپردازی مجدد (Relight)"],
+                ["examples/5.jpg", "سایه‌ها را حذف کن.", "اصلاح نور و سایه"],
             ],
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_image, seed, status_box],
             fn=infer_example,
-            cache_examples=False,
-            label="نمونه‌ها (برای تست کلیک کنید)"
+            label="نمونه‌ها"
+        )
+
+        # --- اتصال JS به دکمه اجرا برای ارسال Fingerprint و Status ---
+        # این تابع JS مقادیر ورودی‌ها + متغیرهای گلوبال (Fingerprint, Status) را می‌خواند و به تابع پایتون می‌فرستد
+        js_click_handler = """
+        (img, p, lora, s, rs, gs, st, ar, cw, ch) => {
+            return [img, p, lora, s, rs, gs, st, ar, cw, ch, window.userFingerprint, window.userSubscriptionStatus];
+        }
+        """
+
+        run_button.click(
+            fn=infer,
+            inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height],
+            outputs=[output_image, seed, status_box],
+            js=js_click_handler
         )
 
-    # اتصال دکمه با تابع 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="window.checkAndDeductCredit" # این خط مهم است: ابتدا JS اجرا می‌شود
-    )
-
-    download_button.click(
-        fn=None,
-        inputs=[output_image],
-        outputs=None,
-        js=js_download_func
-    )
+        download_button.click(fn=None, inputs=[output_image], js=js_download_func)
 
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(show_error=True)