Update app.py

app.py

@@ -33,8 +33,28 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # --- تنظیمات سیستم اعتبار ---
 USAGE_LIMIT = 5
-usage_data_cache = {}  # {fingerprint: {'count': 0, 'last_reset': 'YYYY-MM-DD'}}
-PREMIUM_PAGE_ID = '1149636' # شناسه محصول برای بررسی خرید
+DATA_FILE = "usage_data.json" # فایل ذخیره اطلاعات
+PREMIUM_PAGE_ID = '1149636'
+
+# --- توابع مدیریت فایل JSON ---
+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()
 
 # --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
 print("Loading Safety Checker...")
@@ -178,20 +198,26 @@ def get_success_html(message):
 def check_quota(fingerprint, subscription_status):
     """
     بررسی اعتبار کاربر.
-    اگر پولی باشد -> True
-    اگر رایگان باشد:
-       - بررسی تاریخ
-       - بررسی تعداد استفاده
     """
+    global usage_data_cache
+    
+    # لاگ برای دیباگ
+    print(f"CHECK QUOTA: FP={fingerprint}, STATUS={subscription_status}")
+
     if subscription_status == 'paid':
         return True, "unlimited"
     
-    if not fingerprint:
-        # اگر فینگرپرینت نرسید (مثلا جاوا اسکریپت لود نشد)، به عنوان مهمان یک بار اجازه دهیم یا خطا دهیم.
-        # اینجا فرض می‌کنیم باید خطا دهیم یا اجازه محدود دهیم.
-        return True, 0 # برای جلوگیری از خطا در اولین اجرا
+    if not fingerprint or fingerprint == "unknown":
+        # اگر فینگرپرینت نرسید، موقتا اجازه ندهیم یا یک شناسه موقت بسازیم
+        # اینجا فرض می‌کنیم کاربر ناشناس است
+        print("Warning: No fingerprint received.")
+        return True, 0
 
     today_str = date.today().isoformat()
+    
+    # رفرش کردن دیتا از فایل (در صورت تغییر توسط ترد دیگر)
+    usage_data_cache = load_usage_data()
+    
     user_record = usage_data_cache.get(fingerprint)
 
     # اگر کاربر جدید است یا روز عوض شده
@@ -203,8 +229,13 @@ def check_quota(fingerprint, subscription_status):
     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"]
 
 
@@ -224,12 +255,13 @@ def infer(
     subscription_status,
     progress=gr.Progress(track_tqdm=True)
 ):
+    print(f"INFER CALLED: Fingerprint='{fingerprint}', Sub='{subscription_status}'")
+    
     # --- بررسی اعتبار ---
     is_allowed, usage_count = check_quota(fingerprint, subscription_status)
     
     if not is_allowed:
         msg = "شما به محدودیت ۵ تصویر رایگان روزانه خود رسیده‌اید. لطفاً برای استفاده نامحدود حساب خود را ارتقا دهید."
-        # این پیام باعث نمایش مودال ارتقا می‌شود (به خاطر کلمه Quota یا Limited در JS)
         return None, seed, get_error_html(msg)
 
     if input_image is None:
@@ -296,8 +328,6 @@ 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, "example_user", "paid")
     return res, s, status
 
@@ -321,7 +351,7 @@ async (image) => {
 }
 """
 
-# --- جاوااسکریپت جامع (Fingerprint, Subscription, UI, Errors) ---
+# --- جاوااسکریپت جامع ---
 js_global_content = """
 <script>
 document.addEventListener('DOMContentLoaded', () => {
@@ -329,7 +359,6 @@ document.addEventListener('DOMContentLoaded', () => {
     // 1. Fingerprint & Subscription Logic
     // ---------------------------------------------
     
-    // تابع تولید Fingerprint
     async function getBrowserFingerprint() {
         const components = [
             navigator.userAgent,
@@ -340,17 +369,12 @@ document.addEventListener('DOMContentLoaded', () => {
         try {
             const canvas = document.createElement('canvas');
             const ctx = canvas.getContext('2d');
-            ctx.textBaseline = "top";
-            ctx.font = "14px 'Arial'";
-            ctx.textBaseline = "alphabetic";
-            ctx.fillStyle = "#f60";
-            ctx.fillRect(125, 1, 62, 20);
-            ctx.fillStyle = "#069";
-            ctx.fillText("a1b2c3d4e5f6g7h8i9j0_!@#$%^&*()", 2, 15);
+            ctx.textBaseline = "top"; ctx.font = "14px 'Arial'"; ctx.textBaseline = "alphabetic";
+            ctx.fillStyle = "#f60"; ctx.fillRect(125, 1, 62, 20);
+            ctx.fillStyle = "#069"; ctx.fillText("a1b2c3d4e5f6g7h8i9j0_!@#$%^&*()", 2, 15);
             components.push(canvas.toDataURL());
-        } catch (e) {
-            components.push("canvas-error");
-        }
+        } catch (e) { components.push("canvas-error"); }
+        
         const fingerprintString = components.join('~~~');
         let hash = 0;
         for (let i = 0; i < fingerprintString.length; i++) {
@@ -361,36 +385,32 @@ document.addEventListener('DOMContentLoaded', () => {
         return 'fp_' + Math.abs(hash).toString(16);
     }
 
-    // بررسی کاربر پولی بودن
     function isUserPaid(userObject) {
         const PREMIUM_PAGE_ID = '1149636'; 
         if (userObject && userObject.isLogin && userObject.accessible_pages) {
-            // بررسی می‌کنیم آیا ID صفحه پرمیوم در لیست دسترسی‌ها هست یا نه
             if (Array.isArray(userObject.accessible_pages)) {
-                 return userObject.accessible_pages.some(page => 
-                    String(page) === String(PREMIUM_PAGE_ID)
-                 );
+                 return userObject.accessible_pages.some(page => String(page) === String(PREMIUM_PAGE_ID));
             }
         }
         return false;
     }
 
-    // به‌روزرسانی UI badge
-    function updateSubscriptionBadge(status) {
-        const badge = document.getElementById('user-sub-badge');
+    function updateHiddenInputs(fingerprint, status) {
         const fpInput = document.querySelector('#fingerprint_storage textarea');
         const stInput = document.querySelector('#status_storage textarea');
-
-        // مقداردهی اینپوت‌های مخفی برای ارسال به پایتون
+        
         if(fpInput) {
-            fpInput.value = window.userFingerprint || 'unknown';
+            fpInput.value = fingerprint;
             fpInput.dispatchEvent(new Event('input', { bubbles: true }));
         }
         if(stInput) {
             stInput.value = status;
             stInput.dispatchEvent(new Event('input', { bubbles: true }));
         }
+    }
 
+    function updateSubscriptionBadge(status) {
+        const badge = document.getElementById('user-sub-badge');
         if (!badge) return;
         
         if (status === 'paid') {
@@ -401,20 +421,24 @@ document.addEventListener('DOMContentLoaded', () => {
             badge.style.background = 'linear-gradient(45deg, #4b5563, #6b7280)';
         }
         badge.style.display = 'inline-block';
+        
+        // اطمینان از آپدیت شدن اینپوت‌ها هر بار که وضعیت تغییر می‌کند
+        updateHiddenInputs(window.userFingerprint || 'unknown', status);
     }
 
-    // شروع فرآیند شناسایی
     async function initUserIdentity() {
         window.userFingerprint = await getBrowserFingerprint();
+        window.userStatus = 'free'; // Default
         
-        // درخواست وضعیت از Parent
         window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
-        
-        // پیش‌فرض: رایگان
         updateSubscriptionBadge('free');
+        
+        // Polling to ensure inputs are populated even if DOM is slow
+        setInterval(() => {
+            updateHiddenInputs(window.userFingerprint || 'unknown', window.userStatus || 'free');
+        }, 2000);
     }
 
-    // شنیدن پیام از Parent
     window.addEventListener('message', (event) => {
         if (event.data && event.data.type === 'USER_STATUS_RESPONSE') {
             try {
@@ -423,6 +447,7 @@ document.addEventListener('DOMContentLoaded', () => {
                                    : event.data.payload;
                 
                 const status = isUserPaid(userObject) ? 'paid' : 'free';
+                window.userStatus = status;
                 updateSubscriptionBadge(status);
             } catch (e) {
                 console.error("Error parsing user status:", e);
@@ -434,7 +459,7 @@ document.addEventListener('DOMContentLoaded', () => {
     initUserIdentity();
 
     // ---------------------------------------------
-    // 2. UI Helper & Error Handling
+    // 2. UI Helper
     // ---------------------------------------------
     const forceLight = () => {
         const body = document.querySelector('body');
@@ -448,13 +473,6 @@ document.addEventListener('DOMContentLoaded', () => {
     forceLight();
     setInterval(forceLight, 1000);
 
-    window.retryGeneration = function() {
-        const modal = document.getElementById('custom-quota-modal');
-        if (modal) modal.remove();
-        const runBtn = document.getElementById('run-btn');
-        if(runBtn) runBtn.click();
-    };
-    
     window.closeErrorModal = function() {
          const modal = document.getElementById('custom-quota-modal');
          if (modal) modal.remove();
@@ -469,25 +487,22 @@ document.addEventListener('DOMContentLoaded', () => {
                          <div style="font-size: 2rem;">💎</div>
                         <div>
                             <h2>پایان اعتبار رایگان امروز</h2>
-                            <p>برای ادامه، حساب خود را ارتقا دهید یا فردا برگردید.</p>
+                            <p>برای ادامه، حساب خود را ارتقا دهید.</p>
                         </div>
                     </div>
-                    
                     <div class="guide-content">
                         <div class="info-card" style="border-color: #fbbf24; background: #fffbeb;">
-                            <p>شما از ۵ اعتبار رایگان امروز خود استفاده کرده‌اید. با خرید نسخه نامحدود، بدون هیچ محدودیتی تصویر بسازید.</p>
+                            <p>شما از ۵ اعتبار رایگان امروز خود استفاده کرده‌اید.</p>
                         </div>
-                        
                         <div class="video-button-container">
-                            <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_URL', url: '#/nav/online/news/getSingle/1149636/eyJpdiI6InZSVUdlLzBlR0FzOHZJdXFZeWhER0E9PSIsInZhbHVlIjoiWFhqRXBLc29vSFpHdk9nYmRjZGVuWHRHRHVSZHRlTG1BUENLaE5mNXBNVVRGWFg3ZWN0djJ5K1dIY1RqTHJGaCIsIm1hYyI6IjIzYzFlZTMwYmVmMTdkYjQ0YTQ4YWMxNmFhN2RmNWQ2OTc1NDIyNGVlZGI3ZjJjMjhkNmQxNjM4MDFlZTIxNmUiLCJ0YWciOiIifQ==/20934991' }, '*')" class="elegant-video-button" style="background: linear-gradient(135deg, #f59e0b, #d97706) !important; color: white !important;"> 
+                            <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM' }, '*')" class="elegant-video-button" style="background: linear-gradient(135deg, #f59e0b, #d97706) !important; color: white !important;"> 
                                 <span>⭐️ خرید نسخه نامحدود</span> 
                             </button>
                         </div>
                     </div>
-                    
                     <div class="guide-actions">
                         <button class="action-button back-button" onclick="window.closeErrorModal()"> 
-                            <span>بازگشت</span> 
+                            <span>بستن</span> 
                         </button>
                     </div>
                 </div>
@@ -500,7 +515,6 @@ document.addEventListener('DOMContentLoaded', () => {
         const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar, div[class*="error"]');
         potentialErrors.forEach(el => {
             const text = el.innerText || "";
-            // اگر ارور مربوط به Quota بود (که از سمت پایتون میاد)
             if (text.includes('محدودیت') && text.includes('رایگان')) {
                  showQuotaModal();
                  el.style.display = 'none';
@@ -524,19 +538,14 @@ css_code = """
     font-family: 'Vazirmatn', sans-serif !important;
 }
 
-/* Badge Style */
-#user-sub-badge {
-    padding: 6px 16px;
-    border-radius: 20px;
-    font-size: 0.9em;
-    color: white;
-    margin-top: 10px;
-    box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-    display: none; /* JS will show it */
-    transition: all 0.3s ease;
+/* Hide the storage inputs but keep them in DOM */
+#fingerprint_storage, #status_storage {
+    display: none !important; 
+    /* We use display:none with !important in CSS instead of visible=False in Python
+       to ensure the element exists in DOM for JS to find it. */
 }
 
-/* Modal CSS (Reused from previous request mostly) */
+/* Modal CSS */
 .ip-reset-guide-container {
     text-align: right; 
     direction: rtl; 
@@ -608,10 +617,9 @@ combined_html = css_code + js_global_content
 with gr.Blocks() as demo:
     gr.HTML(combined_html)
     
-    # کامپوننت‌های مخفی برای انتقال اطلاعات JS به Python
-    # مقدار این‌ها توسط JS پر می‌شود
-    fingerprint_box = gr.Textbox(elem_id="fingerprint_storage", visible=False)
-    status_box_input = gr.Textbox(elem_id="status_storage", visible=False)
+    # تغییر مهم: visible=True قرار دادیم تا در DOM باشد، اما با CSS مخفی کردیم
+    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")
@@ -706,14 +714,13 @@ with gr.Blocks() as demo:
             label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-    # اتصال دکمه اجرا با ورودی‌های مخفی Fingerprint و Status
     run_button.click(
         fn=infer,
         inputs=[
             input_image, prompt, lora_adapter, seed, randomize_seed, 
             guidance_scale, steps, aspect_ratio_selection, 
             custom_width, custom_height,
-            fingerprint_box, status_box_input # ارسال اطلاعات کاربر به تابع پایتون
+            fingerprint_box, status_box_input 
         ],
         outputs=[output_image, seed, status_box],
         api_name="predict"