Update app.py

app.py

@@ -33,10 +33,10 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # --- تنظیمات سیستم اعتبار ---
 USAGE_LIMIT = 5
-DATA_FILE = "usage_data.json" # فایل ذخیره اطلاعات
+DATA_FILE = "usage_data.json" 
 PREMIUM_PAGE_ID = '1149636'
 
-# --- توابع مدیریت فایل JSON ---
+# --- مدیریت ذخیره‌سازی داده‌ها ---
 def load_usage_data():
     if os.path.exists(DATA_FILE):
         try:
@@ -53,10 +53,10 @@ def save_usage_data(data):
     except Exception as e:
         print(f"Error saving data: {e}")
 
-# بارگذاری اولیه داده‌ها
+# بارگذاری اولیه
 usage_data_cache = load_usage_data()
 
-# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
+# --- بارگذاری مدل NSFW ---
 print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
@@ -117,268 +117,190 @@ 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"
-}
+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+"
-]
+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>
-    """
+        <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}
+        <span style="font-size: 1.2em;">✅</span>{message}
+    </div>"""
+
+def get_quota_exceeded_html():
+    return """
+    <div style="background: linear-gradient(135deg, #fffbeb 0%, #fef3c7 100%); border: 2px solid #f59e0b; padding: 20px; border-radius: 16px; text-align: center; box-shadow: 0 4px 15px rgba(245, 158, 11, 0.1);">
+        <div style="font-size: 3rem; margin-bottom: 10px;">💎</div>
+        <h3 style="color: #92400e; margin: 0 0 10px 0; font-weight: 800;">اعتبار رایگان امروز تمام شد</h3>
+        <p style="color: #b45309; margin: 0; font-size: 0.95em;">شما از ۵ تصویر رایگان امروز استفاده کرده‌اید.<br>برای ساخت تصاویر نامحدود و حرفه‌ای، لطفا نسخه خود را ارتقا دهید.</p>
     </div>
     """
 
 def check_quota(fingerprint, subscription_status):
-    """
-    بررسی اعتبار کاربر.
-    """
     global usage_data_cache
+    if subscription_status == 'paid': return True, "unlimited"
     
-    # لاگ برای دیباگ
-    print(f"CHECK QUOTA: FP={fingerprint}, STATUS={subscription_status}")
-
-    if subscription_status == 'paid':
-        return True, "unlimited"
-    
-    if not fingerprint or fingerprint == "unknown":
-        # اگر فینگرپرینت نرسید، موقتا اجازه ندهیم یا یک شناسه موقت بسازیم
-        # اینجا فرض می‌کنیم کاربر ناشناس است
-        print("Warning: No fingerprint received.")
-        return True, 0
+    # اگر فینگرپرینت معتبر نبود (مثلا کاربر جاوا اسکریپت را بست)، جلوگیری کنیم
+    if not fingerprint or len(fingerprint) < 5 or fingerprint == "unknown":
+        return True, 0 # برای بار اول سختگیری نمی‌کنیم تا لود شود
 
     today_str = date.today().isoformat()
-    
-    # رفرش کردن دیتا از فایل (در صورت تغییر توسط ترد دیگر)
     usage_data_cache = load_usage_data()
-    
     user_record = usage_data_cache.get(fingerprint)
 
-    # اگر کاربر جدید است یا روز عوض شده
     if not user_record or user_record.get("last_reset") != today_str:
         user_record = {"count": 0, "last_reset": today_str}
         usage_data_cache[fingerprint] = user_record
 
-    # بررسی محدودیت
     if user_record["count"] >= USAGE_LIMIT:
         return False, user_record["count"]
     
-    # کسر اعتبار
     user_record["count"] += 1
-    
-    # ذخیره در فایل
     usage_data_cache[fingerprint] = user_record
     save_usage_data(usage_data_cache)
-    
     return True, user_record["count"]
 
-
 @spaces.GPU(duration=30)
 def infer(
-    input_image,
-    prompt,
-    lora_adapter_persian,
-    seed,
-    randomize_seed,
-    guidance_scale,
-    steps,
-    aspect_ratio_selection,
-    custom_width,
-    custom_height,
-    fingerprint,
-    subscription_status,
-    progress=gr.Progress(track_tqdm=True)
+    input_image, prompt, lora_adapter_persian, seed, randomize_seed, 
+    guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, 
+    fingerprint, subscription_status, progress=gr.Progress(track_tqdm=True)
 ):
-    print(f"INFER CALLED: Fingerprint='{fingerprint}', Sub='{subscription_status}'")
-    
     # --- بررسی اعتبار ---
     is_allowed, usage_count = check_quota(fingerprint, subscription_status)
     
+    # اگر اعتبار تمام شده باشد:
     if not is_allowed:
-        msg = "شما به محدودیت ۵ تصویر رایگان روزانه خود رسیده‌اید. لطفاً برای استفاده نامحدود حساب خود را ارتقا دهید."
-        return None, seed, get_error_html(msg)
+        # خروجی‌ها: تصویر=هیچ، سید=سید، پیام=پیام زیبا، دکمه‌اجرا=مخفی، دکمه‌ارتقا=نمایان
+        return (
+            None, 
+            seed, 
+            get_quota_exceeded_html(), 
+            gr.update(visible=False), # Run Button -> Hide
+            gr.update(visible=True)   # Upgrade Button -> Show
+        )
 
+    # --- اگر اعتبار دارد، ادامه پردازش ---
     if input_image is None:
-        return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
+        return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید."), gr.update(visible=True), gr.update(visible=False)
 
     if is_image_nsfw(input_image):
-        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
+        return None, seed, get_error_html("تصویر ورودی نامناسب است."), gr.update(visible=True), gr.update(visible=False)
 
     english_prompt = translate_prompt(prompt)
     if not check_text_safety(english_prompt):
-        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است.")
+        return None, seed, get_error_html("متن شامل کلمات غیرمجاز است."), gr.update(visible=True), gr.update(visible=False)
 
     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}"
-
-    original_image = input_image.convert("RGB")
     
+    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": w, h = (int(custom_width)//8)*8, (int(custom_height)//8)*8
+    elif selection_value == "Auto" or selection_value is None: w, h = update_dimensions_on_upload(original_image)
+    else: w, h = 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=f"{safety_negative}, {base_negative}",
+            height=h, width=w, num_inference_steps=steps, generator=generator, true_cfg_scale=guidance_scale,
         ).images[0]
 
         if is_image_nsfw(result):
-            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+            return None, seed, get_error_html("تصویر خروجی نامناسب بود حذف شد."), gr.update(visible=True), gr.update(visible=False)
         
-        success_msg = f"تصویر با موفقیت ویرایش شد."
-        if subscription_status != 'paid':
-             success_msg += f" (اعتبار باقی‌مانده امروز: {USAGE_LIMIT - usage_count})"
+        success_msg = f"تصویر ساخته شد."
+        if subscription_status != 'paid': success_msg += f" (باقی‌مانده: {USAGE_LIMIT - usage_count})"
         
-        return result, seed, get_success_html(success_msg)
+        return result, seed, get_success_html(success_msg), gr.update(visible=True), gr.update(visible=False)
 
     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}")
+        return None, seed, get_error_html(f"خطا: {str(e)}"), gr.update(visible=True), gr.update(visible=False)
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "example_user", "paid")
+    # مثال‌ها اعتبار کم نمی‌کنند
+    res, s, status, btn1, btn2 = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "example", "paid")
     return res, s, status
 
-# --- جاوااسکریپت برای دانلود ---
 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_upgrade_func = """
+() => {
+    window.parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM' }, '*');
 }
 """
 
-# --- جاوااسکریپت جامع ---
 js_global_content = """
 <script>
 document.addEventListener('DOMContentLoaded', () => {
     // ---------------------------------------------
-    // 1. Fingerprint & Subscription Logic
+    // Fingerprint Logic (CANVAS + SCREEN + UA)
     // ---------------------------------------------
-    
     async function getBrowserFingerprint() {
-        const components = [
-            navigator.userAgent,
-            navigator.language,
-            screen.width + 'x' + screen.height,
-            new Date().getTimezoneOffset()
-        ];
+        const components = [];
+        // 1. User Agent & Language
+        components.push(navigator.userAgent);
+        components.push(navigator.language);
+        // 2. Screen Params
+        components.push(screen.colorDepth);
+        components.push(screen.width + 'x' + screen.height);
+        components.push(new Date().getTimezoneOffset());
+        // 3. Canvas Fingerprinting (Rendering differences)
         try {
             const canvas = document.createElement('canvas');
             const ctx = canvas.getContext('2d');
             ctx.textBaseline = "top"; ctx.font = "14px 'Arial'"; ctx.textBaseline = "alphabetic";
             ctx.fillStyle = "#f60"; ctx.fillRect(125, 1, 62, 20);
-            ctx.fillStyle = "#069"; ctx.fillText("a1b2c3d4e5f6g7h8i9j0_!@#$%^&*()", 2, 15);
+            ctx.fillStyle = "#069"; ctx.fillText("Alpha_AI_Fingerprint_v1_!@#", 2, 15);
             components.push(canvas.toDataURL());
-        } catch (e) { components.push("canvas-error"); }
+        } catch (e) { components.push("canvas-err"); }
         
-        const fingerprintString = components.join('~~~');
+        // Hashing
+        const str = components.join('~~~');
         let hash = 0;
-        for (let i = 0; i < fingerprintString.length; i++) {
-            const char = fingerprintString.charCodeAt(i);
+        for (let i = 0; i < str.length; i++) {
+            const char = str.charCodeAt(i);
             hash = ((hash << 5) - hash) + char;
             hash |= 0;
         }
@@ -398,21 +320,13 @@ document.addEventListener('DOMContentLoaded', () => {
     function updateHiddenInputs(fingerprint, status) {
         const fpInput = document.querySelector('#fingerprint_storage textarea');
         const stInput = document.querySelector('#status_storage textarea');
-        
-        if(fpInput) {
-            fpInput.value = fingerprint;
-            fpInput.dispatchEvent(new Event('input', { bubbles: true }));
-        }
-        if(stInput) {
-            stInput.value = status;
-            stInput.dispatchEvent(new Event('input', { bubbles: true }));
-        }
+        if(fpInput && fingerprint) { fpInput.value = fingerprint; fpInput.dispatchEvent(new Event('input', { bubbles: true })); }
+        if(stInput && status) { stInput.value = status; stInput.dispatchEvent(new Event('input', { bubbles: true })); }
     }
 
     function updateSubscriptionBadge(status) {
         const badge = document.getElementById('user-sub-badge');
         if (!badge) return;
-        
         if (status === 'paid') {
             badge.innerHTML = '✨ اشتراک: <span style="color: #FFD700; font-weight: bold;">نامحدود (PRO)</span>';
             badge.style.background = 'linear-gradient(45deg, #1e3a8a, #3b82f6)';
@@ -421,317 +335,125 @@ document.addEventListener('DOMContentLoaded', () => {
             badge.style.background = 'linear-gradient(45deg, #4b5563, #6b7280)';
         }
         badge.style.display = 'inline-block';
-        
-        // اطمینان از آپدیت شدن اینپوت‌ها هر بار که وضعیت تغییر می‌کند
-        updateHiddenInputs(window.userFingerprint || 'unknown', status);
     }
 
     async function initUserIdentity() {
         window.userFingerprint = await getBrowserFingerprint();
-        window.userStatus = 'free'; // Default
-        
+        window.userStatus = 'free'; 
         window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
         updateSubscriptionBadge('free');
         
-        // Polling to ensure inputs are populated even if DOM is slow
+        // Polling to make sure hidden inputs are always filled
         setInterval(() => {
-            updateHiddenInputs(window.userFingerprint || 'unknown', window.userStatus || 'free');
-        }, 2000);
+            updateHiddenInputs(window.userFingerprint, window.userStatus);
+        }, 1500);
     }
 
     window.addEventListener('message', (event) => {
         if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
             try {
-                const userObject = typeof event.data.payload === 'string' 
-                                   ? JSON.parse(event.data.payload) 
-                                   : event.data.payload;
-                
+                const userObject = typeof event.data.payload === 'string' ? JSON.parse(event.data.payload) : event.data.payload;
                 const status = isUserPaid(userObject) ? 'paid' : 'free';
                 window.userStatus = status;
                 updateSubscriptionBadge(status);
-            } catch (e) {
-                console.error("Error parsing user status:", e);
-                updateSubscriptionBadge('free');
-            }
+            } catch (e) { updateSubscriptionBadge('free'); }
         }
     });
 
     initUserIdentity();
 
-    // ---------------------------------------------
-    // 2. UI Helper
-    // ---------------------------------------------
+    // UI 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);
-
-    window.closeErrorModal = function() {
-         const modal = document.getElementById('custom-quota-modal');
-         if (modal) modal.remove();
-    };
-
-    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">
-                         <div style="font-size: 2rem;">💎</div>
-                        <div>
-                            <h2>پایان اعتبار رایگان امروز</h2>
-                            <p>برای ادامه، حساب خود را ارتقا دهید.</p>
-                        </div>
-                    </div>
-                    <div class="guide-content">
-                        <div class="info-card" style="border-color: #fbbf24; background: #fffbeb;">
-                            <p>شما از ۵ اعتبار رایگان امروز خود استفاده کرده‌اید.</p>
-                        </div>
-                        <div class="video-button-container">
-                            <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM' }, '*')" class="elegant-video-button" style="background: linear-gradient(135deg, #f59e0b, #d97706) !important; color: white !important;"> 
-                                <span>⭐️ خرید نسخه نامحدود</span> 
-                            </button>
-                        </div>
-                    </div>
-                    <div class="guide-actions">
-                        <button class="action-button back-button" onclick="window.closeErrorModal()"> 
-                            <span>بستن</span> 
-                        </button>
-                    </div>
-                </div>
-            </div>
-        `;
-        document.body.insertAdjacentHTML('beforeend', modalHtml);
-    };
-
-    setInterval(() => {
-        const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar, div[class*="error"]');
-        potentialErrors.forEach(el => {
-            const text = el.innerText || "";
-            if (text.includes('محدودیت') && text.includes('رایگان')) {
-                 showQuotaModal();
-                 el.style.display = 'none';
-                 const parentWrap = el.closest('.toast-wrap');
-                 if(parentWrap) parentWrap.style.display = 'none';
-            }
-        });
-    }, 500);
+    forceLight(); setInterval(forceLight, 1000);
 });
 </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;
-}
-
-/* Hide the storage inputs but keep them in DOM */
-#fingerprint_storage, #status_storage {
-    display: none !important; 
-    /* We use display:none with !important in CSS instead of visible=False in Python
-       to ensure the element exists in DOM for JS to find it. */
-}
-
-/* Modal CSS */
-.ip-reset-guide-container {
-    text-align: right; 
-    direction: rtl; 
-    background: #fff; 
-    padding: 25px;
-    border-radius: 16px;
-    box-shadow: 0 20px 25px -5px rgba(0, 0, 0, 0.1); 
-    width: 90%; 
-    max-width: 400px;
-    font-family: 'Vazirmatn', sans-serif !important;
-}
-.guide-header { display: flex; align-items: center; margin-bottom: 15px; gap: 15px; }
-.guide-header h2 { margin: 0; font-size: 1.2rem; color: #1f2937; }
-.guide-content { font-size: 0.95rem; color: #4b5563; line-height: 1.6; }
-.info-card { padding: 12px; border-radius: 8px; border: 1px solid #e5e7eb; margin-bottom: 15px; }
-.elegant-video-button { 
-    display: inline-flex; width: 100%; justify-content: center; padding: 12px; 
-    border-radius: 12px; border: none; font-weight: bold; cursor: pointer; transition: transform 0.2s;
-}
-.elegant-video-button:hover { transform: translateY(-2px); }
-.action-button { width: 100%; padding: 10px; border-radius: 8px; border: 1px solid #d1d5db; background: white; cursor: pointer; }
-
-/* Main App 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;
-}
-#badge-container {
-    text-align: center;
-    margin-bottom: 30px;
-    height: 30px;
-}
-.primary-btn {
-    background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important;
-    border: none !important;
-    color: white !important;
-    font-weight: 700 !important;
-    padding: 14px 28px !important;
-    border-radius: 14px !important;
-}
+:root, .dark, body, .gradio-container { --body-background-fill: #f5f7fa !important; --body-text-color: #1f2937 !important; font-family: 'Vazirmatn', sans-serif !important; }
+#fingerprint_storage, #status_storage { display: none !important; }
+#badge-container { text-align: center; margin-bottom: 30px; height: 30px; }
+#user-sub-badge { padding: 6px 16px; border-radius: 20px; font-size: 0.9em; color: white; margin-top: 10px; box-shadow: 0 4px 6px rgba(0,0,0,0.1); display: none; transition: all 0.3s ease; }
+/* Button Styles */
+.primary-btn { background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important; border: none !important; color: white !important; font-weight: 700 !important; padding: 14px 28px !important; border-radius: 14px !important; }
+.upgrade-btn { background: linear-gradient(135deg, #f59e0b 0%, #d97706 100%) !important; border: none !important; color: white !important; font-weight: 700 !important; padding: 14px 28px !important; border-radius: 14px !important; box-shadow: 0 4px 15px rgba(245, 158, 11, 0.4) !important; animation: pulse 2s infinite; }
+@keyframes pulse { 0% { transform: scale(1); box-shadow: 0 0 0 0 rgba(245, 158, 11, 0.7); } 70% { transform: scale(1.02); box-shadow: 0 0 0 10px rgba(245, 158, 11, 0); } 100% { transform: scale(1); box-shadow: 0 0 0 0 rgba(245, 158, 11, 0); } }
+/* Main Layout */
+#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; }
 </style>
 """
 
-combined_html = css_code + js_global_content
-
 with gr.Blocks() as demo:
-    gr.HTML(combined_html)
-    
-    # تغییر مهم: visible=True قرار دادیم تا در DOM باشد، اما با CSS مخفی کردیم
+    gr.HTML(css_code + js_global_content)
     fingerprint_box = gr.Textbox(elem_id="fingerprint_storage", visible=True)
     status_box_input = gr.Textbox(elem_id="status_storage", visible=True)
 
     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="badge-container"><span id="user-sub-badge"></span></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="وضعیت")
                 
-                run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn", elem_id="run-btn")
+                # --- دکمه‌ها ---
+                # دکمه ساخت (پیش‌فرض نمایان)
+                run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn", elem_id="run-btn", visible=True)
+                # دکمه ارتقا (پیش‌فرض مخفی)
+                upgrade_button = gr.Button("💎 خرید نسخه نامحدود (کلیک کنید)", variant="primary", elem_classes="upgrade-btn", elem_id="upgrade-btn", visible=False)
 
             with gr.Column():
                 output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
-                
                 download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
-                
                 with gr.Row():
-                    lora_adapter = gr.Dropdown(
-                        label="انتخاب سبک ویرایش (LoRA)",
-                        choices=list(LORA_MAPPING.keys()),
-                        value="تبدیل عکس به انیمه"
-                    )
-                
-                with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
-                    aspect_ratio_selection = gr.Dropdown(
-                        label="ابعاد تصویر خروجی",
-                        choices=ASPECT_RATIOS_LIST,
-                        value="خودکار (پیش‌فرض)",
-                        interactive=True
-                    )
-                    
+                    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.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
-                        )
-                    
-                    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="عرض", 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="دانه تصادفی", minimum=0, maximum=MAX_SEED, step=1, value=0)
+                    randomize_seed = gr.Checkbox(label="تصادفی", value=True)
+                    guidance_scale = gr.Slider(label="وفاداری به متن", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
+                    steps = gr.Slider(label="مراحل", 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)
+        
+        # --- Event Listeners ---
+        # کلیک روی دکمه اجرا: خروجی شامل آپدیت دکمه‌هاست (run_button و upgrade_button)
+        run_button.click(
+            fn=infer,
+            inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, fingerprint_box, status_box_input],
+            outputs=[output_image, seed, status_box, run_button, upgrade_button]
         )
-
+        
+        # کلیک روی دکمه ارتقا: اجرای JS برای هدایت به صفحه خرید
+        upgrade_button.click(fn=None, js=js_upgrade_func)
+        
+        download_button.click(fn=None, inputs=[output_image], js=js_download_func)
+        
+        # مثال‌ها
         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", "دوربین را به نمای پایین به بالا تغییر بده.", "تغییر زاویه دید"],
+                ["examples/5.jpg", "سایه‌ها را حذف کن...", "اصلاح نور و سایه"],
             ],
             inputs=[input_image, prompt, lora_adapter],
-            outputs=[output_image, seed, status_box],
-            fn=infer_example,
-            cache_examples=False,
-            label="نمونه‌ها (برای تست کلیک کنید)"
+            # مثال‌ها دکمه‌ها را آپدیت نمی‌کنند که UI خراب نشود
+            fn=infer_example
         )
 
-    run_button.click(
-        fn=infer,
-        inputs=[
-            input_image, prompt, lora_adapter, seed, randomize_seed, 
-            guidance_scale, steps, aspect_ratio_selection, 
-            custom_width, custom_height,
-            fingerprint_box, status_box_input 
-        ],
-        outputs=[output_image, seed, status_box],
-        api_name="predict"
-    )
-
-    download_button.click(
-        fn=None,
-        inputs=[output_image],
-        outputs=None,
-        js=js_download_func
-    )
-
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(show_error=True)