Update app.py

app.py

@@ -2347,7 +2347,6 @@ def enhance_prompt_on_cpu(image_path: str, prompt: str, mode: str):
     """
     این تابع روی CPU اجرا می‌شود و هیچ سهمیه‌ای از GPU کم نمی‌کند.
     """
-    # اگر نیازی به بهینه سازی نیست
     if mode not in ['تبدیل تصویر به ویدیو', 'تکمیل فریم‌های میانی'] or not image_path:
         return prompt
 
@@ -2357,7 +2356,6 @@ def enhance_prompt_on_cpu(image_path: str, prompt: str, mode: str):
         with open(image_path, 'rb') as f:
             files = {'image': f}
             data = {'prompt': prompt, 'is_extension': 'false'}
-            # تایم اوت را بالا می بریم تا مطمئن شویم پاسخ می گیریم
             response = requests.post(url, files=files, data=data, timeout=60)
             
             if response.status_code == 200:
@@ -2388,23 +2386,19 @@ def ui_lock_and_enhance(first_frame, prompt, mode):
     """
     مرحله ۱: قفل کردن UI و اجرای بهینه سازی روی CPU
     """
-    # تغییر دکمه به حالت لودینگ
     yield (
         gr.Button(value="⏳ در حال بهینه سازی پرامپت (CPU)...", interactive=False),
-        gr.update(), # prompt_ui (no change yet)
-        gr.update(), # آپدیت تکست‌باکس مخفی
-        None # output clear
+        gr.update(), 
+        gr.update(), 
+        None 
     )
     
-    # اجرای بهینه سازی (سنگین برای شبکه، سبک برای پردازنده)
-    # چون دکوراتور @spaces.GPU ندارد، سهمیه شما مصرف نمی شود.
     enhanced = enhance_prompt_on_cpu(first_frame, prompt, mode)
     
-    # آماده سازی برای مرحله بعد
     yield (
         gr.Button(value="🚀 در حال ساخت ویدیو (GPU)...", interactive=False),
-        gr.update(value=enhanced), # نمایش پرامپت جدید به کاربر در UI
-        enhanced, # پاس دادن پرامپت بهینه شده به تکست‌باکس مخفی برای استفاده در تولید
+        gr.update(value=enhanced), 
+        enhanced, 
         None
     )
 
@@ -2416,7 +2410,6 @@ def ui_generate_video(
 ):
     """
     مرحله ۲: ساخت ویدیو با استفاده از GPU
-    در اینجا تمام زمان تخصیص داده شده صرف ساخت ویدیو می شود.
     """
     video_path = generate_video(
         first_frame, end_frame, enhanced_prompt, duration, input_video,
@@ -2430,6 +2423,52 @@ def ui_generate_video(
     )
 
 
+# کدهای جاوااسکریپت برای انتقال المان‌ها و نظارت بر ارور محدودیت GPU
+relocate_js = r"""
+(() => { 
+    function moveIntoFooter() { 
+        const p = document.querySelector("#prompt_ui"), f = p ? p.querySelector(".ds-footer") : null; 
+        if (!f) return false; 
+        const d = document.querySelector("#duration_ui .cd-wrap"), r = document.querySelector("#resolution_ui .cd-wrap"), c = document.querySelector("#camera_ui .cd-wrap"); 
+        if (!d || !r || !c) return false; 
+        f.appendChild(d); f.appendChild(r); f.appendChild(c); 
+        return true; 
+    } 
+    
+    function setupErrorObserver() {
+        const getRoot = () => {
+            const app = document.querySelector('gradio-app');
+            return app && app.shadowRoot ? app.shadowRoot : document;
+        };
+        const observer = new MutationObserver((mutations) => {
+            mutations.forEach(m => {
+                m.addedNodes.forEach(node => {
+                    if (node.nodeType === 1) {
+                        // بررسی ظهور پاپ‌آپ‌های خطا یا محدودیت
+                        const isToast = node.classList && (node.classList.contains('toast-wrap') || node.classList.contains('error') || node.tagName === 'TOAST');
+                        const nodeText = (node.innerText || "").toLowerCase();
+                        if (isToast || nodeText.includes('quota')) {
+                            if (nodeText.includes("quota") || nodeText.includes("exceeded") || nodeText.includes("error") || nodeText.includes("خطا")) {
+                                // در صورت مشاهده خطا دکمه ریست نامرئی کلیک می‌شود
+                                const rb = getRoot().querySelector("#reset_btn");
+                                if (rb) rb.click();
+                            }
+                        }
+                    }
+                });
+            });
+        });
+        observer.observe(document.body, {childList: true, subtree: true});
+    }
+    
+    const tick = () => { 
+        if (!moveIntoFooter()) requestAnimationFrame(tick); 
+        else setupErrorObserver(); 
+    }; 
+    requestAnimationFrame(tick); 
+})();
+"""
+
 with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
     gr.HTML(
         """
@@ -2453,13 +2492,14 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                     first_frame = gr.Image(label="تصویر اول (برای بهینه‌سازی پرامپت الزامی است)", type="filepath", height=256)
                     end_frame = gr.Image(label="تصویر آخر (اختیاری)", type="filepath", height=256, visible=False)
                     input_video = gr.Video(label="Motion Reference Video", height=256, visible=False)
-                relocate = gr.HTML(value="", html_template="<div></div>", js_on_load=r"""(() => { function moveIntoFooter() { const p = document.querySelector("#prompt_ui"), f = p ? p.querySelector(".ds-footer") : null; if (!f) return false; const d = document.querySelector("#duration_ui .cd-wrap"), r = document.querySelector("#resolution_ui .cd-wrap"), c = document.querySelector("#camera_ui .cd-wrap"); if (!d || !r || !c) return false; f.appendChild(d); f.appendChild(r); f.appendChild(c); return true; } const tick = () => { if (!moveIntoFooter()) requestAnimationFrame(tick); }; requestAnimationFrame(tick); })();""")
+                
+                relocate = gr.HTML(value="", html_template="<div></div>", js_on_load=relocate_js)
+                
                 prompt_ui = PromptBox(value="این تصویر را با حرکت سینمایی و انیمیشن روان زنده کن", elem_id="prompt_ui")
                 audio_input = gr.File(label="Audio (Optional)", file_types=["audio"], type="filepath", elem_id="audio_input_hidden")
                 audio_ui = AudioDropUpload(target_audio_elem_id="audio_input_hidden", elem_id="audio_ui")
                 prompt = gr.Textbox(label="Prompt", value="این تصویر را با حرکت سینمایی و انیمیشن روان زنده کن", lines=3, max_lines=3, visible=False)
                 
-                # مقدار اولیه به False تغییر کرد تا تابع داخلی مدل دوباره تلاش نکند پرامپت را تغییر دهد
                 enhance_prompt_flag_input = gr.Checkbox(label="Enhance Prompt", value=False, visible=False) 
                 
                 with gr.Accordion("تنظیمات پیشرفته", open=False, visible=False):
@@ -2480,7 +2520,9 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                     camera_ui = CameraDropdown(choices=[name for name, _ in VISIBLE_RUNTIME_LORA_CHOICES], value="No LoRA", title="افکت دوربین (LoRA)", elem_id="camera_ui")
                     camera_lora = gr.Dropdown(label="Camera Control LoRA", choices=[name for name, _ in VISIBLE_RUNTIME_LORA_CHOICES], value="No LoRA", visible=False)
                 
-                generate_btn = gr.Button("🤩 ساخت ویدیو", variant="primary", elem_classes="button-gradient")
+                generate_btn = gr.Button("🤩 ساخت ویدیو", variant="primary", elem_classes="button-gradient", elem_id="generate_btn")
+                # دکمه پنهان که در صورت وقوع ارور توسط جاوااسکریپت کلیک می‌شود تا رابط کاربری بازیابی شود
+                reset_btn = gr.Button("Reset", visible=False, elem_id="reset_btn")
 
     # اتصال رویدادها
     camera_ui.change(fn=lambda x: x, inputs=camera_ui, outputs=camera_lora, api_visibility="private")
@@ -2491,9 +2533,16 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
     audio_ui.change(fn=lambda v: None if v == "__CLEAR__" or not v else gr.update(), inputs=audio_ui, outputs=audio_input, api_visibility="private")
     download_btn.click(fn=None, inputs=[output_video], js=js_download_video)
     
-    # --- زنجیره اصلی ساخت ویدیو (Main Chain) ---
+    # رویداد بازیابی دکمه در زمان وقوع محدودیت (وصل شده به دکمه مخفی)
+    reset_btn.click(
+        fn=lambda: gr.Button(value="🤩 ساخت ویدیو", interactive=True),
+        inputs=[],
+        outputs=[generate_btn],
+        api_visibility="private"
+    )
     
-    gen_event = generate_btn.click(
+    # زنجیره اصلی ساخت ویدیو
+    generate_btn.click(
         fn=ui_lock_and_enhance,
         inputs=[first_frame, prompt, radioanimated_mode],
         outputs=[generate_btn, prompt_ui, prompt, output_video]
@@ -2507,15 +2556,6 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
         outputs=[generate_btn, output_video, download_btn]
     )
 
-    # --- هندل کردن خطای محدودیت GPU ---
-    # در صورتی که ساخت ویدیو (مثلا به خاطر اتمام سهمیه GPU) با خطا مواجه شود، 
-    # این بخش وارد عمل شده و فوراً دکمه را به حالت اولیه برمی‌گرداند.
-    gen_event.catch(
-        fn=lambda: gr.Button(value="🤩 ساخت ویدیو", interactive=True),
-        inputs=[],
-        outputs=[generate_btn]
-    )
-
     # منطق نمونه ها
     examples_list = [
         ["examples/supergirl-2.png", "A fuzzy puppet superhero...", "Static", "16:9", "تبدیل تصویر به ویدیو", None, "examples/supergirl.m4a", None],