Update app.py

app.py

@@ -38,7 +38,7 @@ def is_image_nsfw(image):
     try:
         results = safety_classifier(image)
         for result in results:
-            if result['label'] == 'nsfw' and result['score'] > 0.75: # کمی سخت‌گیرتر
+            if result['label'] == 'nsfw' and result['score'] > 0.75:
                 return True
         return False
     except Exception as e:
@@ -67,30 +67,14 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
 ).to(device)
 
 # بارگذاری LoRA ها
-pipe.load_lora_weights("autoweeb/Qwen-Image-Edit-2509-Photo-to-Anime",
-                       weight_name="Qwen-Image-Edit-2509-Photo-to-Anime_000001000.safetensors",
-                       adapter_name="anime")
-pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Multiple-angles",
-                       weight_name="镜头转换.safetensors",
-                       adapter_name="multiple-angles")
-pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Light_restoration",
-                       weight_name="移除光影.safetensors",
-                       adapter_name="light-restoration")
-pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Relight",
-                       weight_name="Qwen-Edit-Relight.safetensors",
-                       adapter_name="relight")
-pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Multi-Angle-Lighting",
-                       weight_name="多角度灯光-251116.safetensors",
-                       adapter_name="multi-angle-lighting")
-pipe.load_lora_weights("tlennon-ie/qwen-edit-skin",
-                       weight_name="qwen-edit-skin_1.1_000002750.safetensors",
-                       adapter_name="edit-skin")
-pipe.load_lora_weights("lovis93/next-scene-qwen-image-lora-2509",
-                       weight_name="next-scene_lora-v2-3000.safetensors",
-                       adapter_name="next-scene")
-pipe.load_lora_weights("vafipas663/Qwen-Edit-2509-Upscale-LoRA",
-                       weight_name="qwen-edit-enhance_64-v3_000001000.safetensors",
-                       adapter_name="upscale-image")
+pipe.load_lora_weights("autoweeb/Qwen-Image-Edit-2509-Photo-to-Anime", weight_name="Qwen-Image-Edit-2509-Photo-to-Anime_000001000.safetensors", adapter_name="anime")
+pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Multiple-angles", weight_name="镜头转换.safetensors", adapter_name="multiple-angles")
+pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Light_restoration", weight_name="移除光影.safetensors", adapter_name="light-restoration")
+pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Relight", weight_name="Qwen-Edit-Relight.safetensors", adapter_name="relight")
+pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Multi-Angle-Lighting", weight_name="多角度灯光-251116.safetensors", adapter_name="multi-angle-lighting")
+pipe.load_lora_weights("tlennon-ie/qwen-edit-skin", weight_name="qwen-edit-skin_1.1_000002750.safetensors", adapter_name="edit-skin")
+pipe.load_lora_weights("lovis93/next-scene-qwen-image-lora-2509", weight_name="next-scene_lora-v2-3000.safetensors", adapter_name="next-scene")
+pipe.load_lora_weights("vafipas663/Qwen-Edit-2509-Upscale-LoRA", weight_name="qwen-edit-enhance_64-v3_000001000.safetensors", adapter_name="upscale-image")
 
 pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 MAX_SEED = np.iinfo(np.int32).max
@@ -149,9 +133,7 @@ def translate_prompt(text):
 def update_dimensions_on_upload(image):
     if image is None:
         return 1024, 1024
-    
     original_width, original_height = image.size
-    
     if original_width > original_height:
         new_width = 1024
         aspect_ratio = original_height / original_width
@@ -160,15 +142,13 @@ def update_dimensions_on_upload(image):
         new_height = 1024
         aspect_ratio = original_width / original_height
         new_width = int(new_height * aspect_ratio)
-        
     new_width = (new_width // 8) * 8
     new_height = (new_height // 8) * 8
-    
     return new_width, new_height
 
-# تابع برای نمایش/مخفی کردن اسلایدرهای ابعاد
-def toggle_custom_dimensions(value):
-    if value == "شخصی‌سازی (Custom)":
+# تابع به‌روزرسانی وضعیت نمایش اسلایدرها
+def update_sliders_visibility(choice):
+    if choice == "شخصی‌سازی (Custom)":
         return gr.update(visible=True), gr.update(visible=True)
     else:
         return gr.update(visible=False), gr.update(visible=False)
@@ -206,11 +186,9 @@ def infer(
     if input_image is None:
         return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید.")
 
-    # 1. بررسی امنیت تصویر ورودی
     if is_image_nsfw(input_image):
         return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمی‌شود.")
 
-    # 2. ترجمه و بررسی متن
     english_prompt = translate_prompt(prompt)
     if not check_text_safety(english_prompt):
         return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز یا غیراخلاقی است.")
@@ -230,7 +208,6 @@ def infer(
 
     original_image = input_image.convert("RGB")
     
-    # --- منطق ابعاد ---
     selection_value = ASPECT_RATIOS_MAP.get(aspect_ratio_selection)
     
     if selection_value == "Custom":
@@ -253,7 +230,6 @@ def infer(
             true_cfg_scale=guidance_scale,
         ).images[0]
 
-        # 4. بررسی امنیت تصویر خروجی
         if is_image_nsfw(result):
             return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد.")
         
@@ -264,7 +240,6 @@ def infer(
 
 @spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    # برای نمونه‌ها
     res, s, status = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024)
     return res, s, status
 
@@ -391,6 +366,10 @@ textarea:focus, input[type="text"]:focus {
     box-shadow: 0 8px 25px rgba(16, 185, 129, 0.45) !important;
 }
 
+.primary-btn:active, button.primary:active {
+    transform: translateY(1px);
+}
+
 #download-btn {
     background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
     box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3) !important;
@@ -434,6 +413,11 @@ textarea:focus, input[type="text"]:focus {
     padding: 10px !important;
 }
 
+.gradio-container tbody td span, 
+.gradio-container tbody td p {
+    color: #374151 !important;
+}
+
 footer { display: none !important; }
 .flagging { display: none !important; }
 
@@ -485,9 +469,9 @@ with gr.Blocks() as demo:
                     lines=3
                 )
                 
-                # --- اضافه کردن بخش نمایش وضعیت (پیام‌ها) ---
                 status_box = gr.HTML(label="وضعیت")
-
+                
+                # اضافه کردن api_name="predict" برای فعال کردن حالت API
                 run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn")
 
             with gr.Column():
@@ -510,17 +494,15 @@ with gr.Blocks() as demo:
                         interactive=True
                     )
                     
-                    # اسلایدرها به طور پیش‌فرض مخفی هستند
-                    with gr.Row():
+                    # اسلایدرها با visible=False شروع می‌شوند
+                    with gr.Row(visible=False) as custom_dims_row:
                         custom_width = gr.Slider(
                             label="عرض دلخواه (Width)",
-                            minimum=256, maximum=2048, step=8, value=1024,
-                            visible=False
+                            minimum=256, maximum=2048, step=8, value=1024
                         )
                         custom_height = gr.Slider(
                             label="ارتفاع دلخواه (Height)",
-                            minimum=256, maximum=2048, step=8, value=1024,
-                            visible=False
+                            minimum=256, maximum=2048, step=8, value=1024
                         )
                     
                     seed = gr.Slider(label="دانه تصادفی (Seed)", minimum=0, maximum=MAX_SEED, step=1, value=0)
@@ -528,11 +510,16 @@ with gr.Blocks() as demo:
                     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)
         
-        # --- اتصال رویداد برای نمایش/مخفی کردن ابعاد شخصی ---
+        # اصلاح منطق نمایش/مخفی کردن ردیف اسلایدرها
+        def toggle_row(choice):
+            if choice == "شخصی‌سازی (Custom)":
+                return gr.update(visible=True)
+            return gr.update(visible=False)
+
         aspect_ratio_selection.change(
-            fn=toggle_custom_dimensions,
+            fn=toggle_row,
             inputs=aspect_ratio_selection,
-            outputs=[custom_width, custom_height]
+            outputs=custom_dims_row # کنترل کل ردیف به جای تک تک اسلایدرها
         )
 
         gr.Examples(
@@ -550,17 +537,18 @@ with gr.Blocks() as demo:
                 ["examples/6.jpg", "دوربین را به نمای پایین به بالا تغییر بده.", "تغییر زاویه دید"],
             ],
             inputs=[input_image, prompt, lora_adapter],
-            outputs=[output_image, seed, status_box], # خروجی وضعیت اضافه شد
+            outputs=[output_image, seed, status_box],
             fn=infer_example,
             cache_examples=False,
             label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-    # اتصال دکمه اجرا (خروجی‌ها: تصویر، سید، پیام وضعیت)
+    # اتصال دکمه اجرا با قابلیت API
     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, status_box]
+        outputs=[output_image, seed, status_box],
+        api_name="predict" # فعال‌سازی API
     )
 
     download_button.click(