Update app.py

app.py

@@ -236,6 +236,22 @@ LORA_MAPPING_PERSIAN = {
     "انتقال متریال (Material-Transfer)": "Material-Transfer"
 }
 
+ASPECT_RATIOS_LIST = [
+    "خودکار (پیش‌فرض)",
+    "۱:۱ (مربع - 1024x1024)",
+    "۱۶:۹ (افقی - 1344x768)",
+    "۹:۱۶ (عمودی - 768x1344)",
+    "شخصی‌سازی (Custom)"
+]
+
+ASPECT_RATIOS_MAP = {
+    "خودکار (پیش‌فرض)": "Auto",
+    "۱:۱ (مربع - 1024x1024)": (1024, 1024),
+    "۱۶:۹ (افقی - 1344x768)": (1344, 768),
+    "۹:۱۶ (عمودی - 768x1344)": (768, 1344),
+    "شخصی‌سازی (Custom)": "Custom"
+}
+
 def update_dimensions_on_upload(image):
     if image is None: return 1024, 1024
     original_width, original_height = image.size
@@ -251,6 +267,12 @@ def update_dimensions_on_upload(image):
     new_height = (new_height // 16) * 16
     return new_width, new_height
 
+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)
+
 # ==========================================
 # 5. HTML Helpers
 # ==========================================
@@ -290,6 +312,9 @@ def infer(
     randomize_seed,
     guidance_scale,
     steps,
+    aspect_ratio_selection,
+    custom_width,
+    custom_height,
     fingerprint,
     subscription_status,
     progress=gr.Progress(track_tqdm=True)
@@ -352,7 +377,16 @@ def infer(
     img1_pil = image_1.convert("RGB")
     img2_pil = image_2.convert("RGB")
     
-    width, height = update_dimensions_on_upload(img1_pil)
+    # 7. تنظیم ابعاد تصویر
+    selection_value = ASPECT_RATIOS_MAP.get(aspect_ratio_selection)
+    
+    if selection_value == "Custom":
+        width = (int(custom_width) // 16) * 16
+        height = (int(custom_height) // 16) * 16
+    elif selection_value == "Auto" or selection_value is None:
+        width, height = update_dimensions_on_upload(img1_pil)
+    else:
+        width, height = selection_value
 
     try:
         result = pipe(
@@ -366,11 +400,11 @@ def infer(
             true_cfg_scale=guidance_scale,
         ).images[0]
 
-        # 7. بررسی ایمنی خروجی
+        # 8. بررسی ایمنی خروجی
         if is_image_nsfw(result):
             return None, seed, get_error_html("تصویر خروجی حاوی محتوای نامناسب بود."), gr.update(visible=True), gr.update(visible=False)
 
-        # 8. پیام موفقیت
+        # 9. پیام موفقیت
         user_record = get_user_record(fingerprint)
         remaining = USAGE_LIMIT - user_record["count"] if user_record else 0
         success_msg = "تصویر با موفقیت ترکیب شد."
@@ -390,7 +424,8 @@ def infer(
 
 @spaces.GPU(duration=30)
 def infer_example(image_1, image_2, prompt, lora_adapter):
-    res, s, status, btn1, btn2 = infer(image_1, image_2, prompt, lora_adapter, 0, True, 1.0, 4, "example", "paid")
+    # مثال‌ها: ابعاد خودکار، اعتبار رایگان
+    res, s, status, btn1, btn2 = infer(image_1, image_2, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیش‌فرض)", 1024, 1024, "example", "paid")
     return res, s
 
 # ==========================================
@@ -526,6 +561,12 @@ css_code = """
 :root, body, .gradio-container { font-family: 'Vazirmatn', sans-serif !important; }
 #fingerprint_storage, #status_storage { display: none !important; }
 #col-container { margin: 0 auto; max-width: 1100px; direction: rtl; text-align: right; background: white; padding: 30px; border-radius: 20px; }
+
+/* Centered Titles */
+#main-title, #main-description {
+    text-align: center !important;
+}
+
 #badge-container { text-align: center; margin-bottom: 20px; height: 30px; }
 #user-sub-badge { padding: 6px 16px; border-radius: 20px; font-size: 0.9em; color: white; display: none; }
 .primary-btn { width: 100%; margin-top: 15px; background: linear-gradient(135deg, #FF4500, #FF8533); color: white; border: none; }
@@ -545,7 +586,10 @@ with gr.Blocks(theme=orange_red_theme) as demo:
 
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **ویرایشگر ترکیبی هوشمند (Fusion)**", elem_id="main-title")
-        gr.Markdown("ترکیب، انتقال بافت و ویرایش دو تصویر با هوش مصنوعی آلفا.")
+        gr.Markdown(
+            "ترکیب، انتقال بافت و ویرایش دو تصویر با هوش مصنوعی آلفا",
+            elem_id="main-description"
+        )
         gr.HTML('<div id="badge-container"><span id="user-sub-badge"></span></div>')
 
         with gr.Row(equal_height=True):
@@ -567,6 +611,23 @@ with gr.Blocks(theme=orange_red_theme) as demo:
                 upgrade_button = gr.Button("💎 ارتقا به نسخه نامحدود", variant="primary", elem_classes="upgrade-btn", elem_id="upgrade-btn", visible=False)
 
                 with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
+                    aspect_ratio_selection = gr.Dropdown(
+                        label="ابعاد تصویر خروجی",
+                        choices=ASPECT_RATIOS_LIST,
+                        value="خودکار (پیش‌فرض)",
+                        interactive=True
+                    )
+                    
+                    with gr.Row(visible=False) as custom_dims_row:
+                        custom_width = gr.Slider(
+                            label="عرض دلخواه (Width)",
+                            minimum=256, maximum=2048, step=16, value=1024
+                        )
+                        custom_height = gr.Slider(
+                            label="ارتفاع دلخواه (Height)",
+                            minimum=256, maximum=2048, step=16, value=1024
+                        )
+
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
                     randomize_seed = gr.Checkbox(label="تصادفی", value=True)
                     guidance_scale = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
@@ -588,8 +649,11 @@ with gr.Blocks(theme=orange_red_theme) as demo:
                 ["examples/M1.jpg", "examples/M2.jpg", "نورپردازی تصویر اول را حذف کن و بر اساس نور و رنگ تصویر دوم آن را مجدداً نورپردازی کن.", "انتقال نور (Light-Migration)"],
                 ["examples/Cloth2.jpg", "examples/Design2.png", "این طرح را روی پیراهن قرار بده.", "طراحی روی لباس (Cloth-Design)"],
                 ["examples/Cup1.png", "examples/Wood1.png", "بافت چوب را روی ماگ اعمال کن.", "ویرایش بافت (Texture Edit)"],
+                ["examples/Cloth1.jpg", "examples/Design1.png", "این طرح را روی پیراهن قرار بده.", "طراحی روی لباس (Cloth-Design)"],
                 ["examples/F3.jpg", "examples/F4.jpg", "عینک او را با عینک جدید از تصویر ۱ جایگزین کن.", "ترکیب پیشرفته (Super-Fusion)"],
                 ["examples/Chair.jpg", "examples/Material.jpg", "متریال تصویر ۱ را دقیقاً شبیه به مرجع تصویر ۲ کن.", "انتقال متریال (Material-Transfer)"],
+                ["examples/F1.jpg", "examples/F2.jpg", "بطری کوچک را روی میز قرار بده.", "ترکیب پیشرفته (Super-Fusion)"],
+                ["examples/Mug1.jpg", "examples/Texture1.jpg", "طرح تصویر ۲ را روی ماگ اعمال کن.", "ویرایش بافت (Texture Edit)"],
                 ["examples/Cat1.jpg", "examples/Glass1.webp", "یک گربه که عینک تصویر دوم را زده است.", "ترکیب اشیاء (Fuse-Objects)"],
             ],
             inputs=[image_1, image_2, prompt, lora_adapter],
@@ -599,12 +663,24 @@ with gr.Blocks(theme=orange_red_theme) as demo:
             label="نمونه‌ها"
         )
 
+    # تابع نمایش/مخفی کردن اسلایدر ابعاد
+    def toggle_row(choice):
+        if choice == "شخصی‌سازی (Custom)":
+            return gr.update(visible=True)
+        return gr.update(visible=False)
+
+    aspect_ratio_selection.change(fn=toggle_row, inputs=aspect_ratio_selection, outputs=custom_dims_row)
+
     # اتصال رویدادها
     fingerprint_box.change(fn=check_initial_quota, inputs=[fingerprint_box, status_box_input], outputs=[run_button, upgrade_button, status_box])
     
     run_button.click(
         fn=infer,
-        inputs=[image_1, image_2, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, fingerprint_box, status_box_input],
+        inputs=[
+            image_1, image_2, prompt, lora_adapter, seed, randomize_seed, 
+            guidance_scale, steps, aspect_ratio_selection, custom_width, custom_height,
+            fingerprint_box, status_box_input
+        ],
         outputs=[output_image, seed, status_box, run_button, upgrade_button]
     )