Update app.py

app.py

@@ -265,7 +265,7 @@ async (image) => {
 }
 """
 
-# --- جاوااسکریپت سراسری (بهینه شده برای رفع مشکل عدم نمایش بار دوم) ---
+# --- جاوااسکریپت سراسری (Fixed to ensure retry works every time) ---
 js_global_content = """
 <script>
 document.addEventListener('DOMContentLoaded', () => {
@@ -282,17 +282,30 @@ document.addEventListener('DOMContentLoaded', () => {
     forceLight();
     setInterval(forceLight, 1000);
 
-    // 2. NEW GPU Quota Replacer (ROBUST VERSION)
-    // This function handles the replacement logic
+    // Function to completely remove the error toast
+    // This forces Gradio to create a brand new one next time
+    window.removeErrorToast = function(btn) {
+        const toast = btn.closest('.toast-wrap') || btn.closest('.toast-body');
+        if (toast) {
+            toast.remove();
+        }
+        // Also try to find any other lingering error messages and remove them
+        document.querySelectorAll('.toast-body, .toast-wrap').forEach(el => el.remove());
+    };
+
+    // 2. NEW GPU Quota Replacer logic
     const replaceError = (node) => {
-        if (!node || node.dataset.translated) return;
+        if (!node) return;
         
+        // Prevent re-processing if our custom container is already inside
+        if (node.querySelector('.ip-reset-guide-container')) return;
+
         const originalText = node.innerText || "";
         // Regex to catch Quota Exceeded error
         const regex = /(\d+)s requested vs. (\d+)s left/;
         const match = originalText.match(regex);
         
-        // Also check if it's a generic error that might have been caused by quota but lost details
+        // Check triggers
         if (match || originalText.includes("GPU quota") || originalText.includes("exceeded")) {
             
             const prettyHtml = `
@@ -338,15 +351,14 @@ document.addEventListener('DOMContentLoaded', () => {
                     </div>
                     
                     <div class="guide-actions">
-                        <button class="action-button back-button" onclick="this.closest('.toast-wrap').remove()"> 
+                        <button class="action-button back-button" onclick="window.removeErrorToast(this)"> 
                             <svg class="action-button-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M19 12H5M12 19l-7-7 7-7" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
                             <span>بازگشت</span> 
                         </button>
                         <button class="action-button retry-button" onclick="
                             const runBtn = document.getElementById('run-btn');
                             if(runBtn) runBtn.click();
-                            // Optional: Close current toast, though clicking run usually clears it
-                            try { this.closest('.toast-wrap').remove(); } catch(e){}
+                            window.removeErrorToast(this);
                         "> 
                             <svg class="action-button-icon" viewbox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"><path d="M23 4v6h-6M1 20v-6h6" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path><path d="M20.49 9A9 9 0 0 0 5.64 5.64L1 10m22 4l-4.64 4.36A9 9 0 0 1 3.51 15" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"></path></svg> 
                             <span>تلاش مجدد</span> 
@@ -356,24 +368,28 @@ document.addEventListener('DOMContentLoaded', () => {
             `;
             
             node.innerHTML = prettyHtml;
-            node.dataset.translated = 'true';
+            // Force removal of default styling to avoid conflicts
             node.style.padding = '0';
             node.style.backgroundColor = 'transparent';
+            node.style.boxShadow = 'none';
+            node.style.border = 'none';
         }
     };
 
-    // Method A: Mutation Observer (Fast reaction)
+    // Mutation Observer to watch for NEW errors
     const observer = new MutationObserver((mutations) => {
         mutations.forEach((mutation) => {
             if (mutation.addedNodes.length) {
                 mutation.addedNodes.forEach((node) => {
-                    if (node.nodeType === 1 && (node.classList.contains('toast-body') || node.classList.contains('error'))) {
-                        replaceError(node);
-                    }
-                    // Also check children just in case
-                    if (node.nodeType === 1 && node.querySelector) {
-                         const childError = node.querySelector('.toast-body, .error');
-                         if (childError) replaceError(childError);
+                    // Gradio usually adds a .toast-body inside a .toast-wrap
+                    if (node.nodeType === 1) {
+                         if (node.classList.contains('toast-body') || node.classList.contains('error')) {
+                            replaceError(node);
+                         } else if (node.querySelector) {
+                            // Check children
+                            const childError = node.querySelector('.toast-body, .error');
+                            if (childError) replaceError(childError);
+                         }
                     }
                 });
             }
@@ -381,16 +397,18 @@ document.addEventListener('DOMContentLoaded', () => {
     });
     observer.observe(document.body, { childList: true, subtree: true });
 
-    // Method B: Interval Scanner (Backup for missed events/re-renders)
-    // This fixes the issue where the second error doesn't show
+    // CONTINUOUS SCANNER: This is critical for the "second attempt" issue.
+    // Sometimes Gradio updates the text node without adding a new element,
+    // or recycles the container. This interval ensures we catch it.
     setInterval(() => {
         const errorNodes = document.querySelectorAll('.toast-body, .error');
         errorNodes.forEach(node => {
-            if (!node.dataset.translated) {
+            // Only try to replace if it doesn't already have our custom UI
+            if (!node.querySelector('.ip-reset-guide-container')) {
                 replaceError(node);
             }
         });
-    }, 500);
+    }, 200); // Check every 200ms
 });
 </script>
 """