Update app.py

app.py

@@ -235,9 +235,9 @@ def infer(
         return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
 
     except Exception as e:
-        # اگر خطا مربوط به GPU بود، متن خاص را برمی‌گردانیم تا JS آن را شکار کند
+        # مدیریت خطای GPU: این متن دقیقا باید همین باشد تا JS آن را تشخیص دهد
         error_str = str(e)
-        if "quota" in error_str.lower():
+        if "quota" in error_str.lower() or "exceeded" in error_str.lower():
             raise gr.Error("You have exceeded your GPU quota")
         return None, seed, get_error_html(f"خطا در پردازش: {error_str}")
 
@@ -259,6 +259,7 @@ async (image) => {
     } else if (!fileUrl && image.path) {
        fileUrl = window.location.origin + "/file=" + image.path;
     }
+    console.log("Sending download request for:", fileUrl);
     window.parent.postMessage({
         type: 'DOWNLOAD_REQUEST',
         url: fileUrl
@@ -266,7 +267,7 @@ async (image) => {
 }
 """
 
-# --- جاوااسکریپت اصلی (شامل مدیریت هوشمند خطای GPU و حالت هواپیما) ---
+# --- جاوااسکریپت هوشمند برای جایگزینی پیام خطا ---
 js_global_func = """
 function() {
     // 1. اجبار تم روشن
@@ -282,7 +283,7 @@ function() {
     forceLight();
     setInterval(forceLight, 1000);
 
-    // 2. مدیریت خطای GPU (شناسایی و جایگزینی با پیام حالت هواپیما)
+    // 2. مدیریت خطای GPU (شناسایی و جایگزینی کامل)
     const observer = new MutationObserver((mutations) => {
         mutations.forEach((mutation) => {
             if (mutation.addedNodes.length) {
@@ -290,48 +291,75 @@ function() {
                     if (node.nodeType === 1) { 
                         // بررسی متن داخل نود برای پیدا کردن خطا
                         const textContent = node.innerText || "";
-                        if (textContent.includes("exceeded your GPU quota") || textContent.includes("GPU quota")) {
-                            
+                        
+                        if (textContent.includes("exceeded your GPU quota")) {
                             // پیدا کردن کانتینر اصلی ارور (معمولاً کلاس toast-body دارد)
-                            let targetNode = node;
-                            if (!targetNode.classList.contains('toast-body')) {
-                                const parentToast = targetNode.closest('.toast-body');
-                                if (parentToast) targetNode = parentToast;
+                            // ما تا بالاترین سطح ممکن (toast-wrap) بالا می‌رویم تا کل کادر زشت را حذف کنیم
+                            
+                            let targetNode = node.classList.contains('toast-wrap') ? node : node.closest('.toast-wrap');
+                            
+                            // اگر toast-wrap پیدا نشد، toast-body را پیدا کن
+                            if (!targetNode) {
+                                targetNode = node.classList.contains('toast-body') ? node : node.closest('.toast-body');
                             }
 
-                            // جایگزینی محتوا
-                            if (!targetNode.dataset.fixed) {
+                            if (targetNode && !targetNode.dataset.fixed) {
+                                // جایگزینی کل محتوای HTML با پیام فارسی زیبا
                                 targetNode.innerHTML = `
-                                    <div style="font-family: 'Vazirmatn', sans-serif; direction: rtl; text-align: right; padding: 15px; background: white; border-radius: 12px; border: 2px solid #e74c3c;">
-                                        <div style="display: flex; align-items: center; gap: 12px; margin-bottom: 10px;">
-                                            <span style="font-size: 32px;">✈️</span>
+                                    <div style="
+                                        font-family: 'Vazirmatn', sans-serif; 
+                                        direction: rtl; 
+                                        text-align: right; 
+                                        padding: 20px; 
+                                        background: white; 
+                                        border-radius: 15px; 
+                                        border: 2px solid #e74c3c;
+                                        box-shadow: 0 10px 30px rgba(231, 76, 60, 0.2);
+                                        min-width: 300px;
+                                    ">
+                                        <div style="display: flex; align-items: center; gap: 15px; margin-bottom: 15px; border-bottom: 1px solid #eee; padding-bottom: 10px;">
+                                            <span style="font-size: 35px;">✈️</span>
                                             <div>
-                                                <h3 style="margin: 0; color: #c0392b; font-size: 18px; font-weight: bold;">پایان سهمیه پردازش</h3>
+                                                <h3 style="margin: 0; color: #c0392b; font-size: 18px; font-weight: 800;">ظرفیت سرور تکمیل شد!</h3>
+                                                <span style="font-size: 12px; color: #7f8c8d;">خطای محدودیت GPU</span>
                                             </div>
                                         </div>
-                                        <p style="margin: 0 0 12px 0; color: #555; font-size: 14px;">
-                                            سهمیه رایگان شما برای استفاده از سرور تمام شده است.
+                                        
+                                        <p style="margin: 0 0 15px 0; color: #34495e; font-size: 14px; line-height: 1.6;">
+                                            متأسفانه سهمیه رایگان پردازش شما تمام شده است. برای ادامه باید IP خود را تغییر دهید.
                                         </p>
-                                        <div style="background-color: #e3f2fd; border: 1px solid #64b5f6; border-radius: 8px; padding: 10px;">
-                                            <p style="margin: 0; color: #1565c0; font-weight: bold; font-size: 13px;">💡 راه حل پیشنهادی:</p>
-                                            <p style="margin: 5px 0 0 0; font-size: 13px; color: #0d47a1;">
-                                                لطفاً <b>حالت هواپیما (Airplane Mode)</b> گوشی خود را روشن و سپس خاموش کنید تا IP شما تغییر کند، سپس دوباره تلاش کنید.
+                                        
+                                        <div style="background-color: #e8f6f3; border: 1px dashed #1abc9c; border-radius: 10px; padding: 12px;">
+                                            <p style="margin: 0; color: #16a085; font-weight: bold; font-size: 13px; display: flex; align-items: center; gap: 5px;">
+                                                <span>💡</span> راه حل سریع:
+                                            </p>
+                                            <p style="margin: 8px 0 0 0; font-size: 13px; color: #0e6655; line-height: 1.5;">
+                                                لطفاً <b>حالت هواپیما (Airplane Mode)</b> را روشن/خاموش کنید و صفحه را رفرش نمایید.
                                             </p>
                                         </div>
+                                        
+                                        <button onclick="this.closest('.toast-wrap').remove()" style="
+                                            margin-top: 15px;
+                                            width: 100%;
+                                            padding: 8px;
+                                            background: #ecf0f1;
+                                            border: none;
+                                            border-radius: 8px;
+                                            color: #7f8c8d;
+                                            cursor: pointer;
+                                            font-family: inherit;
+                                            font-size: 12px;
+                                        ">بستن پیام</button>
                                     </div>
                                 `;
                                 targetNode.dataset.fixed = "true";
-                                // حذف استایل‌های پیش‌فرض زشت Gradio
+                                
+                                // حذف استایل‌های مزاحم پیش‌فرض Gradio از این نود
                                 targetNode.style.backgroundColor = "transparent";
                                 targetNode.style.border = "none";
                                 targetNode.style.boxShadow = "none";
                                 targetNode.style.padding = "0";
-                                
-                                // اگر والدی دارد که استایل مزاحم دارد، آن را هم درست کن
-                                if(targetNode.parentElement && targetNode.parentElement.classList.contains('toast-wrap')) {
-                                    targetNode.parentElement.style.boxShadow = "none";
-                                    targetNode.parentElement.style.border = "none";
-                                }
+                                targetNode.className = "toast-wrap"; // حذف کلاس‌های error و ...
                             }
                         }
                     }