Update app.py

app.py

@@ -4,6 +4,9 @@ import numpy as np
 import spaces
 import torch
 import random
+import sqlite3
+import json
+from datetime import datetime
 from PIL import Image, ImageFilter
 from typing import Iterable
 from gradio.themes import Soft
@@ -11,6 +14,71 @@ from gradio.themes.utils import colors, fonts, sizes
 from deep_translator import GoogleTranslator
 from transformers import pipeline
 
+# --- تنظیمات دیتابیس برای مدیریت کردیت کاربران ---
+DB_NAME = "user_limits.db"
+
+def init_db():
+    """ایجاد جدول برای ذخیره تعداد استفاده کاربران"""
+    conn = sqlite3.connect(DB_NAME)
+    c = conn.cursor()
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS usage_logs (
+            user_id TEXT,
+            date TEXT,
+            count INTEGER,
+            PRIMARY KEY (user_id, date)
+        )
+    ''')
+    conn.commit()
+    conn.close()
+
+init_db()
+
+def check_and_update_quota(user_id, is_premium):
+    """
+    بررسی مجاز بودن کاربر برای تولید تصویر
+    کاربران پریمیوم: همیشه مجاز
+    کاربران رایگان: حداکثر 5 تصویر در روز
+    """
+    if not user_id:
+        return True, "Guest" # اگر یوزر آیدی نبود، سختگیری نمی‌کنیم (یا می‌توان محدود کرد)
+
+    if is_premium:
+        return True, "نامحدود"
+
+    today = datetime.now().strftime("%Y-%m-%d")
+    
+    conn = sqlite3.connect(DB_NAME)
+    c = conn.cursor()
+    
+    c.execute("SELECT count FROM usage_logs WHERE user_id = ? AND date = ?", (user_id, today))
+    result = c.fetchone()
+    
+    current_count = result[0] if result else 0
+    limit = 5
+    
+    conn.close()
+    
+    if current_count >= limit:
+        return False, f"0 از {limit}"
+    
+    return True, f"{limit - current_count} از {limit}"
+
+def increment_usage(user_id, is_premium):
+    """افزایش شمارنده استفاده کاربر پس از تولید موفق"""
+    if not user_id or is_premium:
+        return
+
+    today = datetime.now().strftime("%Y-%m-%d")
+    conn = sqlite3.connect(DB_NAME)
+    c = conn.cursor()
+    
+    c.execute("INSERT OR IGNORE INTO usage_logs (user_id, date, count) VALUES (?, ?, 0)", (user_id, today))
+    c.execute("UPDATE usage_logs SET count = count + 1 WHERE user_id = ? AND date = ?", (user_id, today))
+    
+    conn.commit()
+    conn.close()
+
 # --- تعریف تم ---
 colors.steel_blue = colors.Color(
     name="steel_blue",
@@ -44,22 +112,20 @@ def is_image_nsfw(image):
         # بررسی با مدل اول
         results1 = safety_classifier_1(image)
         for result in results1:
-            if result['label'] == 'nsfw' and result['score'] > 0.5: # حساسیت متوسط
+            if result['label'] == 'nsfw' and result['score'] > 0.5:
                 print(f"Safety Check 1 Failed: {result['score']}")
                 return True
 
-        # بررسی با مدل دوم (سخت‌گیرانه برای موارد نیمه برهنه)
+        # بررسی با مدل دوم
         results2 = safety_classifier_2(image)
         for result in results2:
             label = result['label'].lower()
             score = result['score']
             
-            # اگر مدل تشخیص دهد nsfw است (حتی با احتمال ۳۰ درصد)
             if label == 'nsfw' and score > 0.3:
                 print(f"Safety Check 2 (NSFW) Failed: {score}")
                 return True
             
-            # اگر مدل تشخیص دهد تصویر سکسی یا نیمه برهنه است
             if label in ['sexy', 'porn', 'hentai'] and score > 0.4:
                 print(f"Safety Check 2 (Partial) Failed: {label} - {score}")
                 return True
@@ -67,11 +133,9 @@ def is_image_nsfw(image):
         return False
     except Exception as e:
         print(f"Safety check error: {e}")
-        # در صورت خطا در سیستم امنیتی، جانب احتیاط رعایت شود
         return True
 
-# --- لیست کلمات ممنوعه (انگلیسی) ---
-# این لیست بعد از ترجمه متن کاربر چک می‌شود
+# --- لیست کلمات ممنوعه ---
 BANNED_WORDS = [
     "nsfw", "nude", "naked", "sex", "porn", "erotic", "xxx", "18+", "uncensored",
     "breast", "nipple", "areola", "cleavage", "topless", "open chest",
@@ -84,11 +148,9 @@ BANNED_WORDS = [
 ]
 
 def check_text_safety(text):
-    """بررسی متن انگلیسی برای کلمات ممنوعه"""
     if not text: return True
     text_lower = text.lower()
     for word in BANNED_WORDS:
-        # بررسی کلمه به صورت مستقل یا چسبیده
         if word in text_lower: 
             print(f"Banned word found: {word}")
             return False
@@ -104,7 +166,7 @@ def translate_prompt(text):
         print(f"Translation Error: {e}")
         return text
 
-# --- بارگذاری مدل اصلی ویرایش تصویر ---
+# --- بارگذاری مدل اصلی ---
 from diffusers import FlowMatchEulerDiscreteScheduler
 from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
@@ -197,6 +259,16 @@ def get_success_html(message):
     </div>
     """
 
+def get_limit_error_html():
+    return f"""
+    <div style="background-color: #fff3cd; border: 1px solid #ffc107; color: #856404; padding: 16px; border-radius: 12px; text-align: center; margin-bottom: 15px; direction: rtl;">
+        <h3 style="margin: 0 0 10px 0;">🚫 محدودیت اعتبار روزانه</h3>
+        <p style="margin: 0;">شما از سهمیه ۵ تصویر رایگان امروز خود استفاده کرده‌اید.</p>
+        <p style="margin: 8px 0 0 0; font-size: 0.9em;">برای استفاده نامحدود، حساب خود را به نسخه ویژه ارتقا دهید.</p>
+        <button onclick="window.parent.postMessage({{type: 'NAVIGATE_TO_PREMIUM'}}, '*')" style="margin-top: 12px; background: linear-gradient(90deg, #FFD700, #FFA500); border: none; padding: 8px 16px; border-radius: 20px; color: black; font-weight: bold; cursor: pointer;">ارتقا به نسخه ویژه ⭐️</button>
+    </div>
+    """
+
 @spaces.GPU(duration=30)
 def infer(
     input_image,
@@ -209,17 +281,26 @@ def infer(
     aspect_ratio_selection,
     custom_width,
     custom_height,
+    user_id_hidden, # دریافت شناسه کاربر از فرانت
+    is_premium_hidden, # دریافت وضعیت پریمیوم از فرانت
     progress=gr.Progress(track_tqdm=True)
 ):
-    # 1. بررسی وجود تصویر
+    # --- بررسی اعتبار کاربر ---
+    is_premium = str(is_premium_hidden).lower() == "true"
+    allowed, limit_msg = check_and_update_quota(user_id_hidden, is_premium)
+    
+    if not allowed:
+        return None, seed, get_limit_error_html()
+
+    # --- بررسی وجود تصویر ---
     if input_image is None:
         return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
 
-    # 2. بررسی امنیتی تصویر ورودی (با دو مدل)
+    # --- بررسی امنیتی ورودی ---
     if is_image_nsfw(input_image):
         return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
 
-    # 3. ترجمه و بررسی متن
+    # --- ترجمه و بررسی متن ---
     english_prompt = translate_prompt(prompt)
     if not check_text_safety(english_prompt):
         return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است.")
@@ -233,8 +314,6 @@ def infer(
 
     generator = torch.Generator(device=device).manual_seed(seed)
     
-    # 4. تنظیم دستورات منفی (Negative Prompt) سخت‌گیرانه
-    # این کلمات به هوش مصنوعی می‌گویند که چه چیزهایی نسازد
     safety_negative = (
         "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, areola, genital, "
         "vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, "
@@ -269,10 +348,13 @@ def infer(
             true_cfg_scale=guidance_scale,
         ).images[0]
 
-        # 5. بررسی امنیتی تصویر خروجی (با دو مدل)
+        # --- بررسی امنیتی خروجی ---
         if is_image_nsfw(result):
             return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
         
+        # --- کسر اعتبار (فقط برای کاربران رایگان) ---
+        increment_usage(user_id_hidden, is_premium)
+        
         return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
 
     except Exception as e:
@@ -283,7 +365,8 @@ 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)
+    # برای مثال‌ها، آیدی کاربر فرضی و پریمیوم در نظر گرفته می‌شود تا محدودیت اعمال نشود
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "example_user", "true")
     return res, s, status
 
 # --- جاوااسکریپت برای دکمه دانلود ---
@@ -306,11 +389,85 @@ async (image) => {
 }
 """
 
-# --- جاوااسکریپت سراسری (استراتژی موبایل + بستن خودکار ۱۰ ثانیه) ---
+# --- جاوااسکریپت سراسری (شناسایی کاربر، پیام رفع محدودیت، دانلود) ---
 js_global_content = """
 <script>
 document.addEventListener('DOMContentLoaded', () => {
-    // 1. Force Light Mode
+    // --------------------------------------------------------
+    // 1. User Identification & Credit Logic (PostMessage)
+    // --------------------------------------------------------
+    const PREMIUM_PAGE_ID = '1149636'; 
+    let userId = "guest_" + Math.random().toString(36).substr(2, 9);
+    let isPremium = false;
+
+    // Listen for response from Parent (Android/Iframe)
+    window.addEventListener('message', (event) => {
+        if (event.data && event.data.type === 'USER_DATA_RESPONSE') {
+            const payload = event.data.payload;
+            
+            if (payload) {
+                try {
+                    const userObj = JSON.parse(payload);
+                    if (userObj.id) userId = userObj.id;
+                    
+                    // Check Premium Status
+                    if (userObj.isLogin && userObj.accessible_pages) {
+                        const pages = userObj.accessible_pages;
+                        // Check if premium page ID exists in allowed pages
+                        if (pages.includes(PREMIUM_PAGE_ID) || pages.includes(parseInt(PREMIUM_PAGE_ID))) {
+                            isPremium = true;
+                        }
+                    }
+                } catch (e) {
+                    console.error("Error parsing user data:", e);
+                }
+            }
+            updateUserStatusUI();
+            updateHiddenInputs();
+        }
+    });
+
+    // Request User Data on Load
+    window.parent.postMessage({ type: 'REQUEST_USER_DATA' }, '*');
+
+    function updateUserStatusUI() {
+        const badge = document.getElementById('user-status-badge');
+        if (!badge) return;
+
+        if (isPremium) {
+            badge.className = 'status-badge premium';
+            badge.innerHTML = '⭐️ نسخه نامحدود ویژه';
+        } else {
+            badge.className = 'status-badge free';
+            badge.innerHTML = '👤 نسخه رایگان (۵ تصویر در روز)';
+        }
+        badge.style.display = 'inline-flex';
+    }
+
+    function updateHiddenInputs() {
+        // Find hidden textboxes in Gradio (hacky but standard for Gradio)
+        // We look for elements with specific IDs assigned in Python
+        const idInput = document.querySelector('#hidden_user_id textarea');
+        const premiumInput = document.querySelector('#hidden_user_premium textarea');
+        
+        if (idInput) {
+            idInput.value = userId;
+            idInput.dispatchEvent(new Event('input', { bubbles: true }));
+        }
+        if (premiumInput) {
+            premiumInput.value = isPremium ? "true" : "false";
+            premiumInput.dispatchEvent(new Event('input', { bubbles: true }));
+        }
+    }
+    
+    // Retry finding elements if Gradio hasn't fully loaded them
+    setTimeout(updateHiddenInputs, 2000);
+    setTimeout(updateHiddenInputs, 5000);
+
+
+    // --------------------------------------------------------
+    // 2. Force Light Mode
+    // --------------------------------------------------------
     const forceLight = () => {
         const body = document.querySelector('body');
         if (body) {
@@ -323,22 +480,24 @@ document.addEventListener('DOMContentLoaded', () => {
     forceLight();
     setInterval(forceLight, 1000);
 
-    // 2. RETRY FUNCTION
+    // --------------------------------------------------------
+    // 3. Retry Functionality
+    // --------------------------------------------------------
     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
+    // --------------------------------------------------------
+    // 4. Quota Modal (IP Reset Guide)
+    // --------------------------------------------------------
     const showQuotaModal = () => {
         if (document.getElementById('custom-quota-modal')) return;
 
@@ -359,7 +518,6 @@ document.addEventListener('DOMContentLoaded', () => {
                             <p>نیازمند تغییر نقطه دستیابی</p>
                         </div>
                     </div>
-                    
                     <div class="guide-content">
                         <div class="info-card">
                             <div class="info-card-header">
@@ -368,15 +526,13 @@ document.addEventListener('DOMContentLoaded', () => {
                             </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 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> 
@@ -384,7 +540,6 @@ document.addEventListener('DOMContentLoaded', () => {
                             </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> 
@@ -398,30 +553,22 @@ document.addEventListener('DOMContentLoaded', () => {
                 </div>
             </div>
         `;
-        
         document.body.insertAdjacentHTML('beforeend', modalHtml);
-
-        // Auto close after 10 seconds
-        setTimeout(() => {
-            window.closeErrorModal();
-        }, 10000);
+        setTimeout(() => { window.closeErrorModal(); }, 10000);
     };
 
-    // 4. SCANNER
+    // --------------------------------------------------------
+    // 5. Error 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';
             }
@@ -431,7 +578,7 @@ document.addEventListener('DOMContentLoaded', () => {
 </script>
 """
 
-# --- CSS Updated (Larger & Auto Scroll) ---
+# --- CSS Updated ---
 css_code = """
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
@@ -439,441 +586,145 @@ css_code = """
 :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: 16px;
-    --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); }
-}
-
-.ip-reset-guide-container {
-    text-align: right; 
-    direction: rtl; 
-    background: var(--guide-bg); 
-    backdrop-filter: blur(10px);
-    padding: 20px; /* Increased Padding */
-    border-radius: var(--radius-lg-guide);
-    box-shadow: var(--shadow-xl); 
-    border: 1px solid var(--guide-border); 
-    animation: slideInUp 0.6s cubic-bezier(0.4, 0, 0.2, 1) both;
-    width: 90%; 
-    max-width: 420px; /* Increased width slightly */
-    max-height: 90vh; /* Prevent overflow on small screens */
-    overflow-y: auto; /* Enable scroll if needed */
-    position: relative; 
-    box-sizing: border-box;
-    font-family: 'Vazirmatn', sans-serif !important;
-}
-.ip-reset-guide-container::before {
-    content: ''; position: absolute; top: 0; left: 0; right: 0; height: 4px; background: var(--primary-gradient-guide);
-}
-.guide-header { display: flex; align-items: center; margin-bottom: 15px; }
-.guide-header-icon { width: 45px; height: 45px; margin-left: 15px; animation: float 3s ease-in-out infinite; flex-shrink: 0; }
-.guide-header h2 { font-size: 1.2rem; color: var(--guide-text-title); font-weight: 700; margin: 0; }
-.guide-header p { color: var(--guide-text-body); font-size: 0.8rem; margin-top: 3px; 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: 12px; margin: 12px 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: 3px; height: 100%; background: var(--primary-gradient-guide); }
-.info-card-header { display: flex; align-items: center; margin-bottom: 8px; }
-.info-card-icon { width: 18px; height: 18px; margin-left: 8px; }
-.info-card-title { font-weight: 600; color: var(--guide-text-title); font-size: 0.95rem; }
-
-.summary-section { margin-top: 12px; padding: 12px; 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: 3px; height: 100%; background: var(--success-gradient-guide); }
-.summary-header { display: flex; align-items: center; margin-bottom: 8px; }
-.summary-icon { width: 18px; height: 18px; margin-left: 8px; }
-.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: 20px 0 15px 0; width: 100%; }
-.elegant-video-button { 
-    display: inline-flex !important; 
-    align-items: center; 
-    justify-content: center; 
-    padding: 10px 20px !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);
+/* User Status Badge CSS */
+.status-badge {
+    display: none; /* Initially hidden until JS detects user */
+    align-items: center;
+    justify-content: center;
+    padding: 8px 16px;
+    border-radius: 50px;
+    font-size: 0.95em;
+    font-weight: 700;
+    margin: 10px auto 20px auto;
+    width: fit-content;
+    box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);
+    animation: fadeInBadge 0.5s ease-out;
 }
 
-#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;
+.status-badge.premium {
+    background: linear-gradient(135deg, #FFD700 0%, #FDB931 100%);
+    color: #333;
+    border: 2px solid #fff;
+    box-shadow: 0 0 15px rgba(255, 215, 0, 0.4);
 }
 
-#main-description {
-    text-align: center;
-    font-size: 1.15em;
-    color: #4b5563 !important;
-    margin-bottom: 40px;
-    line-height: 1.6;
+.status-badge.free {
+    background: #e2e8f0;
+    color: #475569;
+    border: 1px solid #cbd5e1;
 }
 
-.gr-input-label, span.label-wrap, label span {
-    font-weight: 700 !important;
-    color: #374151 !important;
-    font-size: 0.95em !important;
-    margin-bottom: 8px !important;
+@keyframes fadeInBadge {
+    from { opacity: 0; transform: translateY(-10px); }
+    to { opacity: 1; transform: translateY(0); }
 }
 
-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;
+/* Existing CSS for Guide, Buttons, etc. */
+.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(102, 126, 234, 0.2); 
+    width: 90%; max-width: 420px; max-height: 90vh; overflow-y: auto; 
     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;
-}
-
+.guide-header { display: flex; align-items: center; margin-bottom: 15px; }
+.guide-header-icon { width: 45px; height: 45px; margin-left: 15px; }
+.guide-header h2 { font-size: 1.2rem; margin: 0; }
+.guide-content { font-size: 0.9rem; line-height: 1.6; }
+.info-card, .summary-section { margin-top: 12px; padding: 12px; border-radius: 12px; }
+.info-card { background: #f0f4ff; border: 1px solid #c3dafe; }
+.summary-section { background: #f0fff4; border: 1px solid #c6f6d5; }
+.video-button-container { text-align: center; margin-top: 15px; }
+.elegant-video-button { padding: 8px 16px; border-radius: 20px; border: 1px solid #ddd; background: white; cursor: pointer; display: inline-flex; align-items: center; }
+.guide-actions { display: flex; gap: 10px; margin-top: 20px; }
+.action-button { flex: 1; padding: 10px; border-radius: 10px; border: none; cursor: pointer; display: flex; align-items: center; justify-content: center; font-weight: bold; }
+.back-button { background: white; border: 1px solid #ddd; }
+.retry-button { background: #667eea; color: white; }
+
+/* Main UI Tweaks */
+#col-container { max-width: 980px; margin: 0 auto; direction: rtl; text-align: right; background: white; padding: 20px; border-radius: 20px; }
+#main-title h1 { text-align: center; color: #2563eb; }
+#main-description { text-align: center; color: #4b5563; }
+.primary-btn { background: linear-gradient(135deg, #10b981, #059669); color: white; border: none; }
 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
 
-# استفاده از gr.Blocks
 with gr.Blocks() as demo:
-    # تزریق کدها به عنوان HTML
     gr.HTML(combined_html)
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
-        gr.Markdown(
-            "با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", 
-            elem_id="main-description"
-        )
+        gr.Markdown("با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", elem_id="main-description")
+        
+        # بج وضعیت کاربر (HTML خالی که توسط JS پر می‌شود)
+        gr.HTML('<div id="user-status-badge" class="status-badge"></div>')
+
+        # فیلدهای مخفی برای انتقال اطلاعات کاربر از JS به پایتون
+        user_id_storage = gr.Textbox(visible=False, elem_id="hidden_user_id")
+        premium_status_storage = gr.Textbox(visible=False, elem_id="hidden_user_premium")
 
         with gr.Row(equal_height=True):
             with gr.Column():
                 input_image = gr.Image(label="بارگذاری تصویر", type="pil", height=320)
-                
-                prompt = gr.Text(
-                    label="دستور ویرایش (به فارسی)",
-                    show_label=True,
-                    placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...",
-                    rtl=True,
-                    lines=3
-                )
-                
+                prompt = gr.Text(label="دستور ویرایش (به فارسی)", placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...", rtl=True, lines=3)
                 status_box = gr.HTML(label="وضعیت")
-                
-                # IMPORTANT: Added elem_id="run-btn" here for JS targeting
-                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")
 
             with gr.Column():
                 output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
-                
                 download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
                 
                 with gr.Row():
-                    lora_adapter = gr.Dropdown(
-                        label="انتخاب سبک ویرایش (LoRA)",
-                        choices=list(LORA_MAPPING.keys()),
-                        value="تبدیل عکس به انیمه"
-                    )
+                    lora_adapter = gr.Dropdown(label="انتخاب سبک", choices=list(LORA_MAPPING.keys()), value="تبدیل عکس به انیمه")
                 
-                with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
-                    aspect_ratio_selection = gr.Dropdown(
-                        label="ابعاد تصویر خروجی",
-                        choices=ASPECT_RATIOS_LIST,
-                        value="خودکار (پیش‌فرض)",
-                        interactive=True
-                    )
-                    
+                with gr.Accordion("تنظیمات پیشرفته", open=False):
+                    aspect_ratio_selection = gr.Dropdown(label="ابعاد", choices=ASPECT_RATIOS_LIST, value="خودکار (پیش‌فرض)")
                     with gr.Row(visible=False) as custom_dims_row:
-                        custom_width = gr.Slider(
-                            label="عرض دلخواه (Width)",
-                            minimum=256, maximum=2048, step=8, value=1024
-                        )
-                        custom_height = gr.Slider(
-                            label="ارتفاع دلخواه (Height)",
-                            minimum=256, maximum=2048, step=8, value=1024
-                        )
-                    
-                    seed = gr.Slider(label="دانه تصادفی (Seed)", minimum=0, maximum=MAX_SEED, step=1, value=0)
-                    randomize_seed = gr.Checkbox(label="استفاده از Seed تصادفی", value=True)
-                    guidance_scale = gr.Slider(label="میزان وفاداری به متن (Guidance Scale)", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
-                    steps = gr.Slider(label="تعداد مراحل پردازش (Steps)", minimum=1, maximum=50, step=1, value=4)
+                        custom_width = gr.Slider(label="عرض", minimum=256, maximum=2048, step=8, value=1024)
+                        custom_height = gr.Slider(label="ارتفاع", minimum=256, maximum=2048, step=8, value=1024)
+                    seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
+                    randomize_seed = gr.Checkbox(label="Seed تصادفی", value=True)
+                    guidance_scale = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
+                    steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=4)
         
-        # اصلاح تابع نمایش ردیف اسلایدرها
+        # لاجیک نمایش تنظیمات ابعاد
         def toggle_row(choice):
-            if choice == "شخصی‌سازی (Custom)":
-                return gr.update(visible=True)
-            return gr.update(visible=False)
-
-        aspect_ratio_selection.change(
-            fn=toggle_row,
-            inputs=aspect_ratio_selection,
-            outputs=custom_dims_row
+            return gr.update(visible=True) if choice == "شخصی‌سازی (Custom)" else gr.update(visible=False)
+
+        aspect_ratio_selection.change(fn=toggle_row, inputs=aspect_ratio_selection, outputs=custom_dims_row)
+
+        # اجرای مدل
+        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_id_storage, premium_status_storage  # ارسال اطلاعات کاربر به تابع infer
+            ],
+            outputs=[output_image, seed, status_box],
+            api_name="predict"
         )
 
+        download_button.click(fn=None, inputs=[output_image], outputs=None, js=js_download_func)
+        
+        # نمونه‌ها
         gr.Examples(
             examples=[
                 ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
-                ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم به تصویر بده.", "اصلاح نور و سایه"],
-                ["examples/4.jpg", "از فیلتر ساعت طلایی با پخش نور ملایم استفاده کن.", "نورپردازی مجدد (Relight)"],
-                ["examples/2.jpeg", "دوربین را ۴۵ درجه به سمت چپ بچرخان.", "تغییر زاویه دید"],
-                ["examples/7.jpg", "منبع نور را از سمت راست عقب قرار بده.", "نورپردازی چند زاویه‌ای"],
-                ["examples/10.jpeg", "کیفیت تصویر را افزایش بده (Upscale).", "افزایش کیفیت (Upscale)"],
-                ["examples/7.jpg", "منبع نور را از پایین بتابان.", "نورپردازی چند زاویه‌ای"],
-                ["examples/2.jpeg", "زاویه دوربین را به نمای بالا گوشه راست تغییر بده.", "تغییر زاویه دید"],
-                ["examples/9.jpg", "دوربین کمی به جلو حرکت می‌کند در حالی که نور خورشید از میان ابرها می‌تابد و درخششی نرم اطراف شبح شخصیت در مه ایجاد می‌کند. سبک سینمایی واقعی.", "صحنه بعدی (سینمایی)"],
-                ["examples/8.jpg", "جزئیات پوست سوژه را برجسته‌تر و طبیعی‌تر کن.", "روتوش پوست"],
-                ["examples/6.jpg", "دوربین را به نمای پایین به بالا تغییر بده.", "تغییر زاویه دید"],
+                ["examples/5.jpg", "سایه‌ها را حذف کن.", "اصلاح نور و سایه"],
             ],
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_image, seed, status_box],
             fn=infer_example,
             cache_examples=False,
-            label="نمونه‌ها (برای تست کلیک کنید)"
+            label="نمونه‌ها"
         )
 
-    run_button.click(
-        fn=infer,
-        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height],
-        outputs=[output_image, seed, status_box],
-        api_name="predict"
-    )
-
-    download_button.click(
-        fn=None,
-        inputs=[output_image],
-        outputs=None,
-        js=js_download_func
-    )
-
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(show_error=True)