Update app.py

app.py

@@ -90,6 +90,14 @@ LORA_MAPPING = {
     "افزایش کیفیت (Upscale)": "upscale-image"
 }
 
+# --- گزینه‌های نسبت ابعاد ---
+ASPECT_RATIOS = {
+    "خودکار (پیش‌فرض)": "Auto",
+    "۱:۱ (مربع)": (1024, 1024),
+    "۱۶:۹ (افقی - لنداسکیپ)": (1344, 768),
+    "۹:۱۶ (عمودی - استوری)": (768, 1344)
+}
+
 def translate_prompt(text):
     """ترجمه متن فارسی به انگلیسی"""
     if not text:
@@ -130,6 +138,7 @@ def infer(
     randomize_seed,
     guidance_scale,
     steps,
+    aspect_ratio_selection, # پارامتر جدید
     progress=gr.Progress(track_tqdm=True)
 ):
     if input_image is None:
@@ -148,7 +157,18 @@ def infer(
     negative_prompt = "worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
 
     original_image = input_image.convert("RGB")
-    width, height = update_dimensions_on_upload(original_image)
+    
+    # --- منطق تعیین ابعاد ---
+    if aspect_ratio_selection == "خودکار (پیش‌فرض)" or aspect_ratio_selection is None:
+        width, height = update_dimensions_on_upload(original_image)
+    else:
+        # دریافت ابعاد از دیکشنری
+        dims = ASPECT_RATIOS.get(aspect_ratio_selection)
+        if dims and isinstance(dims, tuple):
+            width, height = dims
+        else:
+             # fallback to auto
+            width, height = update_dimensions_on_upload(original_image)
 
     result = pipe(
         image=original_image,
@@ -168,7 +188,8 @@ 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)
+    # برای نمونه‌ها حالت خودکار را پیش‌فرض قرار می‌دهیم
+    result, seed = infer(input_pil, prompt, lora_adapter, 0, True, guidance_scale, steps, "خودکار (پیش‌فرض)")
     return result, seed
 
 
@@ -266,6 +287,7 @@ textarea, input[type="text"] {
     background-color: #ffffff !important;
     color: #111827 !important;
     padding: 12px !important;
+    transition: all 0.3s ease;
     font-family: 'Vazirmatn', sans-serif !important;
 }
 
@@ -313,7 +335,7 @@ textarea:focus, input[type="text"]:focus {
     box-shadow: 0 8px 25px rgba(59, 130, 246, 0.45) !important;
 }
 
-/* --- 6. بخش نمونه‌ها (Examples) - رفع کامل سیاهی --- */
+/* --- 6. بخش نمونه‌ها --- */
 .gradio-container .prose table, 
 .gradio-container table {
     background-color: #ffffff !important;
@@ -427,10 +449,17 @@ with gr.Blocks() as demo:
                         choices=list(LORA_MAPPING.keys()),
                         value="تبدیل عکس به انیمه"
                     )
+                
+                # --- تنظیمات پیشرفته با قابلیت انتخاب سایز ---
                 with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
+                    aspect_ratio_selection = gr.Dropdown(
+                        label="ابعاد تصویر خروجی",
+                        choices=list(ASPECT_RATIOS.keys()),
+                        value="خودکار (پیش‌فرض)",
+                        interactive=True
+                    )
                     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)
         
@@ -459,7 +488,7 @@ with gr.Blocks() as demo:
     # اتصال دکمه اجرا
     run_button.click(
         fn=infer,
-        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps],
+        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, aspect_ratio_selection],
         outputs=[output_image, seed]
     )