Update app.py

app.py

@@ -10,8 +10,12 @@ from gradio.themes import Soft
 from gradio.themes.utils import colors, fonts, sizes
 from deep_translator import GoogleTranslator
 from transformers import pipeline
+from datetime import date
+import json
 
-# --- تنظیمات تم و رنگ ---
+# ==========================================
+# 1. تنظیمات تم (Orange Red Theme)
+# ==========================================
 colors.orange_red = colors.Color(
     name="orange_red",
     c50="#FFF0E5",
@@ -27,15 +31,114 @@ colors.orange_red = colors.Color(
     c950="#802200",
 )
 
-# --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
+class OrangeRedTheme(Soft):
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.gray,
+        secondary_hue: colors.Color | str = colors.orange_red,
+        neutral_hue: colors.Color | str = colors.slate,
+        text_size: sizes.Size | str = sizes.text_lg,
+        font: fonts.Font | str | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("Outfit"), "Arial", "sans-serif",
+        ),
+        font_mono: fonts.Font | str | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace",
+        ),
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_hue,
+            neutral_hue=neutral_hue,
+            text_size=text_size,
+            font=font,
+            font_mono=font_mono,
+        )
+        super().set(
+            background_fill_primary="*primary_50",
+            background_fill_primary_dark="*primary_900",
+            body_background_fill="linear-gradient(135deg, *primary_200, *primary_100)",
+            body_background_fill_dark="linear-gradient(135deg, *primary_900, *primary_800)",
+            button_primary_text_color="white",
+            button_primary_text_color_hover="white",
+            button_primary_background_fill="linear-gradient(90deg, *secondary_500, *secondary_600)",
+            button_primary_background_fill_hover="linear-gradient(90deg, *secondary_600, *secondary_700)",
+            button_primary_background_fill_dark="linear-gradient(90deg, *secondary_600, *secondary_700)",
+            button_primary_background_fill_hover_dark="linear-gradient(90deg, *secondary_500, *secondary_600)",
+            button_secondary_text_color="black",
+            button_secondary_text_color_hover="white",
+            button_secondary_background_fill="linear-gradient(90deg, *primary_300, *primary_300)",
+            button_secondary_background_fill_hover="linear-gradient(90deg, *primary_400, *primary_400)",
+            button_secondary_background_fill_dark="linear-gradient(90deg, *primary_500, *primary_600)",
+            button_secondary_background_fill_hover_dark="linear-gradient(90deg, *primary_500, *primary_500)",
+            slider_color="*secondary_500",
+            slider_color_dark="*secondary_600",
+            block_title_text_weight="600",
+            block_border_width="3px",
+            block_shadow="*shadow_drop_lg",
+            button_primary_shadow="*shadow_drop_lg",
+            button_large_padding="11px",
+            color_accent_soft="*primary_100",
+            block_label_background_fill="*primary_200",
+        )
+
+orange_red_theme = OrangeRedTheme()
+
+# ==========================================
+# 2. تنظیمات سیستم اعتبار (Limit = 3)
+# ==========================================
+USAGE_LIMIT = 3
+DATA_FILE = "usage_data_fusion.json" 
+PREMIUM_PAGE_ID = '1149636'
+
+def load_usage_data():
+    if os.path.exists(DATA_FILE):
+        try:
+            with open(DATA_FILE, 'r') as f:
+                return json.load(f)
+        except:
+            return {}
+    return {}
+
+def save_usage_data(data):
+    try:
+        with open(DATA_FILE, 'w') as f:
+            json.dump(data, f)
+    except Exception as e:
+        print(f"Error saving data: {e}")
+
+usage_data_cache = load_usage_data()
+
+def get_user_record(fingerprint):
+    global usage_data_cache
+    if not fingerprint: return None
+    usage_data_cache = load_usage_data()
+    today_str = date.today().isoformat()
+    user_record = usage_data_cache.get(fingerprint)
+    if not user_record or user_record.get("last_reset") != today_str:
+        return {"count": 0, "last_reset": today_str}
+    return user_record
+
+def consume_quota(fingerprint):
+    global usage_data_cache
+    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}
+    user_record["count"] += 1
+    usage_data_cache[fingerprint] = user_record
+    save_usage_data(usage_data_cache)
+    return user_record["count"]
+
+# ==========================================
+# 3. سیستم ایمنی (Safety)
+# ==========================================
 print("Loading Safety Checker...")
-try:
-    safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
-except:
-    safety_classifier = None 
+safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
 def is_image_nsfw(image):
-    if image is None or safety_classifier is None: return False
+    if image is None: return False
     try:
         results = safety_classifier(image)
         for result in results:
@@ -46,24 +149,34 @@ def is_image_nsfw(image):
         print(f"Safety check error: {e}")
         return False
 
-# --- توابع کمکی متن ---
 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+"
+    "nsfw", "nude", "naked", "sex", "porn", "erotic", "xxx", "18+", "adult",
+    "explicit", "uncensored", "sexual", "lewd", "sensual", "lust", "horny",
+    "breast", "breasts", "nipple", "nipples", "vagina", "pussy", "cunt",
+    "penis", "dick", "cock", "genital", "genitals", "groin", "pubic",
+    "ass", "butt", "buttocks", "anus", "anal", "rectum",
+    "intercourse", "masturbation", "orgasm", "blowjob", "bj", "cum", "sperm",
+    "ejaculation", "penetration", "fucking", "sucking", "licking",
+    "lingerie", "bikini", "swimwear", "underwear", "panties", "bra", "thong",
+    "topless", "bottomless", "undressed", "unclothed", "skimpy", "transparent",
+    "see-through", "fetish", "bdsm", "bondage", "latex", "hentai", "ecchi", "ahegao",
+    "exhibitionism", "voyeur", "harem", "gore", "bloody", "blood", "kill", "murder", 
+    "dead", "torture", "abuse"
 ]
 
 def check_text_safety(text):
+    if not text: return True
     text_lower = text.lower()
+    padded_text = f" {text_lower} "
+    for char in [".", ",", "!", "?", "-", "_", "(", ")", "[", "]", "{", "}"]:
+        padded_text = padded_text.replace(char, " ")
     for word in BANNED_WORDS:
-        if f" {word} " in f" {text_lower} ": 
+        if f" {word} " in padded_text: 
             return False
     return True
 
 def translate_prompt(text):
-    if not text:
-        return ""
+    if not text: return ""
     try:
         translated = GoogleTranslator(source='auto', target='en').translate(text)
         return translated
@@ -71,442 +184,17 @@ def translate_prompt(text):
         print(f"Translation Error: {e}")
         return text
 
-# --- تنظیمات HTML/JS برای مدیریت خطا (نسخه اصلاح شده و قوی) ---
-# این بخش دقیقاً همان لاجیک برنامه قبلی برای حذف خطا و نمایش پنجره راهنماست
-js_global_content = """
-<script>
-document.addEventListener('DOMContentLoaded', () => {
-    // 1. Force Light Mode
-    const forceLight = () => {
-        const body = document.querySelector('body');
-        if (body) {
-            body.classList.remove('dark');
-            body.style.backgroundColor = '#f5f7fa';
-            body.style.color = '#333333';
-        }
-        document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
-    };
-    forceLight();
-    setInterval(forceLight, 1000);
-
-    // 2. RETRY FUNCTION
-    window.retryGeneration = function() {
-        const modal = document.getElementById('custom-quota-modal');
-        if (modal) modal.remove();
-
-        const runBtn = document.getElementById('run-btn');
-        if(runBtn) runBtn.click();
-    };
-    
-    // Close function
-    window.closeErrorModal = function() {
-         const modal = document.getElementById('custom-quota-modal');
-         if (modal) modal.remove();
-    };
-
-    // 3. SHOW MODAL FUNCTION
-    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" style="direction: rtl; text-align: right; max-width: 450px; width: 90%; background: white; border-radius: 20px; padding: 25px; box-shadow: 0 10px 40px rgba(0,0,0,0.3); animation: slideInUp 0.4s ease;">
-                    <div class="guide-header">
-                        <svg class="guide-header-icon" viewbox="0 0 100 100" xmlns="http://www.w3.org/2000/svg"> 
-                            <defs><lineargradient id="grad1" x1="0%" y1="0%" x2="100%" y2="100%"><stop offset="0%" style="stop-color: #667eea; stop-opacity: 1;"></stop><stop offset="100%" style="stop-color: #764ba2; stop-opacity: 1;"></stop></lineargradient></defs> 
-                            <circle cx="50" cy="50" r="45" fill="url(#grad1)" opacity="0.1"></circle>
-                            <circle cx="50" cy="50" r="35" fill="none" stroke="url(#grad1)" stroke-width="2" opacity="0.3"></circle>
-                            <path d="M35 50 L45 60 L65 40" stroke="url(#grad1)" stroke-width="4" fill="none" stroke-linecap="round" stroke-linejoin="round"></path>
-                            <circle cx="65" cy="35" r="8" fill="#fee140"></circle>
-                            <path d="M62 35 L68 35 M65 32 L65 38" stroke="white" stroke-width="2" stroke-linecap="round"></path> 
-                        </svg>
-                        <div>
-                            <h2>یک قدم تا ساخت تصاویر جدید</h2>
-                            <p>نیازمند تغییر نقطه دستیابی</p>
-                        </div>
-                    </div>
-                    
-                    <div class="guide-content">
-                        <div class="info-card">
-                            <div class="info-card-header">
-                                <svg class="info-card-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z" fill="#667eea" opacity="0.2"></path><path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z" stroke="#667eea" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
-                                <span class="info-card-title">راه حل سریع</span>
-                            </div>
-                            <p>طبق ویدیو آموزشی پایین بین نقطه دستیابی جابجا شوید تلاش مجدد بزنید تا تصاویر مجدداً تولید بشه.</p>
-                        </div>
-                        
-                        <div class="summary-section">
-                            <div class="summary-header">
-                                <svg class="summary-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><circle cx="12" cy="12" r="10" fill="#56ab2f" opacity="0.2"></circle><path d="M12 2C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2z" stroke="#56ab2f" stroke-width="2"></path><path d="M9 12l2 2 4-4" stroke="#56ab2f" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
-                                <span class="summary-title">خلاصه راهنما</span>
-                            </div>
-                            <div class="summary-text">هربار که این صفحه را مشاهده کردید: از اینترنت سیم‌کارت استفاده کنید، VPN را خاموش کرده و طبق ویدیو آموزشی پایین نقطه دستیابی رو تغییر دهید. «تلاش مجدد» کلیک کنید. با این روش ساده می‌توانید به صورت نامحدود تصویر بسازید! ☘️</div>
-                        </div>
-                        
-                        <div class="video-button-container">
-                            <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_URL', url: '#/nav/online/news/getSingle/1149635/eyJpdiI6IjhHVGhPQWJwb3E0cjRXbnFWTW5BaUE9PSIsInZhbHVlIjoiS1V0dTdvT21wbXAwSXZaK1RCTG1pVXZqdlFJa1hXV1RKa2FLem9zU3pXMjd5MmlVOGc2YWY0NVdNR3h3Smp1aSIsIm1hYyI6IjY1NTA5ZDYzMjAzMTJhMGQyMWQ4NjA4ZDgyNGZjZDVlY2MyNjdiMjA2NWYzOWRjY2M4ZmVjYWRlMWNlMWQ3ODEiLCJ0YWciOiIifQ==/21135210' }, '*')" class="elegant-video-button"> 
-                                <svg class="elegant-video-button-icon" xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M8 5v14l11-7z"></path></svg> 
-                                <span>دیدن ویدیو آموزشی استفاده نامحدود</span> 
-                            </button>
-                        </div>
-                    </div>
-                    
-                    <div class="guide-actions">
-                        <button class="action-button back-button" onclick="window.closeErrorModal()"> 
-                            <svg class="action-button-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M19 12H5M12 19l-7-7 7-7" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
-                            <span>بازگشت</span> 
-                        </button>
-                        <button class="action-button retry-button" onclick="window.retryGeneration()"> 
-                            <svg class="action-button-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M23 4v6h-6M1 20v-6h6" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path><path d="M20.49 9A9 9 0 0 0 5.64 5.64L1 10m22 4l-4.64 4.36A9 9 0 0 1 3.51 15" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
-                            <span>تلاش مجدد</span> 
-                        </button>
-                    </div>
-                </div>
-            </div>
-        `;
-        
-        document.body.insertAdjacentHTML('beforeend', modalHtml);
-    };
-
-    // 4. SCANNER
-    setInterval(() => {
-        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')) {
-                 
-                 showQuotaModal();
-                 
-                 // Immediately hide the Gradio error
-                 el.style.display = 'none';
-                 el.style.opacity = '0';
-                 el.innerText = '';
-                 
-                 const parentWrap = el.closest('.toast-wrap');
-                 if(parentWrap) parentWrap.style.display = 'none';
-            }
-        });
-    }, 100);
-});
-</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;
-    --background-fill-primary: #ffffff !important;
-    --background-fill-secondary: #f3f4f6 !important;
-    --border-color-primary: #e5e7eb !important;
-    --block-background-fill: #ffffff !important;
-    --block-label-text-color: #374151 !important;
-    --block-title-text-color: #111827 !important;
-    --input-background-fill: #ffffff !important;
-    color-scheme: light !important;
-}
-
-/* --- IP Reset Guide CSS --- */
-:root {
-    --guide-bg: rgba(255, 255, 255, 0.98);
-    --guide-border: rgba(102, 126, 234, 0.2);
-    --guide-text-title: #2d3748;
-    --guide-text-body: #4a5568;
-    --guide-accent: #667eea;
-    --primary-gradient-guide: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-    --success-gradient-guide: linear-gradient(135deg, #56ab2f 0%, #a8e063 100%);
-    --radius-md-guide: 12px;
-    --radius-lg-guide: 20px;
-    --shadow-sm: 0 1px 2px 0 rgba(26, 32, 44, 0.03);
-    --shadow-md: 0 4px 6px -1px rgba(26, 32, 44, 0.05), 0 2px 4px -2px rgba(26, 32, 44, 0.04);
-    --shadow-xl: 0 20px 25px -5px rgba(26, 32, 44, 0.07), 0 8px 10px -6px rgba(26, 32, 44, 0.05);
-}
-
-@keyframes float {
-    0%, 100% { transform: translateY(0px); }
-    50% { transform: translateY(-10px); }
-}
-@keyframes slideInUp {
-    from { opacity: 0; transform: translateY(30px); }
-    to { opacity: 1; transform: translateY(0); }
-}
-
-.guide-header { display: flex; align-items: center; margin-bottom: 20px; }
-.guide-header-icon { width: 50px; height: 50px; margin-left: 15px; animation: float 3s ease-in-out infinite; flex-shrink: 0; }
-.guide-header h2 { font-size: 1.3rem; color: var(--guide-text-title); font-weight: 700; margin: 0; }
-.guide-header p { color: var(--guide-text-body); font-size: 0.8rem; margin-top: 5px; margin-bottom: 0; }
-
-.guide-content { font-size: 0.9rem; color: var(--guide-text-body); 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: var(--radius-md-guide); padding: 15px; margin: 15px 0; position: relative; overflow: hidden; }
-.info-card p { font-size: 0.85rem; line-height: 1.6; margin: 0; }
-.info-card::before { content: ''; position: absolute; top: 0; right: 0; width: 4px; height: 100%; background: var(--primary-gradient-guide); }
-.info-card-header { display: flex; align-items: center; margin-bottom: 10px; }
-.info-card-icon { width: 20px; height: 20px; margin-left: 10px; }
-.info-card-title { font-weight: 600; color: var(--guide-text-title); font-size: 0.95rem; }
-
-.summary-section { margin-top: 15px; padding: 15px; border-radius: var(--radius-md-guide); background: linear-gradient(135deg, #56ab2f15 0%, #a8e06315 100%); border: 1px solid rgba(86, 171, 47, 0.2); position: relative; overflow: hidden; }
-.summary-section::before { content: ''; position: absolute; top: 0; right: 0; width: 4px; height: 100%; background: var(--success-gradient-guide); }
-.summary-header { display: flex; align-items: center; margin-bottom: 8px; }
-.summary-icon { width: 20px; height: 20px; margin-left: 10px; }
-.summary-title { font-weight: 600; color: #2f5a33; font-size: 0.95rem; }
-.summary-text { color: #2f5a33; font-size: 0.85rem; line-height: 1.6; }
-
-/* Tutorial Button */
-.video-button-container { text-align: center; margin: 25px 0 15px 0; width: 100%; }
-.elegant-video-button { 
-    display: inline-flex !important; 
-    align-items: center; 
-    justify-content: center; 
-    padding: 10px 24px !important; 
-    background-color: #fff !important; 
-    color: var(--guide-accent) !important; 
-    border: 1px solid #e2e8f0 !important; 
-    text-decoration: none; 
-    border-radius: 50px !important; 
-    font-weight: 600 !important; 
-    font-size: 0.9rem !important; 
-    cursor: pointer !important; 
-    font-family: inherit; 
-    transition: all 0.3s ease !important; 
-    box-shadow: 0 2px 10px rgba(0,0,0,0.05) !important; 
-    width: auto !important;
-}
-.elegant-video-button:hover { 
-    background: var(--primary-gradient-guide) !important; 
-    color: white !important; 
-    border-color: transparent !important; 
-    transform: translateY(-2px); 
-    box-shadow: 0 6px 16px rgba(102, 126, 234, 0.3) !important; 
-}
-.elegant-video-button-icon { width: 18px; height: 18px; margin-left: 8px; fill: currentColor; }
-
-/* Action Buttons */
-.guide-actions { 
-    display: flex !important; 
-    gap: 12px !important; 
-    margin-top: 20px; 
-    padding-top: 20px; 
-    border-top: 1px solid #e2e8f0; 
-    width: 100% !important;
-}
-.action-button { 
-    padding: 12px 15px !important; 
-    border: none !important; 
-    border-radius: 12px !important; 
-    font-size: 0.95rem !important; 
-    font-weight: 600 !important; 
-    cursor: pointer !important; 
-    flex: 1 !important; 
-    transition: all 0.3s ease !important; 
-    display: flex !important; 
-    align-items: center; 
-    justify-content: center; 
-    font-family: inherit; 
-    height: 48px !important;
-}
-.action-button-icon { width: 20px; height: 20px; margin-right: 0; margin-left: 8px; }
-
-.back-button { 
-    background: white !important; 
-    color: var(--guide-text-body) !important; 
-    border: 2px solid #e2e8f0 !important; 
-}
-.back-button:hover { 
-    background: #f7fafc !important; 
-    border-color: var(--guide-accent) !important; 
-    transform: translateY(-2px); 
-    box-shadow: var(--shadow-md) !important; 
-}
-
-.retry-button { 
-    background: var(--primary-gradient-guide) !important; 
-    color: white !important; 
-    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.3) !important; 
-}
-.retry-button:hover { 
-    transform: translateY(-2px); 
-    box-shadow: 0 6px 20px rgba(102, 126, 234, 0.4) !important; 
-}
-
-/* --- 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: 40px;
-    line-height: 1.6;
-}
-
-.gr-input-label, span.label-wrap, label span {
-    font-weight: 700 !important;
-    color: #374151 !important;
-    font-size: 0.95em !important;
-    margin-bottom: 8px !important;
-}
-
-textarea, input[type="text"] {
-    border: 2px solid #e2e8f0 !important;
-    border-radius: 12px !important;
-    background-color: #ffffff !important;
-    color: #111827 !important;
-    padding: 12px !important;
-    transition: all 0.3s ease;
-    font-family: 'Vazirmatn', sans-serif !important;
-}
-
-textarea:focus, input[type="text"]:focus {
-    border-color: #3b82f6 !important;
-    box-shadow: 0 0 0 4px rgba(59, 130, 246, 0.1) !important;
-    outline: none;
-}
-
-.gr-dropdown {
-    background: #ffffff !important;
-    border-radius: 12px !important;
-}
-
-.primary-btn, button.primary {
-    background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important;
-    border: none !important;
-    color: white !important;
-    font-weight: 700 !important;
-    font-size: 1.1em !important;
-    padding: 14px 28px !important;
-    border-radius: 14px !important;
-    box-shadow: 0 4px 15px rgba(16, 185, 129, 0.3) !important;
-    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
-    cursor: pointer !important;
-    width: 100%;
-    margin-top: 15px;
-}
-
-.primary-btn:hover, button.primary:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 8px 25px rgba(16, 185, 129, 0.45) !important;
-}
-
-.primary-btn:active, button.primary:active {
-    transform: translateY(1px);
-}
-
-#download-btn {
-    background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
-    box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3) !important;
-}
-#download-btn:hover {
-    box-shadow: 0 8px 25px rgba(59, 130, 246, 0.45) !important;
-}
-
-.gradio-container .prose table, 
-.gradio-container table {
-    background-color: #ffffff !important;
-    color: #111827 !important;
-    border: 1px solid #e5e7eb !important;
-    border-radius: 12px !important;
-    overflow: hidden !important;
-    width: 100% !important;
-    margin-top: 20px !important;
-}
-
-.gradio-container thead th {
-    background-color: #f3f4f6 !important;
-    color: #374151 !important;
-    font-weight: 700 !important;
-    border-bottom: 2px solid #e5e7eb !important;
-    padding: 12px !important;
-    text-align: right !important;
-}
-
-.gradio-container tbody tr {
-    background-color: #ffffff !important;
-    border-bottom: 1px solid #f3f4f6 !important;
-}
-
-.gradio-container tbody tr:hover {
-    background-color: #f9fafb !important;
-}
-
-.gradio-container tbody td {
-    background-color: #ffffff !important;
-    color: #374151 !important;
-    padding: 10px !important;
-}
-
-.gradio-container tbody td span, 
-.gradio-container tbody td p {
-    color: #374151 !important;
-}
-
-footer { display: none !important; }
-.flagging { display: none !important; }
-
-/* Force toast transparency */
-.toast-body {
-    direction: rtl !important;
-    text-align: right !important;
-    background: transparent !important;
-    box-shadow: none !important;
-    border: none !important;
-    padding: 0 !important;
-    max-width: 100% !important;
-    width: auto !important;
-}
-.toast-wrap {
-    background: transparent !important;
-    border: none !important;
-    box-shadow: none !important;
-}
-
-@media (prefers-color-scheme: dark) {
-    body, .gradio-container, .prose, table, tr, td, th {
-        background-color: #ffffff !important;
-        color: #333333 !important;
-    }
-}
-</style>
-"""
-
-# ادغام CSS و JS
-combined_html = css_code + js_global_content
-
+# ==========================================
+# 4. بارگذاری مدل و LoRA ها
+# ==========================================
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dtype = torch.bfloat16
 
+from diffusers import FlowMatchEulerDiscreteScheduler
+from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
+from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
+from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
+
 print("Loading Qwen Image Edit Pipeline...")
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "Qwen/Qwen-Image-Edit-2509",
@@ -535,17 +223,17 @@ try:
     pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
     print("Flash Attention 3 Processor set successfully.")
 except Exception as e:
-    print(f"Could not set FA3 processor (likely hardware mismatch): {e}. using default attention.")
+    print(f"Could not set FA3 processor: {e}. using default attention.")
 
 MAX_SEED = np.iinfo(np.int32).max
 
 LORA_MAPPING_PERSIAN = {
-    "ویرایش بافت (Texture)": "Texture Edit",
-    "طراحی لباس (Shirt Design)": "Cloth-Design-Fuse",
-    "ترکیب اشیاء (Fusion)": "Fuse-Objects",
-    "ترکیب پیشرفته (Super Fusion)": "Super-Fusion",
-    "انتقال نور (Light Migration)": "Light-Migration",
-    "انتقال متریال (Material Transfer)": "Material-Transfer"
+    "ویرایش بافت (Texture Edit)": "Texture Edit",
+    "طراحی روی لباس (Cloth-Design)": "Cloth-Design-Fuse",
+    "ترکیب اشیاء (Fuse-Objects)": "Fuse-Objects",
+    "ترکیب پیشرفته (Super-Fusion)": "Super-Fusion",
+    "انتقال نور (Light-Migration)": "Light-Migration",
+    "انتقال متریال (Material-Transfer)": "Material-Transfer"
 }
 
 def update_dimensions_on_upload(image):
@@ -563,21 +251,34 @@ def update_dimensions_on_upload(image):
     new_height = (new_height // 16) * 16
     return new_width, new_height
 
+# ==========================================
+# 5. HTML Helpers
+# ==========================================
 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; direction: rtl;">
-        <span style="font-size: 1.2em;">⛔</span>
-        {message}
-    </div>
-    """
+    return f"""<div style="background-color: #fee2e2; border: 1px solid #ef4444; color: #b91c1c; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;"><span style="font-size: 1.2em;">⛔</span>{message}</div>"""
 
 def get_success_html(message):
-    return f"""
-    <div style="background-color: #dcfce7; border: 1px solid #22c55e; color: #15803d; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px; direction: rtl;">
-        <span style="font-size: 1.2em;">✅</span>
-        {message}
-    </div>
-    """
+    return f"""<div style="background-color: #dcfce7; border: 1px solid #22c55e; color: #15803d; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;"><span style="font-size: 1.2em;">✅</span>{message}</div>"""
+
+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>"""
+
+# ==========================================
+# 6. منطق اصلی پردازش (Infer)
+# ==========================================
+def check_initial_quota(fingerprint, subscription_status):
+    if not fingerprint:
+        return gr.update(visible=True), gr.update(visible=False), None
+    if subscription_status == 'paid':
+        return gr.update(visible=True), gr.update(visible=False), None
+    
+    user_record = get_user_record(fingerprint)
+    current_usage = user_record["count"] if user_record else 0
+    
+    if current_usage >= USAGE_LIMIT:
+        return gr.update(visible=False), gr.update(visible=True), get_quota_exceeded_html()
+    else:
+        return gr.update(visible=True), gr.update(visible=False), None
 
 @spaces.GPU(duration=30)
 def infer(
@@ -589,19 +290,33 @@ def infer(
     randomize_seed,
     guidance_scale,
     steps,
+    fingerprint,
+    subscription_status,
     progress=gr.Progress(track_tqdm=True)
 ):
+    # 1. بررسی اعتبار
+    if subscription_status != 'paid':
+        user_record = get_user_record(fingerprint)
+        if user_record and user_record["count"] >= USAGE_LIMIT:
+            return None, seed, get_quota_exceeded_html(), gr.update(visible=False), gr.update(visible=True)
+
+    # 2. بررسی ورودی‌ها
     if image_1 is None or image_2 is None:
-        return None, seed, get_error_html("لطفاً هر دو تصویر (تصویر پایه و تصویر مرجع) را بارگذاری کنید.")
+        return None, seed, get_error_html("لطفاً هر دو تصویر پایه و مرجع را بارگذاری کنید."), gr.update(visible=True), gr.update(visible=False)
 
+    # 3. بررسی ایمنی تصاویر ورودی
     if is_image_nsfw(image_1) or is_image_nsfw(image_2):
-        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
+        return None, seed, get_error_html("یکی از تصاویر ورودی حاوی محتوای نامناسب است."), gr.update(visible=True), gr.update(visible=False)
 
+    # 4. ترجمه و بررسی ایمنی متن
     lora_adapter = LORA_MAPPING_PERSIAN.get(lora_adapter_persian, "Texture Edit")
     
     english_prompt = translate_prompt(prompt)
+    if not check_text_safety(english_prompt):
+        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز است."), gr.update(visible=True), gr.update(visible=False)
+
+    # اگر پرامپت خالی بود، پرامپت پیش‌فرض انگلیسی تنظیم می‌شود (نیاز به ترجمه ندارد)
     if not english_prompt:
-        # Default prompts based on internal logic
         if lora_adapter == "Cloth-Design-Fuse": english_prompt = "Put this design on their shirt."
         elif lora_adapter == "Texture Edit": english_prompt = "Apply texture to object."
         elif lora_adapter == "Fuse-Objects": english_prompt = "Fuse object into background."
@@ -609,9 +324,11 @@ def infer(
         elif lora_adapter == "Material-Transfer": english_prompt = "change materials of image1 to match the reference in image2"
         elif lora_adapter == "Light-Migration": english_prompt = "Refer to the color tone, remove the original lighting from Image 1, and relight Image 1 based on the lighting and color tone of Image 2."
 
-    if not check_text_safety(english_prompt):
-        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز است.")
+    # 5. کسر اعتبار (اگر کاربر رایگان است)
+    if subscription_status != 'paid':
+        consume_quota(fingerprint)
 
+    # 6. تنظیم آداپتر
     adapters_map = {
         "Texture Edit": "texture",
         "Fuse-Objects": "fusion",
@@ -620,7 +337,6 @@ def infer(
         "Material-Transfer": "material-transfer",
         "Light-Migration": "light-migration",
     }
-    
     active_adapter = adapters_map.get(lora_adapter)
     if active_adapter:
         pipe.set_adapters([active_adapter], adapter_weights=[1.0])
@@ -650,116 +366,250 @@ def infer(
             true_cfg_scale=guidance_scale,
         ).images[0]
 
+        # 7. بررسی ایمنی خروجی
         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)
+
+        # 8. پیام موفقیت
+        user_record = get_user_record(fingerprint)
+        remaining = USAGE_LIMIT - user_record["count"] if user_record else 0
+        success_msg = "تصویر با موفقیت ترکیب شد."
+        if subscription_status != 'paid':
+             success_msg += f" (اعتبار باقی‌مانده: {remaining})"
+        
+        if subscription_status != 'paid' and remaining <= 0:
+             return result, seed, get_success_html(success_msg), gr.update(visible=False), gr.update(visible=True)
 
-        return result, seed, get_success_html("تصویر با موفقیت ترکیب و ویرایش شد.")
+        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}")
+             raise e # Raise for JS Scanner
+        return None, seed, get_error_html(f"خطا در پردازش: {error_str}"), gr.update(visible=True), gr.update(visible=False)
 
 @spaces.GPU(duration=30)
-def infer_example(image_1, image_2, prompt, lora_adapter_persian):
-    res, s, status = infer(image_1, image_2, prompt, lora_adapter_persian, 0, True, 1.0, 4)
-    return res, s, status
+def infer_example(image_1, image_2, prompt, lora_adapter):
+    res, s, status, btn1, btn2 = infer(image_1, image_2, prompt, lora_adapter, 0, True, 1.0, 4, "example", "paid")
+    return res, s
 
-# --- رابط کاربری ---
+# ==========================================
+# 7. JS & CSS
+# ==========================================
 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 }, '*');
 }
 """
 
-with gr.Blocks() as demo:
-    gr.HTML(combined_html)
+js_upgrade_func = """() => { window.parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM' }, '*'); }"""
+
+js_global = """
+<script>
+document.addEventListener('DOMContentLoaded', () => {
+    async function getBrowserFingerprint() {
+        const components = [navigator.userAgent, navigator.language, screen.colorDepth, 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("Fusion_FP_v1", 2, 15);
+            components.push(canvas.toDataURL());
+        } catch (e) { components.push("err"); }
+        const str = components.join('~~~');
+        let hash = 0; for (let i = 0; i < str.length; i++) { hash = ((hash << 5) - hash) + str.charCodeAt(i); hash |= 0; }
+        return 'fp_' + Math.abs(hash).toString(16);
+    }
+
+    function isUserPaid(userObject) {
+        const PREMIUM_PAGE_ID = '1149636'; 
+        if (userObject && userObject.isLogin && userObject.accessible_pages) {
+            if (Array.isArray(userObject.accessible_pages)) { return userObject.accessible_pages.some(page => String(page) === String(PREMIUM_PAGE_ID)); }
+        }
+        return false;
+    }
+
+    function updateHiddenInputs(fingerprint, status) {
+        const fpInput = document.querySelector('#fingerprint_storage textarea');
+        const stInput = document.querySelector('#status_storage textarea');
+        if(fpInput && fingerprint && fpInput.value !== fingerprint) { fpInput.value = fingerprint; fpInput.dispatchEvent(new Event('input', { bubbles: true })); }
+        if(stInput && status && stInput.value !== status) { stInput.value = status; stInput.dispatchEvent(new Event('input', { bubbles: true })); }
+    }
+
+    function updateBadge(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, #FF4500, #FF8533)';
+        } else {
+            badge.innerHTML = '👤 اشتراک: <span style="color: #fff; font-weight: bold;">رایگان (۳ تصویر روزانه)</span>';
+            badge.style.background = 'linear-gradient(45deg, #4b5563, #6b7280)';
+        }
+        badge.style.display = 'inline-block';
+    }
+
+    async function init() {
+        window.userFingerprint = await getBrowserFingerprint();
+        window.userStatus = 'free'; 
+        window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
+        updateBadge('free');
+        updateHiddenInputs(window.userFingerprint, window.userStatus);
+        setInterval(() => { if(window.userFingerprint) updateHiddenInputs(window.userFingerprint, window.userStatus || 'free'); }, 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 status = isUserPaid(userObject) ? 'paid' : 'free';
+                window.userStatus = status;
+                updateBadge(status);
+                updateHiddenInputs(window.userFingerprint, status);
+            } catch (e) { console.error(e); }
+        }
+    });
+
+    init();
+
+    // GPU QUOTA MODAL
+    window.retryGeneration = function() { document.getElementById('custom-quota-modal')?.remove(); document.getElementById('run-btn')?.click(); };
+    window.closeErrorModal = function() { document.getElementById('custom-quota-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" style="background: rgba(255, 255, 255, 0.98); padding: 20px; border-radius: 16px; width: 90%; max-width: 420px; text-align: right; direction: rtl;">
+                    <h2 style="color: #2d3748; margin-bottom: 10px;">یک قدم تا ساخت تصاویر جدید</h2>
+                    <p style="color: #4a5568; font-size: 0.9rem;">نیازمند تغییر نقطه دستیابی (IP) هستید.</p>
+                    <div style="background: #FFF0E5; padding: 12px; border-radius: 8px; margin: 10px 0; border: 1px solid #FF8533;">
+                        <p style="margin:0; font-size: 0.85rem; color: #992900;">از اینترنت سیم‌کارت استفاده کنید، VPN را خاموش کرده، نقطه دستیابی را طبق ویدیو تغییر دهید و «تلاش مجدد» بزنید.</p>
+                    </div>
+                     <div style="text-align:center; margin: 15px 0;">
+                        <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_URL', url: '#/nav/online/news/getSingle/1149635/eyJpdiI6IjhHVGhPQWJwb3E0cjRXbnFWTW5BaUE9PSIsInZhbHVlIjoiS1V0dTdvT21wbXAwSXZaK1RCTG1pVXZqdlFJa1hXV1RKa2FLem9zU3pXMjd5MmlVOGc2YWY0NVdNR3h3Smp1aSIsIm1hYyI6IjY1NTA5ZDYzMjAzMTJhMGQyMWQ4NjA4ZDgyNGZjZDVlY2MyNjdiMjA2NWYzOWRjY2M4ZmVjYWRlMWNlMWQ3ODEiLCJ0YWciOiIifQ==/21135210' }, '*')" style="padding: 8px 16px; background: white; border: 1px solid #ccc; border-radius: 20px; cursor: pointer;">ویدیو آموزشی</button>
+                    </div>
+                    <div style="display: flex; gap: 10px;">
+                        <button onclick="window.closeErrorModal()" style="flex:1; padding: 10px; background: white; border: 1px solid #ccc; border-radius: 8px; cursor: pointer;">بازگشت</button>
+                        <button onclick="window.retryGeneration()" style="flex:1; padding: 10px; background: linear-gradient(135deg, #FF4500, #FF8533); color: white; border: none; border-radius: 8px; cursor: pointer;">تلاش مجدد</button>
+                    </div>
+                </div>
+            </div>`;
+        document.body.insertAdjacentHTML('beforeend', modalHtml);
+        setTimeout(() => window.closeErrorModal(), 10000);
+    };
+
+    setInterval(() => {
+        document.querySelectorAll('.toast-body, .error, div[class*="error"]').forEach(el => {
+            if ((el.innerText||"").toLowerCase().includes('quota')) {
+                showQuotaModal();
+                el.style.display = 'none';
+            }
+        });
+    }, 100);
+    
+    // Force Light Mode
+    setInterval(() => {
+        document.body.classList.remove('dark');
+        document.body.style.backgroundColor = '#FFF0E5';
+    }, 1000);
+});
+</script>
+"""
+
+css_code = """
+<style>
+@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
+:root, body, .gradio-container { font-family: 'Vazirmatn', sans-serif !important; }
+#fingerprint_storage, #status_storage { display: none !important; }
+#col-container { margin: 0 auto; max-width: 1100px; direction: rtl; text-align: right; background: white; padding: 30px; border-radius: 20px; }
+#badge-container { text-align: center; margin-bottom: 20px; height: 30px; }
+#user-sub-badge { padding: 6px 16px; border-radius: 20px; font-size: 0.9em; color: white; display: none; }
+.primary-btn { width: 100%; margin-top: 15px; background: linear-gradient(135deg, #FF4500, #FF8533); color: white; border: none; }
+.upgrade-btn { width: 100%; margin-top: 15px; background: linear-gradient(135deg, #f59e0b, #d97706); color: white; border: none; animation: pulse 2s infinite; }
+@keyframes pulse { 0% { transform: scale(1); } 50% { transform: scale(1.02); } 100% { transform: scale(1); } }
+footer { display: none !important; }
+</style>
+"""
+
+# ==========================================
+# 8. رابط کاربری (Gradio Blocks)
+# ==========================================
+with gr.Blocks(theme=orange_red_theme) as demo:
+    gr.HTML(css_code + js_global)
+    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("# **ویرایشگر ترکیبی هوشمند (Fusion)**", elem_id="main-title")
+        gr.Markdown("ترکیب، انتقال بافت و ویرایش دو تصویر با هوش مصنوعی آلفا.")
+        gr.HTML('<div id="badge-container"><span id="user-sub-badge"></span></div>')
+
         with gr.Row(equal_height=True):
             with gr.Column(scale=1):
                 with gr.Row():
-                    image_1 = gr.Image(label="۱. تصویر پایه (اصلی)", type="pil", height=290)
-                    image_2 = gr.Image(label="۲. تصویر مرجع (استایل/بافت)", type="pil", height=290)
+                    image_1 = gr.Image(label="تصویر پایه (Base)", type="pil", height=290)
+                    image_2 = gr.Image(label="تصویر مرجع (Reference)", type="pil", height=290)
 
                 prompt = gr.Text(
-                    label="دستور ویرایش (به فارسی)",
+                    label="دستور ویرایش (فارسی)",
                     show_label=True,
-                    placeholder="مثال: بافت چوب را روی لیوان اعمال کن...",
-                    rtl=True,
-                    lines=3
+                    placeholder="مثال: بافت تصویر دوم را روی لیوان اعمال کن...",
+                    rtl=True
                 )
                 
                 status_box = gr.HTML(label="وضعیت")
                 
-                # IMPORTANT: elem_id="run-btn" for JS
-                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")
+                upgrade_button = gr.Button("💎 ارتقا به نسخه نامحدود", variant="primary", elem_classes="upgrade-btn", elem_id="upgrade-btn", visible=False)
 
                 with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
-                    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)
+                    seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
+                    randomize_seed = gr.Checkbox(label="تصادفی", 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)
 
             with gr.Column(scale=1):
                 output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=350)
-                download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
+                download_button = gr.Button("📥 دانلود تصویر", variant="secondary", elem_classes="primary-btn")
                 
                 with gr.Row():
                     lora_adapter = gr.Dropdown(
-                        label="انتخاب سبک ویرایش",
+                        label="نوع عملیات (LoRA)",
                         choices=list(LORA_MAPPING_PERSIAN.keys()),
-                        value="ویرایش بافت (Texture)",
+                        value="ویرایش بافت (Texture Edit)",
                     )        
  
         gr.Examples(
             examples=[
-                ["examples/M1.jpg", "examples/M2.jpg", "با توجه به تن رنگ، نورپردازی اصلی تصویر ۱ را حذف کن و بر اساس نور و رنگ تصویر ۲ مجدداً نورپردازی کن.", "انتقال نور (Light Migration)"],
-                ["examples/Cloth2.jpg", "examples/Design2.png", "این طرح را روی پیراهن قرار بده.", "طراحی لباس (Shirt Design)"],
-                ["examples/Cup1.png", "examples/Wood1.png", "بافت چوب را روی لیوان اعمال کن.", "ویرایش بافت (Texture)"],
-                ["examples/Cloth1.jpg", "examples/Design1.png", "این طرح را روی پیراهن قرار بده.", "طراحی لباس (Shirt Design)"],
-                ["examples/F3.jpg", "examples/F4.jpg", "عینک او را با عینک جدید از تصویر ۱ جایگزین کن.", "ترکیب پیشرفته (Super Fusion)"],
-                ["examples/Chair.jpg", "examples/Material.jpg", "متریال تصویر ۱ را تغییر بده تا با مرجع تصویر ۲ مطابقت داشته باشد.", "انتقال متریال (Material Transfer)"],
-                ["examples/F1.jpg", "examples/F2.jpg", "بطری کوچک را روی میز قرار بده.", "ترکیب پیشرفته (Super Fusion)"],
-                ["examples/Mug1.jpg", "examples/Texture1.jpg", "طرح تصویر ۲ را روی لیوان اعمال کن.", "ویرایش بافت (Texture)"],
-                ["examples/Cat1.jpg", "examples/Glass1.webp", "یک گربه که عینک زده است.", "ترکیب اشیاء (Fusion)"],
+                ["examples/M1.jpg", "examples/M2.jpg", "نورپردازی تصویر اول را حذف کن و بر اساس نور و رنگ تصویر دوم آن را مجدداً نورپردازی کن.", "انتقال نور (Light-Migration)"],
+                ["examples/Cloth2.jpg", "examples/Design2.png", "این طرح را روی پیراهن قرار بده.", "طراحی روی لباس (Cloth-Design)"],
+                ["examples/Cup1.png", "examples/Wood1.png", "بافت چوب را روی ماگ اعمال کن.", "ویرایش بافت (Texture Edit)"],
+                ["examples/F3.jpg", "examples/F4.jpg", "عینک او را با عینک جدید از تصویر ۱ جایگزین کن.", "ترکیب پیشرفته (Super-Fusion)"],
+                ["examples/Chair.jpg", "examples/Material.jpg", "متریال تصویر ۱ را دقیقاً شبیه به مرجع تصویر ۲ کن.", "انتقال متریال (Material-Transfer)"],
+                ["examples/Cat1.jpg", "examples/Glass1.webp", "یک گربه که عینک تصویر دوم را زده است.", "ترکیب اشیاء (Fuse-Objects)"],
             ],
             inputs=[image_1, image_2, prompt, lora_adapter],
-            outputs=[output_image, seed, status_box],
+            outputs=[output_image, seed],
             fn=infer_example,
             cache_examples=False,
-            label="نمونه‌ها (برای تست کلیک کنید)"
+            label="نمونه‌ها"
         )
 
+    # اتصال رویدادها
+    fingerprint_box.change(fn=check_initial_quota, inputs=[fingerprint_box, status_box_input], outputs=[run_button, upgrade_button, status_box])
+    
     run_button.click(
         fn=infer,
-        inputs=[image_1, image_2, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps],
-        outputs=[output_image, seed, status_box],
-        api_name="predict"
+        inputs=[image_1, image_2, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, fingerprint_box, status_box_input],
+        outputs=[output_image, seed, status_box, run_button, upgrade_button]
     )
-    
-    download_button.click(
-        fn=None,
-        inputs=[output_image],
-        outputs=None,
-        js=js_download_func
-    )
-    
+
+    upgrade_button.click(fn=None, js=js_upgrade_func)
+    download_button.click(fn=None, inputs=[output_image], js=js_download_func)
+
 if __name__ == "__main__":
-    demo.queue(max_size=50).launch(show_error=True)
+    demo.queue(max_size=30).launch(show_error=True)