Update app.py

app.py

@@ -34,10 +34,11 @@ print("Loading Safety Checker...")
 safety_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device=-1)
 
 def is_image_nsfw(image):
+    if image is None: return False
     try:
         results = safety_classifier(image)
         for result in results:
-            if result['label'] == 'nsfw' and result['score'] > 0.8:
+            if result['label'] == 'nsfw' and result['score'] > 0.75: # کمی سخت‌گیرتر
                 return True
         return False
     except Exception as e:
@@ -105,7 +106,7 @@ LORA_MAPPING = {
     "افزایش کیفیت (Upscale)": "upscale-image"
 }
 
-ASPECT_RATIOS = [
+ASPECT_RATIOS_LIST = [
     "خودکار (پیش‌فرض)",
     "۱:۱ (مربع - 1024x1024)",
     "۱۶:۹ (افقی - 1344x768)",
@@ -165,12 +166,29 @@ def update_dimensions_on_upload(image):
     
     return new_width, new_height
 
+# تابع برای نمایش/مخفی کردن اسلایدرهای ابعاد
 def toggle_custom_dimensions(value):
     if value == "شخصی‌سازی (Custom)":
         return gr.update(visible=True), gr.update(visible=True)
     else:
         return gr.update(visible=False), gr.update(visible=False)
 
+def get_error_html(message):
+    return f"""
+    <div style="background-color: #fee2e2; border: 1px solid #ef4444; color: #b91c1c; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;">
+        <span style="font-size: 1.2em;">⛔</span>
+        {message}
+    </div>
+    """
+
+def get_success_html(message):
+    return f"""
+    <div style="background-color: #dcfce7; border: 1px solid #22c55e; color: #15803d; padding: 12px; border-radius: 8px; text-align: center; margin-bottom: 10px; font-weight: bold; display: flex; align-items: center; justify-content: center; gap: 8px;">
+        <span style="font-size: 1.2em;">✅</span>
+        {message}
+    </div>
+    """
+
 @spaces.GPU(duration=30)
 def infer(
     input_image,
@@ -186,14 +204,16 @@ def infer(
     progress=gr.Progress(track_tqdm=True)
 ):
     if input_image is None:
-        raise gr.Error("لطفاً یک تصویر برای ویرایش بارگذاری کنید.")
+        return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
 
+    # 1. بررسی امنیت تصویر ورودی
     if is_image_nsfw(input_image):
-        raise gr.Error("تصویر ورودی دارای محتوای نامناسب تشخیص داده شد.")
+        return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
 
+    # 2. ترجمه و بررسی متن
     english_prompt = translate_prompt(prompt)
     if not check_text_safety(english_prompt):
-        raise gr.Error("متن درخواست شامل کلمات غیرمجاز است.")
+        return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است.")
 
     adapter_internal_name = LORA_MAPPING.get(lora_adapter_persian)
     if adapter_internal_name:
@@ -210,6 +230,7 @@ def infer(
 
     original_image = input_image.convert("RGB")
     
+    # --- منطق ابعاد ---
     selection_value = ASPECT_RATIOS_MAP.get(aspect_ratio_selection)
     
     if selection_value == "Custom":
@@ -220,84 +241,333 @@ def infer(
     else:
         width, height = selection_value
 
-    result = pipe(
-        image=original_image,
-        prompt=english_prompt,
-        negative_prompt=final_negative_prompt,
-        height=height,
-        width=width,
-        num_inference_steps=steps,
-        generator=generator,
-        true_cfg_scale=guidance_scale,
-    ).images[0]
-
-    if is_image_nsfw(result):
-        raise gr.Error("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+    try:
+        result = pipe(
+            image=original_image,
+            prompt=english_prompt,
+            negative_prompt=final_negative_prompt,
+            height=height,
+            width=width,
+            num_inference_steps=steps,
+            generator=generator,
+            true_cfg_scale=guidance_scale,
+        ).images[0]
+
+        # 4. بررسی امنیت تصویر خروجی
+        if is_image_nsfw(result):
+            return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
+        
+        return result, seed, get_success_html("تصویر با موفقیت ویرایش شد.")
 
-    return result, seed
+    except Exception as e:
+        return None, seed, get_error_html(f"خطا در پردازش: {str(e)}")
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    input_pil = input_image.convert("RGB")
-    guidance_scale = 1.0
-    steps = 4
-    result, seed = infer(input_pil, prompt, lora_adapter, 0, True, guidance_scale, steps, "خودکار (پیش‌فرض)", 1024, 1024)
-    return result, seed
+    # برای نمونه‌ها
+    res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024)
+    return res, s, status
+
+# --- جاوااسکریپت ---
+js_post_message = """
+async (image) => {
+    if (!image) {
+        alert("لطفاً ابتدا تصویر را تولید کنید.");
+        return;
+    }
+    let fileUrl = image.url;
+    if (fileUrl && !fileUrl.startsWith('http')) {
+       fileUrl = window.location.origin + fileUrl;
+    } 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
+    }, '*');
+}
+"""
 
-# --- تنظیمات HTML ---
 html_code = """
 <style>
-/* CSS Styles (Same as before for Gradio UI) */
-/* ... (مخفی برای کوتاه شدن، همان استایل‌های قبلی اینجا قرار می‌گیرد) ... */
+@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
+
+:root, .dark, body, .gradio-container {
+    --body-background-fill: #f5f7fa !important;
+    --body-text-color: #1f2937 !important;
+    --background-fill-primary: #ffffff !important;
+    --background-fill-secondary: #f3f4f6 !important;
+    --border-color-primary: #e5e7eb !important;
+    --block-background-fill: #ffffff !important;
+    --block-label-text-color: #374151 !important;
+    --block-title-text-color: #111827 !important;
+    --input-background-fill: #ffffff !important;
+    color-scheme: light !important;
+}
+
+body {
+    font-family: 'Vazirmatn', sans-serif !important;
+    background-color: #f5f7fa !important;
+    margin: 0;
+    padding: 10px;
+}
+
+#col-container {
+    margin: 0 auto;
+    max-width: 980px;
+    direction: rtl;
+    text-align: right;
+    padding: 30px;
+    background: #ffffff !important;
+    border-radius: 24px;
+    box-shadow: 0 10px 40px -10px rgba(0,0,0,0.08); 
+    border: 1px solid rgba(255,255,255,0.8);
+}
+
+#main-title h1 {
+    font-size: 2.4em !important;
+    text-align: center;
+    color: #1a202c !important;
+    margin-bottom: 15px;
+    font-weight: 800;
+    background: -webkit-linear-gradient(45deg, #2563eb, #1e40af);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+}
+
+#main-description {
+    text-align: center;
+    font-size: 1.15em;
+    color: #4b5563 !important;
+    margin-bottom: 40px;
+    line-height: 1.6;
+}
+
+.gr-input-label, span.label-wrap, label span {
+    font-weight: 700 !important;
+    color: #374151 !important;
+    font-size: 0.95em !important;
+    margin-bottom: 8px !important;
+}
+
+textarea, input[type="text"] {
+    border: 2px solid #e2e8f0 !important;
+    border-radius: 12px !important;
+    background-color: #ffffff !important;
+    color: #111827 !important;
+    padding: 12px !important;
+    font-family: 'Vazirmatn', sans-serif !important;
+}
+
+textarea:focus, input[type="text"]:focus {
+    border-color: #3b82f6 !important;
+    box-shadow: 0 0 0 4px rgba(59, 130, 246, 0.1) !important;
+    outline: none;
+}
+
+.gr-dropdown {
+    background: #ffffff !important;
+    border-radius: 12px !important;
+}
+
+.primary-btn, button.primary {
+    background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important;
+    border: none !important;
+    color: white !important;
+    font-weight: 700 !important;
+    font-size: 1.1em !important;
+    padding: 14px 28px !important;
+    border-radius: 14px !important;
+    box-shadow: 0 4px 15px rgba(16, 185, 129, 0.3) !important;
+    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
+    cursor: pointer !important;
+    width: 100%;
+    margin-top: 15px;
+}
+
+.primary-btn:hover, button.primary:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 8px 25px rgba(16, 185, 129, 0.45) !important;
+}
+
+#download-btn {
+    background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
+    box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3) !important;
+}
+#download-btn:hover {
+    box-shadow: 0 8px 25px rgba(59, 130, 246, 0.45) !important;
+}
+
+.gradio-container .prose table, 
+.gradio-container table {
+    background-color: #ffffff !important;
+    color: #111827 !important;
+    border: 1px solid #e5e7eb !important;
+    border-radius: 12px !important;
+    overflow: hidden !important;
+    width: 100% !important;
+    margin-top: 20px !important;
+}
+
+.gradio-container thead th {
+    background-color: #f3f4f6 !important;
+    color: #374151 !important;
+    font-weight: 700 !important;
+    border-bottom: 2px solid #e5e7eb !important;
+    padding: 12px !important;
+    text-align: right !important;
+}
+
+.gradio-container tbody tr {
+    background-color: #ffffff !important;
+    border-bottom: 1px solid #f3f4f6 !important;
+}
+
+.gradio-container tbody tr:hover {
+    background-color: #f9fafb !important;
+}
+
+.gradio-container tbody td {
+    background-color: #ffffff !important;
+    color: #374151 !important;
+    padding: 10px !important;
+}
+
 footer { display: none !important; }
 .flagging { display: none !important; }
+
+@media (prefers-color-scheme: dark) {
+    body, .gradio-container, .prose, table, tr, td, th {
+        background-color: #ffffff !important;
+        color: #333333 !important;
+    }
+}
 </style>
+
+<script>
+function forceLightMode() {
+    const body = document.querySelector('body');
+    if (body) {
+        body.classList.remove('dark');
+        body.style.backgroundColor = '#f5f7fa';
+        body.style.color = '#333333';
+    }
+    document.querySelectorAll('.dark').forEach(el => {
+        el.classList.remove('dark');
+    });
+}
+forceLightMode();
+setInterval(forceLightMode, 1000);
+document.addEventListener('DOMContentLoaded', forceLightMode);
+</script>
 """
 
 with gr.Blocks() as demo:
     gr.HTML(html_code)
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("# **ویرایشگر هوشمند آلفا**")
-        
+        gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
+        gr.Markdown(
+            "با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", 
+            elem_id="main-description"
+        )
+
         with gr.Row(equal_height=True):
             with gr.Column():
                 input_image = gr.Image(label="بارگذاری تصویر", type="pil", height=320)
-                prompt = gr.Text(label="دستور ویرایش", placeholder="متن فارسی...", rtl=True, lines=3)
-                run_button = gr.Button("✨ شروع پردازش", variant="primary")
+                
+                prompt = gr.Text(
+                    label="دستور ویرایش (به فارسی)",
+                    show_label=True,
+                    placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...",
+                    rtl=True,
+                    lines=3
+                )
+                
+                # --- اضافه کردن بخش نمایش وضعیت (پیام‌ها) ---
+                status_box = gr.HTML(label="وضعیت")
+
+                run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn")
 
             with gr.Column():
-                output_image = gr.Image(label="خروجی", interactive=False, format="png", height=380)
+                output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
+                
+                download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
                 
                 with gr.Row():
-                    lora_adapter = gr.Dropdown(label="سبک", choices=list(LORA_MAPPING.keys()), value="تبدیل عکس به انیمه")
+                    lora_adapter = gr.Dropdown(
+                        label="انتخاب سبک ویرایش (LoRA)",
+                        choices=list(LORA_MAPPING.keys()),
+                        value="تبدیل عکس به انیمه"
+                    )
                 
-                with gr.Accordion("تنظیمات", open=False):
-                    aspect_ratio_selection = gr.Dropdown(label="ابعاد", choices=ASPECT_RATIOS, value="خودکار (پیش‌فرض)")
+                with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
+                    aspect_ratio_selection = gr.Dropdown(
+                        label="ابعاد تصویر خروجی",
+                        choices=ASPECT_RATIOS_LIST,
+                        value="خودکار (پیش‌فرض)",
+                        interactive=True
+                    )
+                    
+                    # اسلایدرها به طور پیش‌فرض مخفی هستند
                     with gr.Row():
-                        custom_width = gr.Slider(label="عرض", minimum=256, maximum=2048, step=8, value=1024, visible=False)
-                        custom_height = gr.Slider(label="ارتفاع", minimum=256, maximum=2048, step=8, value=1024, visible=False)
-                    seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
-                    randomize_seed = gr.Checkbox(label="Seed تصادفی", value=True)
-                    guidance_scale = gr.Slider(label="Guidance", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
-                    steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=4)
-
-        aspect_ratio_selection.change(toggle_custom_dimensions, aspect_ratio_selection, [custom_width, custom_height])
+                        custom_width = gr.Slider(
+                            label="عرض دلخواه (Width)",
+                            minimum=256, maximum=2048, step=8, value=1024,
+                            visible=False
+                        )
+                        custom_height = gr.Slider(
+                            label="ارتفاع دلخواه (Height)",
+                            minimum=256, maximum=2048, step=8, value=1024,
+                            visible=False
+                        )
+                    
+                    seed = gr.Slider(label="دانه تصادفی (Seed)", minimum=0, maximum=MAX_SEED, step=1, value=0)
+                    randomize_seed = gr.Checkbox(label="استفاده از Seed تصادفی", value=True)
+                    guidance_scale = gr.Slider(label="میزان وفاداری به متن (Guidance Scale)", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
+                    steps = gr.Slider(label="تعداد مراحل پردازش (Steps)", minimum=1, maximum=50, step=1, value=4)
+        
+        # --- اتصال رویداد برای نمایش/مخفی کردن ابعاد شخصی ---
+        aspect_ratio_selection.change(
+            fn=toggle_custom_dimensions,
+            inputs=aspect_ratio_selection,
+            outputs=[custom_width, custom_height]
+        )
 
         gr.Examples(
             examples=[
                 ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
-                # ... سایر نمونه‌ها
+                ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم به تصویر بده.", "اصلاح نور و سایه"],
+                ["examples/4.jpg", "از فیلتر ساعت طلایی با پخش نور ملایم استفاده کن.", "نورپردازی مجدد (Relight)"],
+                ["examples/2.jpeg", "دوربین را ۴۵ درجه به سمت چپ بچرخان.", "تغییر زاویه دید"],
+                ["examples/7.jpg", "منبع نور را از سمت راست عقب قرار بده.", "نورپردازی چند زاویه‌ای"],
+                ["examples/10.jpeg", "کیفیت تصویر را افزایش بده (Upscale).", "افزایش کیفیت (Upscale)"],
+                ["examples/7.jpg", "منبع نور را از پایین بتابان.", "نورپردازی چند زاویه‌ای"],
+                ["examples/2.jpeg", "زاویه دوربین را به نمای بالا گوشه راست تغییر بده.", "تغییر زاویه دید"],
+                ["examples/9.jpg", "دوربین کمی به جلو حرکت می‌کند در حالی که نور خورشید از میان ابرها می‌تابد و درخششی نرم اطراف شبح شخصیت در مه ایجاد می‌کند. سبک سینمایی واقعی.", "صحنه بعدی (سینمایی)"],
+                ["examples/8.jpg", "جزئیات پوست سوژه را برجسته‌تر و طبیعی‌تر کن.", "روتوش پوست"],
+                ["examples/6.jpg", "دوربین را به نمای پایین به بالا تغییر بده.", "تغییر زاویه دید"],
             ],
-            inputs=[input_image, prompt, lora_adapter]
+            inputs=[input_image, prompt, lora_adapter],
+            outputs=[output_image, seed, status_box], # خروجی وضعیت اضافه شد
+            fn=infer_example,
+            cache_examples=False,
+            label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-    # *** نکته مهم: اضافه کردن api_name ***
+    # اتصال دکمه اجرا (خروجی‌ها: تصویر، سید، پیام وضعیت)
     run_button.click(
         fn=infer,
         inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height],
-        outputs=[output_image, seed],
-        api_name="predict" # این خط کلید اتصال به فایل HTML است
+        outputs=[output_image, seed, status_box]
+    )
+
+    download_button.click(
+        fn=None,
+        inputs=[output_image],
+        outputs=None,
+        js=js_post_message
     )
 
 if __name__ == "__main__":