Update app.py

app.py

@@ -31,13 +31,10 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # --- بارگذاری مدل تشخیص محتوای نامناسب (NSFW) ---
 print("Loading Safety Checker...")
-try:
-    safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
-except:
-    safety_classifier = None # Fallback if model fails to load
+safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
 def is_image_nsfw(image):
-    if image is None or safety_classifier is None: return False
+    if image is None: return False
     try:
         results = safety_classifier(image)
         for result in results:
@@ -268,7 +265,7 @@ async (image) => {
 }
 """
 
-# --- جاوااسکریپت سراسری (استراتژی حذف کامل برای رفع مشکل رفرش) ---
+# --- جاوااسکریپت سراسری (استراتژی پنجره شناور مستقل) ---
 js_global_content = """
 <script>
 document.addEventListener('DOMContentLoaded', () => {
@@ -285,45 +282,31 @@ document.addEventListener('DOMContentLoaded', () => {
     forceLight();
     setInterval(forceLight, 1000);
 
-    // 2. HELPER: Completely destroy the error element
-    window.destroyErrorToast = function(btn) {
-        // Try to find the container
-        const toast = btn.closest('.toast-wrap') || btn.closest('.toast-body') || btn.closest('.error');
-        
-        // Find the main run button and click it to retry
+    // 2. RETRY FUNCTION
+    window.retryGeneration = function() {
+        // Close modal
+        const modal = document.getElementById('custom-quota-modal');
+        if (modal) modal.remove();
+
+        // Click Run button
         const runBtn = document.getElementById('run-btn');
         if(runBtn) runBtn.click();
-
-        // Remove the error UI immediately
-        if (toast) {
-            toast.remove();
-        }
-        
-        // Aggressive cleanup: remove ANY other error toasts visible
-        document.querySelectorAll('.toast-body, .toast-wrap, .error').forEach(el => {
-            if(el.innerText.includes("قدم تا ساخت") || el.innerText.includes("quota")) {
-                el.remove();
-            }
-        });
     };
     
-    // Just remove toast (for Back button)
-    window.closeErrorToast = function(btn) {
-         const toast = btn.closest('.toast-wrap') || btn.closest('.toast-body') || btn.closest('.error');
-         if(toast) toast.remove();
+    // Close function
+    window.closeErrorModal = function() {
+         const modal = document.getElementById('custom-quota-modal');
+         if (modal) modal.remove();
     };
 
-    // 3. LOGIC: Replace content logic
-    const replaceErrorContent = (node) => {
-        if (node.querySelector('.ip-reset-guide-container')) return;
+    // 3. SHOW MODAL FUNCTION
+    const showQuotaModal = () => {
+        // Prevent duplicate modals
+        if (document.getElementById('custom-quota-modal')) return;
 
-        const text = node.innerText || "";
-        
-        // Check triggers
-        if (text.toLowerCase().includes('quota') || text.toLowerCase().includes('exceeded') || (text.includes('Error') && node.classList.contains('error'))) {
-            
-            const prettyHtml = `
-                <div class="ip-reset-guide-container">
+        const modalHtml = `
+            <div id="custom-quota-modal" style="position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.6); backdrop-filter: blur(5px); z-index: 99999; display: flex; align-items: center; justify-content: center; font-family: 'Vazirmatn', sans-serif;">
+                <div class="ip-reset-guide-container" style="max-width: 450px; width: 90%; background: white; border-radius: 20px; padding: 25px; box-shadow: 0 10px 40px rgba(0,0,0,0.3); animation: slideInUp 0.4s ease;">
                     <div class="guide-header">
                         <svg class="guide-header-icon" viewbox="0 0 100 100" xmlns="http://www.w3.org/2000/svg"> 
                             <defs><lineargradient id="grad1" x1="0%" y1="0%" x2="100%" y2="100%"><stop offset="0%" style="stop-color: #667eea; stop-opacity: 1;"></stop><stop offset="100%" style="stop-color: #764ba2; stop-opacity: 1;"></stop></lineargradient></defs> 
@@ -365,41 +348,47 @@ document.addEventListener('DOMContentLoaded', () => {
                     </div>
                     
                     <div class="guide-actions">
-                        <button class="action-button back-button" onclick="window.closeErrorToast(this)"> 
+                        <button class="action-button back-button" onclick="window.closeErrorModal()"> 
                             <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="window.destroyErrorToast(this)"> 
+                        <button class="action-button retry-button" onclick="window.retryGeneration()"> 
                             <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> 
                         </button>
                     </div>
                 </div>
-            `;
-
-            // Inject the HTML
-            node.innerHTML = prettyHtml;
-            
-            node.style.padding = '0';
-            node.style.border = 'none';
-            node.style.backgroundColor = 'transparent';
-            node.style.boxShadow = 'none';
-            node.classList.remove('error');
-        }
+            </div>
+        `;
+        
+        document.body.insertAdjacentHTML('beforeend', modalHtml);
     };
 
-    // 4. AGGRESSIVE SCANNER
+    // 4. SCANNER
     setInterval(() => {
-        const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar');
+        // Find ALL possible error containers, including small toasts and status bars
+        const potentialErrors = document.querySelectorAll('.toast-body, .error, .toast-wrap, .eta-bar, div[class*="error"]');
         
         potentialErrors.forEach(el => {
-            if (el.innerText && (el.innerText.includes('quota') || el.innerText.includes('exceeded') || (el.innerText.includes('Error') && el.classList.contains('error')))) {
-                 if (!el.querySelector('.ip-reset-guide-container')) {
-                     replaceErrorContent(el);
-                 }
+            const text = el.innerText || "";
+            // Trigger words
+            if (text.toLowerCase().includes('quota') || text.toLowerCase().includes('exceeded')) {
+                 
+                 // Show our custom modal if not already open
+                 showQuotaModal();
+                 
+                 // IMMEDIATELY HIDE/REMOVE the ugly Gradio error
+                 // We use display:none or remove() to ensure it's gone
+                 el.style.display = 'none';
+                 el.style.opacity = '0';
+                 el.innerText = ''; // Clear text so it doesn't trigger again immediately for this specific node
+                 
+                 // Also try to find parent wrapper and hide it
+                 const parentWrap = el.closest('.toast-wrap');
+                 if(parentWrap) parentWrap.style.display = 'none';
             }
         });
-    }, 150); 
+    }, 100); // Check every 100ms
 });
 </script>
 """
@@ -447,26 +436,6 @@ css_code = """
     to { opacity: 1; transform: translateY(0); }
 }
 
-.ip-reset-guide-container {
-    text-align: right; 
-    background: var(--guide-bg); 
-    backdrop-filter: blur(10px);
-    padding: 25px; 
-    border-radius: var(--radius-lg-guide);
-    box-shadow: var(--shadow-xl); 
-    border: 1px solid var(--guide-border); 
-    animation: slideInUp 0.6s cubic-bezier(0.4, 0, 0.2, 1) both;
-    width: 100%; 
-    max-width: 450px; 
-    position: relative; 
-    overflow: hidden; 
-    box-sizing: border-box;
-    font-family: 'Vazirmatn', sans-serif !important;
-    direction: rtl;
-}
-.ip-reset-guide-container::before {
-    content: ''; position: absolute; top: 0; left: 0; right: 0; height: 5px; background: var(--primary-gradient-guide);
-}
 .guide-header { display: flex; align-items: center; margin-bottom: 20px; }
 .guide-header-icon { width: 50px; height: 50px; margin-left: 15px; animation: float 3s ease-in-out infinite; flex-shrink: 0; }
 .guide-header h2 { font-size: 1.3rem; color: var(--guide-text-title); font-weight: 700; margin: 0; }