Update app.py

app.py

@@ -21,7 +21,7 @@ from datetime import date
 
 # رنگ‌ها و تنظیمات ظاهری
 USAGE_LIMIT = 5
-DATA_FILE = "usage_data.json" 
+DATA_FILE = "usage_data.json"
 PREMIUM_PAGE_ID = '1149636'
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
@@ -104,13 +104,13 @@ def is_image_nsfw(image):
     try:
         # اگر ورودی لیست گالری باشد، اولین تصویر را چک کن
         img_to_check = image
-        if isinstance(image, list): 
+        if isinstance(image, list):
              # هندل کردن فرمت گالری گرادیو
              if len(image) > 0:
                 img_to_check = image[0][0] if isinstance(image[0], tuple) else image[0]
              else:
                 return False
-        
+
         results = safety_classifier(img_to_check)
         for result in results:
             if result['label'] == 'nsfw' and result['score'] > 0.75:
@@ -126,9 +126,9 @@ def check_text_safety(text):
     padded_text = f" {text_lower} "
     for char in [".", ",", "!", "?", "-", "_", "(", ")", "[", "]", "{", "}"]:
         padded_text = padded_text.replace(char, " ")
-    
+
     for word in BANNED_WORDS:
-        if f" {word} " in padded_text: 
+        if f" {word} " in padded_text:
             return False
     return True
 
@@ -206,7 +206,7 @@ def upsample_prompt_logic(prompt, image_list):
         else:
             system_content = SYSTEM_PROMPT_TEXT_ONLY
             messages = [{"role": "system", "content": system_content}, {"role": "user", "content": prompt}]
-        
+
         completion = hf_client.chat.completions.create(model=VLM_MODEL, messages=messages, max_tokens=1024)
         return completion.choices[0].message.content
     except Exception as e:
@@ -226,7 +226,7 @@ def get_duration(prompt_embeds, image_list, width, height, num_inference_steps,
 def generate_image(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
     prompt_embeds = prompt_embeds.to(device)
     generator = torch.Generator(device=device).manual_seed(seed)
-    
+
     pipe_kwargs = {
         "prompt_embeds": prompt_embeds,
         "image": image_list,
@@ -236,13 +236,13 @@ def generate_image(prompt_embeds, image_list, width, height, num_inference_steps
         "width": width,
         "height": height,
     }
-    
+
     if progress: progress(0, desc="Starting generation...")
     image = pipe(**pipe_kwargs).images[0]
     return image
 
 def infer(
-    prompt, input_images, seed, randomize_seed, width, height, 
+    prompt, input_images, seed, randomize_seed, width, height,
     num_inference_steps, guidance_scale, prompt_upsampling,
     fingerprint, subscription_status,
     progress=gr.Progress(track_tqdm=True)
@@ -274,36 +274,36 @@ def infer(
     # 4. آماده‌سازی تنظیمات
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-    
+
     try:
         # Upsampling Prompt (Optional)
         final_prompt = english_prompt
         if prompt_upsampling:
             progress(0.2, desc="Enhancing prompt...")
             final_prompt = upsample_prompt_logic(english_prompt, image_list)
-        
+
         # Text Encoding (CPU/Network)
         progress(0.3, desc="Encoding...")
         prompt_embeds = remote_text_encoder(final_prompt)
-        
+
         # Generation (GPU)
         progress(0.4, desc="Generating...")
         result_image = generate_image(
-            prompt_embeds, image_list, width, height, 
+            prompt_embeds, image_list, width, height,
             num_inference_steps, guidance_scale, seed, progress
         )
-        
+
         # 5. بررسی تصویر خروجی
         if is_image_nsfw(result_image):
             return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود."), gr.update(visible=True), gr.update(visible=False)
-        
+
         # 6. محاسبه اعتبار باقی‌مانده
         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" (اعتبار باقی‌مانده امروز: {remaining})"
-        
+
         btn_run_update = gr.update(visible=True)
         btn_upg_update = gr.update(visible=False)
 
@@ -370,9 +370,9 @@ document.addEventListener('DOMContentLoaded', () => {
         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'; 
+        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));
         }
@@ -401,7 +401,7 @@ document.addEventListener('DOMContentLoaded', () => {
 
     async function initUserIdentity() {
         window.userFingerprint = await getBrowserFingerprint();
-        window.userStatus = 'free'; 
+        window.userStatus = 'free';
         window.parent.postMessage({ type: 'REQUEST_USER_STATUS' }, '*');
         updateSubscriptionBadge('free');
         updateHiddenInputs(window.userFingerprint, window.userStatus);
@@ -425,7 +425,7 @@ document.addEventListener('DOMContentLoaded', () => {
     // 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 = `
@@ -437,8 +437,8 @@ document.addEventListener('DOMContentLoaded', () => {
                     <div class="guide-content">
                         <p>برای ادامه ساخت تصویر، لطفاً طبق آموزش زیر IP خود را تغییر دهید (اینترنت را خاموش/روشن کنید یا VPN را قطع کنید) و سپس دکمه تلاش مجدد را بزنید.</p>
                         <div class="video-button-container">
-                            <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_URL', url: '#/nav/online/news/getSingle/1149635' }, '*')" class="elegant-video-button"> 
-                                <span>دیدن ویدیو آموزشی</span> 
+                            <button onclick="parent.postMessage({ type: 'NAVIGATE_TO_URL', url: '#/nav/online/news/getSingle/1149635' }, '*')" class="elegant-video-button">
+                                <span>دیدن ویدیو آموزشی</span>
                             </button>
                         </div>
                     </div>
@@ -505,17 +505,18 @@ footer { display: none !important; }
 # 6. ساخت رابط کاربری (Gradio Blocks)
 # ==========================================
 
-with gr.Blocks(css=css_code) as demo:
+# ******** این خط اصلاح شده است ********
+with gr.Blocks() as demo:
     gr.HTML(js_global_content + css_code)
-    
+
     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("# **ساخت تصویر با FLUX.2 (پیشرفته)**", elem_id="main-title")
         gr.Markdown("با استفاده از مدل قدرتمند FLUX.2 متن فارسی خود را به تصاویر شگفت‌انگیز تبدیل کنید.", elem_id="main-description")
         gr.HTML('<div id="badge-container"><span id="user-sub-badge"></span></div>')
-        
+
         with gr.Row():
             with gr.Column():
                 with gr.Row():
@@ -526,7 +527,7 @@ with gr.Blocks(css=css_code) as demo:
                         placeholder="یک منظره زیبا از...",
                         rtl=True
                     )
-                
+
                 with gr.Accordion("بارگذاری تصویر (اختیاری برای ویرایش/ایده)", open=False):
                     input_images = gr.Gallery(
                         label="تصاویر ورودی",
@@ -535,12 +536,12 @@ with gr.Blocks(css=css_code) as demo:
                         rows=1,
                         height=200
                     )
-                
+
                 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.Accordion("تنظیمات پیشرفته", open=False):
                     prompt_upsampling = gr.Checkbox(label="بهبود خودکار پرامپت (هوشمند)", value=True)
                     seed = gr.Slider(label="دانه تصادفی (Seed)", minimum=0, maximum=MAX_SEED, step=1, value=0)
@@ -557,7 +558,7 @@ with gr.Blocks(css=css_code) as demo:
                 download_button = gr.Button("📥 دانلود تصویر", variant="secondary", elem_id="download-btn")
 
     # اتصال رویدادها
-    
+
     # 1. آپدیت ابعاد بر اساس تصویر آپلودی
     input_images.upload(
         fn=update_dimensions_from_image,
@@ -576,7 +577,7 @@ with gr.Blocks(css=css_code) as demo:
     run_button.click(
         fn=infer,
         inputs=[
-            prompt, input_images, seed, randomize_seed, width, height, 
+            prompt, input_images, seed, randomize_seed, width, height,
             num_inference_steps, guidance_scale, prompt_upsampling,
             fingerprint_box, status_box_input
         ],