Update app.py

app.py

@@ -2423,7 +2423,7 @@ def ui_generate_video(
     )
 
 
-# کدهای جابجایی المان‌های UI به مکان‌های صحیح
+# کدهای جاوااسکریپت قدرتمند برای بازکردن قفل دکمه هنگام خطا
 relocate_js = r"""
 (() => { 
     function getGradioRoot() {
@@ -2442,20 +2442,91 @@ relocate_js = r"""
         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 checkAndReset(mutations) {
+        let errorFound = false;
+        
+        // بررسی افزوده شدن Toast ها
+        if (mutations) {
+            mutations.forEach(m => {
+                m.addedNodes.forEach(node => {
+                    if (node.nodeType === 1) {
+                        const txt = (node.textContent || "").toLowerCase();
+                        if (txt.includes("quota") || txt.includes("exceeded") || txt.includes("error") || txt.includes("خطا")) {
+                            errorFound = true;
+                        }
+                        if (node.classList && node.classList.contains("toast-wrap")) {
+                            errorFound = true; 
+                        }
+                    }
+                });
+            });
+        }
+
+        // بررسی متون در کل صفحه (حتی مواردی که توسط کد شما مخفی شده‌اند)
+        const root = getGradioRoot();
+        const fullText = ((document.body.textContent || "") + " " + (root.textContent || "")).toLowerCase();
+        if (fullText.includes("exceeded your gpu quota") || fullText.includes("gpu quota")) {
+            errorFound = true;
+        }
+
+        if (errorFound && !window._resetFiredBtn) {
+            window._resetFiredBtn = true;
+            console.log("Error or Quota detected! Forcing reset...");
+            
+            // کلیک کردن دکمه مخفی پایتون که صف رو دور می‌زند
+            const rb = root.querySelector('#reset_btn');
+            if (rb) {
+                const btn = rb.tagName === 'BUTTON' ? rb : rb.querySelector('button');
+                if (btn) btn.click();
+            }
+            
+            // باز کردن فیزیکی قفل دکمه تولید (احتیاط ثانویه)
+            const gbWrap = root.querySelector('#generate_btn');
+            if (gbWrap) {
+                const gb = gbWrap.tagName === 'BUTTON' ? gbWrap : gbWrap.querySelector('button');
+                if (gb) {
+                    gb.innerText = "🤩 ساخت ویدیو";
+                    gb.disabled = false;
+                    gb.classList.remove("disabled"); 
+                    gb.style.pointerEvents = "auto";
+                }
+            }
+            
+            // جلوگیری از کلیک مکرر به مدت ۵ ثانیه
+            setTimeout(() => { window._resetFiredBtn = false; }, 5000);
+        }
+    }
+
+    function setupErrorObserver() {
+        const observer = new MutationObserver(checkAndReset);
+        observer.observe(document.body, { childList: true, subtree: true });
+        
+        const root = getGradioRoot();
+        if (root !== document) {
+            observer.observe(root, { childList: true, subtree: true });
+        }
+        
+        // اسکن دائمی صفحه هر 1 ثانیه به عنوان روش تضمینی
+        setInterval(() => checkAndReset(), 1000);
+    }
+    
     const tick = () => { 
         if (!moveIntoFooter()) requestAnimationFrame(tick); 
+        else setupErrorObserver(); 
     }; 
     requestAnimationFrame(tick); 
 })();
 """
 
 with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
+    
+    # مخفی کردن فیزیکی دکمه ریست بدون استفاده از visible=False
+    gr.HTML("<style>#reset_btn { display: none !important; }</style>")
+    
     gr.HTML(
         """
         <div style="text-align: center; padding: 20px;">
@@ -2507,6 +2578,9 @@ 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")
+                
+                # توجه مهم: این دکمه دیگر visible=False نیست تا از DOM حذف نشود. توسط CSS بالا مخفی شده است.
+                reset_btn = gr.Button("Reset", elem_id="reset_btn")
 
     # اتصال رویدادها
     camera_ui.change(fn=lambda x: x, inputs=camera_ui, outputs=camera_lora, api_visibility="private")
@@ -2517,18 +2591,22 @@ 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)
     
-    # تابع بازگرداندن دکمه به حالت اولیه (در صورت وقوع خطای محدودیت)
-    def reset_button_state():
-        return gr.Button(value="🤩 ساخت ویدیو", interactive=True)
-        
-    # --- زنجیره اصلی ساخت ویدیو ---
-    step1 = generate_btn.click(
+    # *** رویداد جادویی برای شکستن قفل گاردریو ***
+    # پارامتر queue=False باعث می‌شود اگر گاردریو در صف گیر کرده باشد، این دستور معطل نشود و فوراً اجرا شود
+    reset_btn.click(
+        fn=lambda: gr.Button(value="🤩 ساخت ویدیو", interactive=True),
+        inputs=[],
+        outputs=[generate_btn],
+        api_visibility="private",
+        queue=False 
+    )
+    
+    # زنجیره اصلی ساخت ویدیو
+    generate_btn.click(
         fn=ui_lock_and_enhance,
         inputs=[first_frame, prompt, radioanimated_mode],
         outputs=[generate_btn, prompt_ui, prompt, output_video]
-    )
-    
-    step2 = step1.then(
+    ).then(
         fn=ui_generate_video,
         inputs=[
             first_frame, end_frame, prompt, 
@@ -2538,11 +2616,6 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
         outputs=[generate_btn, output_video, download_btn]
     )
 
-    # 🔴 کلید حل مشکل: مدیریت استثناها (Exceptions)
-    # با افزودن .error هر وقت سرور به خاطر محدودیت GPU درخواست را رد کند، دکمه سریعاً ریست می‌شود.
-    step1.error(fn=reset_button_state, inputs=[], outputs=[generate_btn], api_visibility="private")
-    step2.error(fn=reset_button_state, inputs=[], outputs=[generate_btn], api_visibility="private")
-
     # منطق نمونه ها
     examples_list = [
         ["examples/supergirl-2.png", "A fuzzy puppet superhero...", "Static", "16:9", "تبدیل تصویر به ویدیو", None, "examples/supergirl.m4a", None],