Update app.py

app.py

@@ -235,7 +235,11 @@ def infer(
         return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
 
     except Exception as e:
-        return None, seed, get_error_html(f"خطا در پردازش: {str(e)}")
+        # اگر خطا مربوط به GPU بود، متن خاص را برمی‌گردانیم تا JS آن را شکار کند
+        error_str = str(e)
+        if "quota" in error_str.lower():
+            raise gr.Error("You have exceeded your GPU quota")
+        return None, seed, get_error_html(f"خطا در پردازش: {error_str}")
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
@@ -255,7 +259,6 @@ 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
@@ -263,8 +266,7 @@ async (image) => {
 }
 """
 
-# --- جاوااسکریپت اصلی (شامل مدیریت هوشمند خطا) ---
-# این اسکریپت با دقت هر تغییری در صفحه را رصد می‌کند و اگر خطای GPU را ببیند، آن را جایگزین می‌کند.
+# --- جاوااسکریپت اصلی (شامل مدیریت هوشمند خطای GPU و حالت هواپیما) ---
 js_global_func = """
 function() {
     // 1. اجبار تم روشن
@@ -280,50 +282,55 @@ function() {
     forceLight();
     setInterval(forceLight, 1000);
 
-    // 2. مدیریت خطای GPU (نسخه پیشرفته)
+    // 2. مدیریت خطای GPU (شناسایی و جایگزینی با پیام حالت هواپیما)
     const observer = new MutationObserver((mutations) => {
         mutations.forEach((mutation) => {
             if (mutation.addedNodes.length) {
                 mutation.addedNodes.forEach((node) => {
-                    // بررسی تمام نودهای اضافه شده (چه المنت چه متن)
-                    if (node.nodeType === 1) { // اگر المنت باشد
+                    if (node.nodeType === 1) { 
+                        // بررسی متن داخل نود برای پیدا کردن خطا
                         const textContent = node.innerText || "";
                         if (textContent.includes("exceeded your GPU quota") || textContent.includes("GPU quota")) {
                             
-                            // پیدا کردن کانتینر اصلی ارور (معمولاً کلاس toast-body یا error دارد)
-                            // اگر خودش کانتینر نبود، والدش را پیدا می‌کنیم یا خودش را تغییر می‌دهیم
+                            // پیدا کردن کانتینر اصلی ارور (معمولاً کلاس toast-body دارد)
                             let targetNode = node;
                             if (!targetNode.classList.contains('toast-body')) {
                                 const parentToast = targetNode.closest('.toast-body');
                                 if (parentToast) targetNode = parentToast;
                             }
 
+                            // جایگزینی محتوا
                             if (!targetNode.dataset.fixed) {
                                 targetNode.innerHTML = `
-                                    <div style="font-family: 'Vazirmatn', sans-serif; direction: rtl; text-align: right; padding: 10px; background: white; border-radius: 8px;">
-                                        <div style="display: flex; align-items: center; gap: 10px; margin-bottom: 8px;">
-                                            <span style="font-size: 30px;">✈️</span>
-                                            <h3 style="margin: 0; color: #c0392b; font-size: 18px; font-weight: bold;">ظرفیت تکمیل شد</h3>
+                                    <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>
+                                                <h3 style="margin: 0; color: #c0392b; font-size: 18px; font-weight: bold;">پایان سهمیه پردازش</h3>
+                                            </div>
                                         </div>
-                                        <p style="margin: 5px 0; color: #333; font-size: 14px;">
-                                            سهمیه رایگان استفاده از سرور تمام شده است.
+                                        <p style="margin: 0 0 12px 0; color: #555; font-size: 14px;">
+                                            سهمیه رایگان شما برای استفاده از سرور تمام شده است.
                                         </p>
-                                        <div style="background-color: #e0f2f1; border: 1px solid #80cbc4; border-radius: 8px; padding: 10px; margin-top: 10px;">
-                                            <p style="margin: 0; color: #00695c; font-weight: bold; font-size: 13px;">💡 راه حل:</p>
-                                            <p style="margin: 5px 0 0 0; font-size: 13px; color: #004d40;">
-                                                لطفاً <b>حالت هواپیما (Airplane Mode)</b> گوشی خود را روشن و سپس خاموش کنید و صفحه را رفرش کنید.
+                                        <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 شما تغییر کند، سپس دوباره تلاش کنید.
                                             </p>
                                         </div>
                                     </div>
                                 `;
                                 targetNode.dataset.fixed = "true";
-                                // مخفی کردن استایل‌های پیش‌فرض زشت
+                                // حذف استایل‌های پیش‌فرض زشت Gradio
+                                targetNode.style.backgroundColor = "transparent";
                                 targetNode.style.border = "none";
-                                targetNode.style.boxShadow = "0 4px 15px rgba(0,0,0,0.1)";
-                                // اگر والدی دارد که بوردر قرمز دارد، آن را هم پاکسازی کن
-                                if(targetNode.parentElement) {
+                                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.parentElement.style.backgroundColor = "transparent";
                                 }
                             }
                         }
@@ -495,7 +502,7 @@ textarea:focus, input[type="text"]:focus {{
 footer {{ display: none !important; }}
 .flagging {{ display: none !important; }}
 
-/* استایل Toast */
+/* استایل Toast برای راست چین کردن */
 .toast-body {{
     direction: rtl !important;
     text-align: right !important;