Update app.py

app.py

@@ -4,8 +4,6 @@ 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
@@ -31,58 +29,6 @@ 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)
@@ -222,7 +168,6 @@ def get_success_html(message):
     </div>
     """
 
-# --- تابع اصلی پردازش (به‌روز شده با سیستم Quota) ---
 @spaces.GPU(duration=30)
 def infer(
     input_image,
@@ -235,19 +180,8 @@ 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("لطفاً ابتدا یک تصویر بارگذاری کنید.")
 
@@ -298,9 +232,6 @@ 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:
@@ -311,249 +242,155 @@ def infer(
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    # مثال‌ها رایگان هستند، پس فینگرپرینت را 'demo' و وضعیت را 'paid' رد می‌کنیم تا محدودیت نداشته باشد
-    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "demo_user", "paid")
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024)
     return res, s, status
 
-# --- جاوااسکریپت و CSS ترکیبی ---
-# این بخش شامل کد تشخیص کاربر، استایل دکمه‌های جدید و لاجیک قبلی است
-
-combined_html_js = """
-<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;
-    --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;
-}
-
-/* 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: inline-block;
-    padding: 6px 16px;
-    border-radius: 20px;
-    font-size: 0.9em;
-    font-weight: 700;
-    margin-top: 1rem;
-    margin-bottom: 1rem;
-    letter-spacing: 0.5px;
-    text-shadow: 0 1px 2px rgba(0,0,0,0.1);
-    animation: badge-fade-in 0.6s 0.5s ease-out backwards;
-    display: none; /* Hidden by default until JS checks status */
-    font-family: var(--app-font);
-}
-#subscription-status-badge.free-badge {
-    background: linear-gradient(45deg, #6c757d, #495057);
-    color: white;
-    box-shadow: 0 4px 10px rgba(108, 117, 125, 0.3);
-}
-#subscription-status-badge.paid-badge {
-    background: linear-gradient(45deg, var(--accent-premium), #ffca2c);
-    color: #333;
-    box-shadow: 0 4px 10px var(--accent-premium-glow);
-}
-
-/* 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 {
-    display: none;
-    width: 100%;
-    margin-top: 1.5rem;
-    padding: 1rem;
-    font-family: var(--app-font);
-    font-size: 1.1em;
-    font-weight: 800;
-    color: #212529;
-    background: linear-gradient(95deg, #FFD54F, 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(-3px);
-    box-shadow: 0 12px 25px -5px rgba(255, 193, 7, 0.4);
-    animation-play-state: paused;
-}
-
-/* --- 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%);
-}
-
-.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;
+# --- جاوااسکریپت برای دکمه دانلود ---
+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
+    }, '*');
 }
-.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>
+"""
 
+# --- منطق جاوااسکریپت برای مدیریت اعتبار و اشتراک ---
+js_credit_logic = """
 <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 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 < fingerprintString.length; i++) {
-            const char = fingerprintString.charCodeAt(i);
-            hash = ((hash << 5) - hash) + char;
+        for (let i = 0; i < str.length; i++) {
+            hash = ((hash << 5) - hash) + str.charCodeAt(i);
             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)));
-    }
-
-    function updateBadge(status) {
-        const badge = document.getElementById('subscription-status-badge');
-        const upgradeBtn = document.getElementById('upgrade-premium-btn');
-        if (!badge) return;
+    const fingerprint = getFingerprint();
+    const today = new Date().toISOString().slice(0, 10); // YYYY-MM-DD
+    const storageKey = `app_credits_${fingerprint}_${today}`;
 
-        if (status === 'paid') {
-            badge.textContent = '⭐️ نسخه نامحدود';
-            badge.className = 'paid-badge';
-            if(upgradeBtn) upgradeBtn.style.display = 'none';
+    function checkCredits() {
+        const stored = localStorage.getItem(storageKey);
+        if (stored === null) {
+            dailyCredits = 5;
+            localStorage.setItem(storageKey, 5);
         } else {
-            badge.textContent = 'نسخه رایگان (۵ تصویر روزانه)';
-            badge.className = 'free-badge';
+            dailyCredits = parseInt(stored);
         }
-        badge.style.display = 'inline-block';
+        updateUI();
     }
 
-    // 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');
+    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 (fpInput) {
-            fpInput.value = userFingerprint;
-            fpInput.dispatchEvent(new Event('input', { bubbles: true }));
-        }
-        if (statusInput) {
-            statusInput.value = userSubscriptionStatus;
-            statusInput.dispatchEvent(new Event('input', { bubbles: true }));
+        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>`;
+            }
         }
     }
 
-    // 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();
+                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);
-                updateBadge('free');
+                isUserPaid = false;
+                updateUI();
             }
         }
     });
 
-    // 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 } }, '*');
-            });
+    // درخواست وضعیت کاربر هنگام لود
+    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"); // جلوگیری از اجرا
         }
-    })();
+    };
     
-    // Constant check to ensure hidden inputs are populated before send
-    setInterval(updateGradioInputs, 1000);
+    // تابع برای دکمه ارتقا
+    window.navigateToPremium = function() {
+        parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM', payload: { url: PREMIUM_URL } }, '*');
+    };
 
-    // ---------------------------------------------
-    // 2. Global Utilities (Force Light, Modals, Download)
-    // ---------------------------------------------
+    // 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) {
@@ -566,67 +403,49 @@ document.addEventListener('DOMContentLoaded', () => {
     forceLight();
     setInterval(forceLight, 1000);
 
-    window.closeErrorModal = function() {
-         const modal = document.getElementById('custom-quota-modal');
-         if (modal) modal.remove();
-    };
-
+    // 2. RETRY FUNCTION
     window.retryGeneration = function() {
-        window.closeErrorModal();
+        const modal = document.getElementById('custom-quota-modal');
+        if (modal) modal.remove();
+        // تلاش مجدد نباید اعتبار کسر کند یا باید مدیریت شود، اما اینجا دکمه اصلی را کلیک می‌کنیم
+        // چون تابع checkAndDeductCredit روی دکمه است، دوباره اعتبار چک می‌شود.
         const runBtn = document.getElementById('run-btn');
         if(runBtn) runBtn.click();
     };
     
-    // 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);
+    // Close function
+    window.closeErrorModal = function() {
+         const modal = document.getElementById('custom-quota-modal');
+         if (modal) modal.remove();
+    };
 
-    // Original Quota/IP Modal Logic (Preserved)
+    // 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;">
+            <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">
-                        <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>
+                        <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 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);
     };
 
+    // 4. SCANNER
     setInterval(() => {
-        const potentialErrors = document.querySelectorAll('.toast-body, .error');
+        const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar, div[class*="error"]');
         potentialErrors.forEach(el => {
             const text = el.innerText || "";
             if (text.toLowerCase().includes('quota') || text.toLowerCase().includes('exceeded')) {
@@ -634,24 +453,111 @@ document.addEventListener('DOMContentLoaded', () => {
                  el.style.display = 'none';
             }
         });
-    }, 500);
+    }, 1000);
 });
+</script>
+"""
 
-// 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 }, '*');
+# --- 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;
 }
-</script>
+
+/* 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-status-container {
+    text-align: center;
+    margin-bottom: 20px;
+    padding: 10px;
+    background: #fff;
+    border-radius: 12px;
+    border: 1px solid #e5e7eb;
+}
+
+#credit-text {
+    margin-top: 8px;
+    color: #4b5563;
+    font-size: 0.95em;
+}
+
+#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;
+}
+
+@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); }
+}
+
+/* Modal and other existing styles... */
+.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
+
 # استفاده از gr.Blocks
 with gr.Blocks() as demo:
     # تزریق کدها به عنوان HTML
-    gr.HTML(combined_html_js)
+    gr.HTML(combined_html)
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
@@ -659,14 +565,16 @@ with gr.Blocks() as demo:
             "با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", 
             elem_id="main-description"
         )
-        
-        # --- المان‌های جدید برای نمایش 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")
+        # --- بخش وضعیت اشتراک و اعتبار ---
+        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>
+        """)
+        # ------------------------------------
 
         with gr.Row(equal_height=True):
             with gr.Column():
@@ -682,6 +590,7 @@ with gr.Blocks() as demo:
                 
                 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():
@@ -719,7 +628,6 @@ 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)
@@ -736,9 +644,6 @@ 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],
@@ -747,24 +652,20 @@ with gr.Blocks() as demo:
             label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-    # اتصال دکمه اجرا با ورودی‌های جدید (fingerprint و status)
+    # اتصال دکمه با تابع 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,
-            user_fingerprint, user_status  # Added these inputs
-        ],
+        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"
+        api_name="predict",
+        js="window.checkAndDeductCredit" # این خط مهم است: ابتدا JS اجرا می‌شود
     )
 
     download_button.click(
         fn=None,
         inputs=[output_image],
         outputs=None,
-        js="handleDownload"
+        js=js_download_func
     )
 
 if __name__ == "__main__":