Update app.py

app.py

@@ -161,29 +161,62 @@ def get_quota_exceeded_html():
     </div>
     """
 
-def check_quota(fingerprint, subscription_status):
+def get_user_record(fingerprint):
+    """خواندن رکورد کاربر از دیتابیس"""
     global usage_data_cache
-    if subscription_status == 'paid': return True, "unlimited"
+    if not fingerprint: return None
     
-    # اگر فینگرپرینت معتبر نبود (مثلا کاربر جاوا اسکریپت را بست)، جلوگیری کنیم
-    if not fingerprint or len(fingerprint) < 5 or fingerprint == "unknown":
-        return True, 0 # برای بار اول سختگیری نمی‌کنیم تا لود شود
+    # Reload to get fresh data
+    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}
-        usage_data_cache[fingerprint] = user_record
-
-    if user_record["count"] >= USAGE_LIMIT:
-        return False, user_record["count"]
     
     user_record["count"] += 1
     usage_data_cache[fingerprint] = user_record
     save_usage_data(usage_data_cache)
-    return True, user_record["count"]
+    return user_record["count"]
+
+# --- تابع جدید برای بررسی اولیه (بدون GPU) ---
+def check_initial_quota(fingerprint, subscription_status):
+    """
+    این تابع فقط برای بررسی وضعیت دکمه‌ها در لحظه لود شدن صفحه است.
+    هیچ GPUای مصرف نمی‌کند.
+    """
+    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(
@@ -191,21 +224,19 @@ def infer(
     guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, 
     fingerprint, subscription_status, progress=gr.Progress(track_tqdm=True)
 ):
-    # --- بررسی اعتبار ---
-    is_allowed, usage_count = check_quota(fingerprint, subscription_status)
-    
-    # اگر اعتبار تمام شده باشد:
-    if not is_allowed:
-        # خروجی‌ها: تصویر=هیچ، سید=سید، پیام=پیام زیبا، دکمه‌اجرا=مخفی، دکمه‌ارتقا=نمایان
-        return (
-            None, 
-            seed, 
-            get_quota_exceeded_html(), 
-            gr.update(visible=False), # Run Button -> Hide
-            gr.update(visible=True)   # Upgrade Button -> Show
-        )
+    # بررسی مجدد اعتبار در لحظه اجرا (جهت اطمینان)
+    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)
+            )
+        
+        # اگر اعتبار داشت، حالا کم می‌کنیم
+        consume_quota(fingerprint)
 
-    # --- اگر اعتبار دارد، ادامه پردازش ---
+    # --- ادامه منطق ساخت تصویر ---
     if input_image is None:
         return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید."), gr.update(visible=True), gr.update(visible=False)
 
@@ -240,8 +271,16 @@ def infer(
         if is_image_nsfw(result):
             return None, seed, get_error_html("تصویر خروجی نامناسب بود حذف شد."), gr.update(visible=True), gr.update(visible=False)
         
+        # دریافت تعداد باقی‌مانده برای نمایش
+        user_record = get_user_record(fingerprint)
+        remaining = USAGE_LIMIT - user_record["count"] if user_record else 0
+        
         success_msg = f"تصویر ساخته شد."
-        if subscription_status != 'paid': success_msg += f" (باقی‌مانده: {USAGE_LIMIT - usage_count})"
+        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(success_msg), gr.update(visible=True), gr.update(visible=False)
 
@@ -250,7 +289,6 @@ def infer(
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    # مثال‌ها اعتبار کم نمی‌کنند
     res, s, status, btn1, btn2 = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "example", "paid")
     return res, s, status
 
@@ -264,7 +302,6 @@ async (image) => {
 }
 """
 
-# جاوا اسکریپت برای دکمه ارتقا
 js_upgrade_func = """
 () => {
     window.parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM' }, '*');
@@ -274,19 +311,13 @@ js_upgrade_func = """
 js_global_content = """
 <script>
 document.addEventListener('DOMContentLoaded', () => {
-    // ---------------------------------------------
-    // Fingerprint Logic (CANVAS + SCREEN + UA)
-    // ---------------------------------------------
     async function getBrowserFingerprint() {
         const components = [];
-        // 1. User Agent & Language
         components.push(navigator.userAgent);
         components.push(navigator.language);
-        // 2. Screen Params
         components.push(screen.colorDepth);
         components.push(screen.width + 'x' + screen.height);
         components.push(new Date().getTimezoneOffset());
-        // 3. Canvas Fingerprinting (Rendering differences)
         try {
             const canvas = document.createElement('canvas');
             const ctx = canvas.getContext('2d');
@@ -296,7 +327,6 @@ document.addEventListener('DOMContentLoaded', () => {
             components.push(canvas.toDataURL());
         } catch (e) { components.push("canvas-err"); }
         
-        // Hashing
         const str = components.join('~~~');
         let hash = 0;
         for (let i = 0; i < str.length; i++) {
@@ -320,8 +350,14 @@ document.addEventListener('DOMContentLoaded', () => {
     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.dispatchEvent(new Event('input', { bubbles: true })); }
-        if(stInput && status) { stInput.value = status; stInput.dispatchEvent(new Event('input', { bubbles: true })); }
+        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 updateSubscriptionBadge(status) {
@@ -343,10 +379,8 @@ document.addEventListener('DOMContentLoaded', () => {
         window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
         updateSubscriptionBadge('free');
         
-        // Polling to make sure hidden inputs are always filled
-        setInterval(() => {
-            updateHiddenInputs(window.userFingerprint, window.userStatus);
-        }, 1500);
+        // Initial Update
+        updateHiddenInputs(window.userFingerprint, window.userStatus);
     }
 
     window.addEventListener('message', (event) => {
@@ -356,13 +390,20 @@ document.addEventListener('DOMContentLoaded', () => {
                 const status = isUserPaid(userObject) ? 'paid' : 'free';
                 window.userStatus = status;
                 updateSubscriptionBadge(status);
+                updateHiddenInputs(window.userFingerprint, status);
             } catch (e) { updateSubscriptionBadge('free'); }
         }
     });
 
     initUserIdentity();
+    
+    // Polling to ensure sync
+    setInterval(() => {
+        if(window.userFingerprint) {
+            updateHiddenInputs(window.userFingerprint, window.userStatus || 'free');
+        }
+    }, 2000);
 
-    // UI Force Light Mode
     const forceLight = () => {
         const body = document.querySelector('body');
         if (body) { body.classList.remove('dark'); body.style.backgroundColor = '#f5f7fa'; body.style.color = '#333333'; }
@@ -408,9 +449,7 @@ with gr.Blocks() as demo:
                 status_box = gr.HTML(label="وضعیت")
                 
                 # --- دکمه‌ها ---
-                # د��مه ساخت (پیش‌فرض نمایان)
                 run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn", elem_id="run-btn", visible=True)
-                # دکمه ارتقا (پیش‌فرض مخفی)
                 upgrade_button = gr.Button("💎 خرید نسخه نامحدود (کلیک کنید)", variant="primary", elem_classes="upgrade-btn", elem_id="upgrade-btn", visible=False)
 
             with gr.Column():
@@ -431,27 +470,26 @@ with gr.Blocks() as demo:
         def toggle_row(choice): return gr.update(visible=(choice == "شخصی‌سازی (Custom)"))
         aspect_ratio_selection.change(fn=toggle_row, inputs=aspect_ratio_selection, outputs=custom_dims_row)
         
-        # --- Event Listeners ---
-        # کلیک روی دکمه اجرا: خروجی شامل آپدیت دکمه‌هاست (run_button و upgrade_button)
+        # --- سیستم چک کردن خودکار اعتبار در هنگام لود صفحه ---
+        # به محض اینکه JS اثرانگشت را در باکس مخفی می‌ریزد، این تابع اجرا می‌شود
+        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=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height, fingerprint_box, status_box_input],
             outputs=[output_image, seed, status_box, run_button, upgrade_button]
         )
         
-        # کلیک روی دکمه ارتقا: اجرای JS برای هدایت به صفحه خرید
         upgrade_button.click(fn=None, js=js_upgrade_func)
-        
         download_button.click(fn=None, inputs=[output_image], js=js_download_func)
         
-        # مثال‌ها
         gr.Examples(
-            examples=[
-                ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
-                ["examples/5.jpg", "سایه‌ها را حذف کن...", "اصلاح نور و سایه"],
-            ],
+            examples=[["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"]],
             inputs=[input_image, prompt, lora_adapter],
-            # مثال‌ها دکمه‌ها را آپدیت نمی‌کنند که UI خراب نشود
             fn=infer_example
         )