Update app.py

app.py

@@ -4,14 +4,14 @@ import numpy as np
 import spaces
 import torch
 import random
+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
-import json
-from datetime import datetime
 
 # --- تعریف تم ---
 colors.steel_blue = colors.Color(
@@ -31,6 +31,58 @@ colors.steel_blue = colors.Color(
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+# --- سیستم مدیریت کاربران و سهمیه (Quota System) ---
+# ساختار: { 'fingerprint': { 'count': 0, 'date': 'YYYY-MM-DD' } }
+USER_QUOTA_DB = {}
+DAILY_LIMIT = 5
+PREMIUM_PAGE_ID = '1149636'
+PREMIUM_URL = '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991'
+
+def check_and_update_quota(fingerprint, is_paid_user):
+    """
+    بررسی می‌کند آیا کاربر مجاز به تولید تصویر است یا خیر.
+    اگر مجاز بود، کنتور را می‌اندازد.
+    """
+    if is_paid_user:
+        return True, "Unlimited"
+
+    if not fingerprint:
+        # اگر فینگرپرینت نرسید، سخت‌گیری نمی‌کنیم یا دیفالت در نظر می‌گیریم
+        return True, "Unknown"
+
+    today = datetime.now().strftime("%Y-%m-%d")
+    
+    # دریافت رکورد کاربر یا ایجاد رکورد جدید
+    user_data = USER_QUOTA_DB.get(fingerprint, {'count': 0, 'date': today})
+
+    # اگر روز عوض شده، کنتور را صفر کن
+    if user_data['date'] != today:
+        user_data = {'count': 0, 'date': today}
+
+    # بررسی محدودیت
+    if user_data['count'] >= DAILY_LIMIT:
+        return False, f"تعداد مجاز روزانه ({DAILY_LIMIT} تصویر) به پایان رسیده است."
+
+    # افزایش کنتور (این تابع باید قبل از شروع پردازش قطعی فراخوانی شود، یا بعد از موفقیت. اینجا فرض بر شروع است)
+    # اما چون ممکن است خطا رخ دهد، بهتر است اینجا فقط چک کنیم و در تابع infer اضافه کنیم.
+    # فعلا برای سادگی اینجا اضافه نمی‌کنیم، فقط وضعیت را برمی‌گردانیم.
+    return True, f"{DAILY_LIMIT - user_data['count']} اعتبار باقی‌مانده"
+
+def consume_quota(fingerprint, is_paid_user):
+    """یکی از اعتبار کاربر کم می‌کند"""
+    if is_paid_user or not fingerprint:
+        return
+    
+    today = datetime.now().strftime("%Y-%m-%d")
+    user_data = USER_QUOTA_DB.get(fingerprint, {'count': 0, 'date': today})
+    
+    if user_data['date'] != today:
+        user_data = {'count': 0, 'date': today}
+        
+    user_data['count'] += 1
+    USER_QUOTA_DB[fingerprint] = user_data
+
+
 # --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
 print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
@@ -170,6 +222,7 @@ def get_success_html(message):
     </div>
     """
 
+# --- تابع اصلی پردازش (به‌روز شده با سیستم Quota) ---
 @spaces.GPU(duration=30)
 def infer(
     input_image,
@@ -182,8 +235,19 @@ def infer(
     aspect_ratio_selection,
     custom_width,
     custom_height,
+    user_fingerprint, # ورودی جدید
+    user_status,      # ورودی جدید
     progress=gr.Progress(track_tqdm=True)
 ):
+    # 1. بررسی وضعیت کاربر و اعتبار
+    is_paid = (user_status == 'paid')
+    allowed, message = check_and_update_quota(user_fingerprint, is_paid)
+    
+    if not allowed:
+        # نمایش خطا و دکمه ارتقا (از طریق HTML خطا و JS که دکمه را نمایان میکند)
+        # ما یک سیگنال خاص در متن خطا می‌گذاریم که JS تشخیص دهد
+        return None, seed, get_error_html(f"{message} <br> لطفا برای استفاده نامحدود حساب خود را ارتقا دهید.")
+
     if input_image is None:
         return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
 
@@ -234,6 +298,9 @@ def infer(
         if is_image_nsfw(result):
             return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
         
+        # اگر موفق بود، اعتبار کسر شود
+        consume_quota(user_fingerprint, is_paid)
+        
         return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
 
     except Exception as e:
@@ -244,385 +311,249 @@ 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)
+    # مثال‌ها رایگان هستند، پس فینگرپرینت را 'demo' و وضعیت را 'paid' رد می‌کنیم تا محدودیت نداشته باشد
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "demo_user", "paid")
     return res, s, status
 
-# --- جاوااسکریپت برای دکمه دانلود ---
-js_download_func = """
-async (image) => {
-    if (!image) {
-        alert("لطفاً ابتدا تصویر را تولید کنید.");
-        return;
-    }
-    let fileUrl = image.url;
-    if (fileUrl && !fileUrl.startsWith('http')) {
-       fileUrl = window.location.origin + fileUrl;
-    } else if (!fileUrl && image.path) {
-       fileUrl = window.location.origin + "/file=" + image.path;
-    }
-    window.parent.postMessage({
-        type: 'DOWNLOAD_REQUEST',
-        url: fileUrl
-    }, '*');
-}
-"""
-
-# --- جاوااسکریپت مدیریت کاربر (رایگان/پولی) و محدودیت‌ها ---
-js_user_auth_logic = """
-<script>
-// --- تنظیمات پریمیوم ---
-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);
-}
-
-// --- بررسی اینکه آیا کاربر پولی است ---
-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)));
-}
-
-// --- به‌روزرسانی 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();
-    }
-}
-
-// --- مدیریت اعتبار کاربر رایگان (ذخیره در 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 {
-             creditInfo.innerHTML = `<span style="color:#ef4444">اعتبار امروز شما تمام شده است.</span>`;
-        }
-    }
-    
-    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'; // نمایش دکمه ارتقا
-        }
-    } 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;
-}
+# --- جاوااسکریپت و CSS ترکیبی ---
+# این بخش شامل کد تشخیص کاربر، استایل دکمه‌های جدید و لاجیک قبلی است
 
-// --- هندلر ارتقا ---
-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');
-            }
-        }
-    });
-    
-    // بررسی اولیه
-    setTimeout(() => {
-        if(window.userSubscriptionStatus === 'free') checkFreeUserCredit();
-    }, 1000);
-});
-</script>
-"""
-
-# --- CSS Updated (With Badges & Upgrade Button) ---
-css_code = """
+combined_html_js = """
 <style>
-@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700;800&display=swap');
+@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;
+    --background-fill-primary: #ffffff !important;
+    --background-fill-secondary: #f3f4f6 !important;
+    --border-color-primary: #e5e7eb !important;
+    --app-font: 'Vazirmatn', sans-serif;
+    --accent-premium: #FFC107;
+    --accent-premium-glow: rgba(255, 193, 7, 0.3);
+    color-scheme: light !important;
 }
 
-/* Badges */
+/* Badge Styles */
+@keyframes badge-fade-in {
+    from { opacity: 0; transform: translateY(-10px) scale(0.95); }
+    to { opacity: 1; transform: translateY(0) scale(1); }
+}
 #subscription-status-badge {
-    display: none; /* توسط JS کنترل می‌شود */
+    display: inline-block;
     padding: 6px 16px;
     border-radius: 20px;
     font-size: 0.9em;
     font-weight: 700;
-    margin-top: 10px;
+    margin-top: 1rem;
+    margin-bottom: 1rem;
     letter-spacing: 0.5px;
     text-shadow: 0 1px 2px rgba(0,0,0,0.1);
-    animation: fadeIn 0.5s ease;
+    animation: badge-fade-in 0.6s 0.5s ease-out backwards;
+    display: none; /* Hidden by default until JS checks status */
+    font-family: var(--app-font);
 }
-
-.free-badge {
+#subscription-status-badge.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);
+#subscription-status-badge.paid-badge {
+    background: linear-gradient(45deg, var(--accent-premium), #ffca2c);
     color: #333;
-    box-shadow: 0 4px 10px rgba(255, 193, 7, 0.4);
+    box-shadow: 0 4px 10px var(--accent-premium-glow);
 }
 
-@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 Button Styles */
+@keyframes upgrade-btn-pulse {
+    0% { transform: scale(1); box-shadow: 0 8px 20px -5px var(--accent-premium-glow); }
+    50% { transform: scale(1.02); box-shadow: 0 12px 25px -5px var(--accent-premium-glow); }
+    100% { transform: scale(1); box-shadow: 0 8px 20px -5px var(--accent-premium-glow); }
 }
-
 #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;
+    display: none;
     width: 100%;
-    animation: upgrade-pulse 2s infinite;
-    cursor: pointer !important;
-    display: none; /* پیش‌فرض مخفی */
+    margin-top: 1.5rem;
+    padding: 1rem;
+    font-family: var(--app-font);
+    font-size: 1.1em;
+    font-weight: 800;
+    color: #212529;
+    background: linear-gradient(95deg, #FFD54F, var(--accent-premium) 100%);
+    border: none;
+    border-radius: 14px;
+    cursor: pointer;
+    transition: transform 0.2s ease, box-shadow 0.2s ease;
+    box-shadow: 0 8px 20px -5px var(--accent-premium-glow);
+    animation: upgrade-btn-pulse 2.5s infinite;
+    text-align: center;
 }
 #upgrade-premium-btn:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 10px 20px -5px rgba(255, 193, 7, 0.5);
-    animation: none;
+    transform: translateY(-3px);
+    box-shadow: 0 12px 25px -5px rgba(255, 193, 7, 0.4);
+    animation-play-state: paused;
 }
 
-/* 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);
+/* --- IP Reset Guide CSS (from previous code) --- */
+:root {
+    --guide-bg: rgba(255, 255, 255, 0.98);
+    --guide-border: rgba(102, 126, 234, 0.2);
+    --primary-gradient-guide: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    --success-gradient-guide: linear-gradient(135deg, #56ab2f 0%, #a8e063 100%);
 }
 
-#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;
+.ip-reset-guide-container {
+    text-align: right; 
+    direction: rtl; 
+    background: var(--guide-bg); 
+    backdrop-filter: blur(10px);
+    padding: 20px;
+    border-radius: 16px;
+    box-shadow: 0 20px 25px -5px rgba(0,0,0,0.1); 
+    border: 1px solid var(--guide-border); 
+    width: 90%; 
+    max-width: 420px;
+    font-family: 'Vazirmatn', sans-serif !important;
 }
+.guide-header { display: flex; align-items: center; margin-bottom: 15px; }
+.guide-content { font-size: 0.9rem; color: #4a5568; line-height: 1.6; }
+.info-card { background: linear-gradient(135deg, #667eea15 0%, #764ba215 100%); border: 1px solid rgba(102, 126, 234, 0.2); border-radius: 12px; padding: 12px; margin: 12px 0; }
+.summary-section { margin-top: 12px; padding: 12px; border-radius: 12px; background: linear-gradient(135deg, #56ab2f15 0%, #a8e06315 100%); border: 1px solid rgba(86, 171, 47, 0.2); }
+.elegant-video-button { display: inline-flex; align-items: center; justify-content: center; padding: 10px 20px; background-color: #fff; color: #667eea; border: 1px solid #e2e8f0; border-radius: 50px; font-weight: 600; cursor: pointer; width: auto; margin-top:15px; }
+.guide-actions { display: flex; gap: 12px; margin-top: 20px; padding-top: 20px; border-top: 1px solid #e2e8f0; }
+.action-button { padding: 12px 15px; border: none; border-radius: 12px; font-weight: 600; cursor: pointer; flex: 1; }
+.back-button { background: white; color: #4a5568; border: 2px solid #e2e8f0; }
+.retry-button { background: var(--primary-gradient-guide); color: white; }
+
+/* Main App CSS */
+body { font-family: 'Vazirmatn', sans-serif !important; background-color: #f5f7fa !important; margin: 0; padding: 10px; }
+#col-container { margin: 0 auto; max-width: 980px; direction: rtl; text-align: right; padding: 30px; background: #ffffff !important; border-radius: 24px; box-shadow: 0 10px 40px -10px rgba(0,0,0,0.08); border: 1px solid rgba(255,255,255,0.8); }
+#main-title h1 { font-size: 2.4em !important; text-align: center; color: #1a202c !important; margin-bottom: 15px; font-weight: 800; background: -webkit-linear-gradient(45deg, #2563eb, #1e40af); -webkit-background-clip: text; -webkit-text-fill-color: transparent; }
+#main-description { text-align: center; font-size: 1.15em; color: #4b5563 !important; margin-bottom: 10px; line-height: 1.6; }
+.primary-btn { 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%; margin-top: 15px; }
+#download-btn { background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important; box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3) !important; }
+textarea, input[type="text"] { border: 2px solid #e2e8f0 !important; border-radius: 12px !important; padding: 12px !important; font-family: 'Vazirmatn', sans-serif !important; text-align: right; }
+footer { display: none !important; }
+</style>
 
-#main-description {
-    text-align: center;
-    font-size: 1.15em;
-    color: #4b5563 !important;
-    margin-bottom: 10px;
-    line-height: 1.6;
-}
+<script>
+document.addEventListener('DOMContentLoaded', () => {
+    // ---------------------------------------------
+    // 1. User Identification & Fingerprinting Logic
+    // ---------------------------------------------
+    
+    const PREMIUM_PAGE_ID = '1149636';
+    const PREMIUM_URL = '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991';
+    
+    let userFingerprint = null;
+    let userSubscriptionStatus = 'free';
+
+    // Simple Fingerprint Generator
+    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);
+    }
 
-/* Center badge container */
-.badge-container {
-    display: flex;
-    flex-direction: column;
-    align-items: center;
-    justify-content: center;
-    margin-bottom: 30px;
-}
+    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)));
+    }
 
-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;
-}
+    function updateBadge(status) {
+        const badge = document.getElementById('subscription-status-badge');
+        const upgradeBtn = document.getElementById('upgrade-premium-btn');
+        if (!badge) return;
 
-.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%;
-}
+        if (status === 'paid') {
+            badge.textContent = '⭐️ نسخه نامحد��د';
+            badge.className = 'paid-badge';
+            if(upgradeBtn) upgradeBtn.style.display = 'none';
+        } else {
+            badge.textContent = 'نسخه رایگان (۵ تصویر روزانه)';
+            badge.className = 'free-badge';
+        }
+        badge.style.display = 'inline-block';
+    }
 
-.primary-btn:disabled {
-    background: #9ca3af !important;
-    box-shadow: none !important;
-    cursor: not-allowed !important;
-}
+    // Function to populate hidden inputs for Gradio
+    function updateGradioInputs() {
+        const fpInput = document.querySelector('#user_fingerprint_input textarea');
+        const statusInput = document.querySelector('#user_status_input textarea');
+        
+        if (fpInput) {
+            fpInput.value = userFingerprint;
+            fpInput.dispatchEvent(new Event('input', { bubbles: true }));
+        }
+        if (statusInput) {
+            statusInput.value = userSubscriptionStatus;
+            statusInput.dispatchEvent(new Event('input', { bubbles: true }));
+        }
+    }
 
-#download-btn {
-    background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
-}
+    // Listen for messages from Parent App
+    window.addEventListener('message', (event) => {
+        if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
+            try {
+                const userObject = JSON.parse(event.data.payload);
+                userSubscriptionStatus = isUserPaid(userObject) ? 'paid' : 'free';
+                updateBadge(userSubscriptionStatus);
+                updateGradioInputs();
+            } catch (e) {
+                console.error("Error parsing user status", e);
+                updateBadge('free');
+            }
+        }
+    });
 
-footer { display: none !important; }
+    // Initialize
+    (async () => {
+        userFingerprint = await getBrowserFingerprint();
+        // Request User Status
+        window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
+        updateGradioInputs();
+        
+        // Setup Upgrade Button Click
+        const upgradeBtn = document.getElementById('upgrade-premium-btn');
+        if(upgradeBtn) {
+            upgradeBtn.addEventListener('click', () => {
+                 window.parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
+            });
+        }
+    })();
+    
+    // Constant check to ensure hidden inputs are populated before send
+    setInterval(updateGradioInputs, 1000);
 
-/* 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>
-"""
+    // ---------------------------------------------
+    // 2. Global Utilities (Force Light, Modals, Download)
+    // ---------------------------------------------
 
-# جاوااسکریپت سراسری (شامل راهنمای 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) {
@@ -630,51 +561,72 @@ document.addEventListener('DOMContentLoaded', () => {
             body.style.backgroundColor = '#f5f7fa';
             body.style.color = '#333333';
         }
+        document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
     };
     forceLight();
     setInterval(forceLight, 1000);
 
-    // 2. IP Reset Guide Functions
+    window.closeErrorModal = function() {
+         const modal = document.getElementById('custom-quota-modal');
+         if (modal) modal.remove();
+    };
+
     window.retryGeneration = function() {
-        const modal = document.getElementById('custom-quota-modal');
-        if (modal) modal.remove();
+        window.closeErrorModal();
         const runBtn = document.getElementById('run-btn');
         if(runBtn) runBtn.click();
     };
     
-    window.closeErrorModal = function() {
-         const modal = document.getElementById('custom-quota-modal');
-         if (modal) modal.remove();
-    };
+    // Check for "Limit Reached" error text in output to show Upgrade Button
+    setInterval(() => {
+        const errorBoxes = document.querySelectorAll('.prose'); // Where HTML output usually goes
+        let limitReached = false;
+        errorBoxes.forEach(box => {
+            if (box.innerHTML.includes('تعداد مجاز روزانه')) {
+                limitReached = true;
+            }
+        });
+        
+        const upgradeBtn = document.getElementById('upgrade-premium-btn');
+        if (upgradeBtn) {
+            if (limitReached && userSubscriptionStatus !== 'paid') {
+                upgradeBtn.style.display = 'block';
+            } else if (!limitReached && userSubscriptionStatus === 'paid') {
+                upgradeBtn.style.display = 'none';
+            }
+        }
+    }, 1000);
 
+    // Original Quota/IP Modal Logic (Preserved)
     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 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">
                     <div class="guide-header">
-                        <div>
-                            <h2>یک قدم تا ساخت تصاویر جدید</h2>
-                            <p>نیازمند تغییر نقطه دستیابی</p>
-                        </div>
+                        <h2>یک قدم تا ساخت تصاویر جدید</h2>
                     </div>
                     <div class="guide-content">
-                        <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 class="info-card"><p>طبق ویدیو آموزشی پایین بین نقطه دستیابی جابجا شوید تلاش مجدد بزنید تا تصاویر مجدداً تولید بشه.</p></div>
+                        <div class="summary-section"><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"> 
+                                <span>دیدن ویدیو آموزشی استفاده نامحدود</span> 
+                            </button>
                         </div>
                     </div>
+                    <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);
         setTimeout(() => window.closeErrorModal(), 10000);
     };
 
-    // 3. Scanner for Quota Errors
     setInterval(() => {
-        const potentialErrors = document.querySelectorAll('.toast-body, .error, div[class*="error"]');
+        const potentialErrors = document.querySelectorAll('.toast-body, .error');
         potentialErrors.forEach(el => {
             const text = el.innerText || "";
             if (text.toLowerCase().includes('quota') || text.toLowerCase().includes('exceeded')) {
@@ -682,18 +634,24 @@ document.addEventListener('DOMContentLoaded', () => {
                  el.style.display = 'none';
             }
         });
-    }, 100);
+    }, 500);
 });
+
+// Download Function
+async function handleDownload(image) {
+    if (!image) { alert("لطفاً ابتدا تصویر را تولید کنید."); return; }
+    let fileUrl = image.url;
+    if (fileUrl && !fileUrl.startsWith('http')) { fileUrl = window.location.origin + fileUrl; } 
+    else if (!fileUrl && image.path) { fileUrl = window.location.origin + "/file=" + image.path; }
+    window.parent.postMessage({ type: 'DOWNLOAD_REQUEST', url: fileUrl }, '*');
+}
 </script>
 """
 
-# ادغام نهایی
-combined_html = css_code + js_global_combined
-
 # استفاده از gr.Blocks
 with gr.Blocks() as demo:
     # تزریق کدها به عنوان HTML
-    gr.HTML(combined_html)
+    gr.HTML(combined_html_js)
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
@@ -702,13 +660,13 @@ with gr.Blocks() as demo:
             elem_id="main-description"
         )
         
-        # --- بخش جدید: بج و اطلاعات اعتبار ---
-        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>
-        """)
+        # --- المان‌های جدید برای نمایش Badge و دکمه ارتقا ---
+        gr.HTML('<div style="text-align: center;"><div id="subscription-status-badge"></div></div>')
+        gr.HTML('<button type="button" id="upgrade-premium-btn">⭐️ ارتقا به نسخه کامل و نامحدود</button>')
+
+        # --- کامپوننت‌های مخفی برای انتقال داده JS به Python ---
+        user_fingerprint = gr.Textbox(elem_id="user_fingerprint_input", visible=False)
+        user_status = gr.Textbox(elem_id="user_status_input", visible=False, value="free")
 
         with gr.Row(equal_height=True):
             with gr.Column():
@@ -724,11 +682,7 @@ with gr.Blocks() as demo:
                 
                 status_box = gr.HTML(label="وضعیت")
                 
-                # دکمه اجرا (با شناسه برای 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)
@@ -765,6 +719,7 @@ with gr.Blocks() as demo:
                     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)
@@ -781,38 +736,35 @@ with gr.Blocks() as demo:
                 ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
                 ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم به تصویر بده.", "اصلاح نور و سایه"],
                 ["examples/4.jpg", "از فیلتر ساعت طلایی با پخش نور ملایم استفاده کن.", "نورپردازی مجدد (Relight)"],
+                ["examples/2.jpeg", "دوربین را ۴۵ درجه به سمت چپ بچرخان.", "تغییر زاویه دید"],
+                ["examples/7.jpg", "منبع نور را از سمت راست عقب قرار بده.", "نورپردازی چند زاویه‌ای"],
+                ["examples/10.jpeg", "کیفیت تصویر را افزایش بده (Upscale).", "افزایش کیفیت (Upscale)"],
             ],
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_image, seed, status_box],
             fn=infer_example,
             cache_examples=False,
-            label="نمونه‌ها"
+            label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-    # --- اتصال رویدادها ---
-    
-    # دکمه اجرا: اول JS اجرا می‌شود (کسر اعتبار)، اگر موفق بود پایتون اجرا می‌شود
+    # اتصال دکمه اجرا با ورودی‌های جدید (fingerprint و status)
     run_button.click(
         fn=infer,
-        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height],
+        inputs=[
+            input_image, prompt, lora_adapter, seed, randomize_seed, 
+            guidance_scale, steps, aspect_ratio_selection, 
+            custom_width, custom_height,
+            user_fingerprint, user_status  # Added these inputs
+        ],
         outputs=[output_image, seed, status_box],
-        api_name="predict",
-        js="deductCreditAndCheck" # این تابع JS قبل از پایتون اجرا می‌شود
-    )
-
-    # دکمه ارتقا: فقط JS اجرا می‌شود (پیام به اندروید)
-    upgrade_button.click(
-        fn=None,
-        inputs=None,
-        outputs=None,
-        js="navigateToPremium"
+        api_name="predict"
     )
 
     download_button.click(
         fn=None,
         inputs=[output_image],
         outputs=None,
-        js=js_download_func
+        js="handleDownload"
     )
 
 if __name__ == "__main__":