Update app.py

app.py

@@ -2423,42 +2423,79 @@ def ui_generate_video(
     )
 
 
-# کدهای جاوااسکریپت برای انتقال المان‌ها و نظارت بر ارور محدودیت GPU
+# کدهای جاوااسکریپت بسیار پیشرفته‌تر برای تشخیص خطای خارج از بدنه گاردریو (ZeroGPU Modals)
 relocate_js = r"""
 (() => { 
+    function getGradioRoot() {
+        const app = document.querySelector('gradio-app');
+        return app && app.shadowRoot ? app.shadowRoot : document;
+    }
+
     function moveIntoFooter() { 
-        const p = document.querySelector("#prompt_ui"), f = p ? p.querySelector(".ds-footer") : null; 
+        const root = getGradioRoot();
+        const p = root.querySelector("#prompt_ui");
+        const 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"); 
+        
+        const d = root.querySelector("#duration_ui .cd-wrap");
+        const r = root.querySelector("#resolution_ui .cd-wrap");
+        const c = root.querySelector("#camera_ui .cd-wrap"); 
         if (!d || !r || !c) return false; 
-        f.appendChild(d); f.appendChild(r); f.appendChild(c); 
+        
+        f.appendChild(d); 
+        f.appendChild(r); 
+        f.appendChild(c); 
         return true; 
     } 
     
+    function forceResetButton() {
+        const root = getGradioRoot();
+        
+        // 1. شبیه‌سازی کلیک روی دکمه مخفی برای ریست بک‌اند پایتون
+        const resetWrap = root.querySelector("#reset_btn");
+        if (resetWrap) {
+            const btn = resetWrap.querySelector("button") || resetWrap;
+            if (btn) btn.click();
+        }
+        
+        // 2. تغییر ظاهری فوری دکمه اصلی جهت رفع سریع قفل
+        const mainWrap = root.querySelector("#generate_btn");
+        if (mainWrap) {
+            const mainBtn = mainWrap.querySelector("button") || mainWrap;
+            if (mainBtn) {
+                mainBtn.innerText = "🤩 ساخت ویدیو";
+                mainBtn.disabled = false;
+                mainBtn.style.pointerEvents = "auto";
+            }
+        }
+    }
+
     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});
+        // یک حلقه بی‌نهایت و امن که هر 800 میلی‌ثانیه کل صفحه را اسکن می‌کند
+        setInterval(() => {
+            let fullText = "";
+            
+            // دریافت محتوای متنی از بدنه اصلی (Hugging Face مودال‌های خطا را اینجا تزریق می‌کند)
+            fullText += document.body.innerText || "";
+            
+            // دریافت محتوای متنی از ShadowDOM (خطاهای داخلی گاردریو)
+            const root = getGradioRoot();
+            if (root !== document) {
+                fullText += root.textContent || "";
+            }
+            
+            fullText = fullText.toLowerCase();
+            
+            // اگر خطای محدودیت کوپون دیده شد
+            if (fullText.includes("exceeded your gpu quota") || fullText.includes("gpu quota")) {
+                if (!window._quotaResetFired) {
+                    forceResetButton();
+                    window._quotaResetFired = true;
+                    // یک توقف کوتاه تا از اسپم شدن کلیک‌ها در طول نمایش خطا جلوگیری شود
+                    setTimeout(() => { window._quotaResetFired = false; }, 6000);
+                }
+            }
+        }, 800);
     }
     
     const tick = () => { 
@@ -2521,7 +2558,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                     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", elem_id="generate_btn")
-                # دکمه پنهان که در صورت وقوع ارور توسط جاوااسکریپت کلیک می‌شود تا رابط کاربری بازیابی شود
+                # دکمه پنهانی که هنگام وقوع خطا کلیک می‌شود تا UI را بازسازی کند
                 reset_btn = gr.Button("Reset", visible=False, elem_id="reset_btn")
 
     # اتصال رویدادها
@@ -2533,7 +2570,7 @@ 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)
     
-    # رویداد بازیابی دکمه در زمان وقوع محدودیت (وصل شده به دکمه مخفی)
+    # رویداد بازیابی دکمه در زمان وقوع محدودیت
     reset_btn.click(
         fn=lambda: gr.Button(value="🤩 ساخت ویدیو", interactive=True),
         inputs=[],