Update app.py

app.py

@@ -62,13 +62,20 @@ PROMPTS = {
     "right": "Move the camera to a side view (profile) from the right so the full body of the character is visible. Background is plain white."
 }
 
+# NEW: 出力解像度プリセット
+RESOLUTIONS = {
+    "1:4": (512, 2048),
+    "1:3": (576, 1728),
+    "nealy 9:16": (768, 1344),
+    "nealy 2:3": (832, 1216),
+    "3:4": (896, 1152),
+}
+
 def _append_prompt(base: str, extra: str) -> str:
-    """末尾にユーザー指定のプロンプトを追記（空なら変更なし）"""
     extra = (extra or "").strip()
     return (base if not extra else f"{base} {extra}").strip()
 
 def generate_single_view(input_images, prompt, seed, num_inference_steps, true_guidance_scale):
-    """単一視点画像生成"""
     generator = torch.Generator(device=device).manual_seed(seed)
     result = pipe(
         image=input_images if input_images else None,
@@ -82,7 +89,6 @@ def generate_single_view(input_images, prompt, seed, num_inference_steps, true_g
     return result[0]
 
 def concat_images_horizontally(images, bg_color=(255, 255, 255)):
-    """画像を横に連結"""
     images = [img.convert("RGB") for img in images if img is not None]
     if not images:
         return None
@@ -101,17 +107,23 @@ def concat_images_horizontally(images, bg_color=(255, 255, 255)):
         x += img.width
     return canvas
 
+# NEW: リサイズユーティリティ
+def resize_to_preset(img: Image.Image, preset_key: str) -> Image.Image:
+    w, h = RESOLUTIONS[preset_key]
+    return img.resize((w, h), Image.LANCZOS)
+
 @spaces.GPU()
 def generate_turnaround(
     image,
     extra_prompt="",
+    preset_key="nealy 9:16",  # NEW: デフォルト
     seed=42,
     randomize_seed=False,
     true_guidance_scale=1.0,
     num_inference_steps=4,
     progress=gr.Progress(track_tqdm=True),
 ):
-    """4視点＋横連結PNG生成（ユーザー追記プロンプト対応）"""
+    """4視点＋横連結PNG生成（ユーザー追記プロンプト対応 & 出力解像度プリセット対応）"""
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     if image is None:
@@ -142,8 +154,16 @@ def generate_turnaround(
     progress(1.0, desc="右側面生成中...")
     right = generate_single_view([front], p_right, seed+3, num_inference_steps, true_guidance_scale)
 
-    concat = concat_images_horizontally([front, right, back, left])
-    return front, back, left, right, concat, seed, "✅ PNG形式で4視点＋連結画像を生成しました"
+    # NEW: ここで指定プリセットにリサイズ
+    front_r = resize_to_preset(front, preset_key)
+    back_r  = resize_to_preset(back,  preset_key)
+    left_r  = resize_to_preset(left,  preset_key)
+    right_r = resize_to_preset(right, preset_key)
+
+    # NEW: リサイズ後を連結（横：正面→右→背面→左）
+    concat = concat_images_horizontally([front_r, right_r, back_r, left_r])
+
+    return front_r, back_r, left_r, right_r, concat, seed, f"✅ {preset_key} にリサイズして4視点＋連結画像を生成しました"
 
 # --- UI ---
 css = """
@@ -167,7 +187,6 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown("アップロードしたキャラクター画像から正面・背面・左右側面、さらに4枚連結のPNG画像を出力します。")
 
     with gr.Column(elem_id="col-container"):
-        # 追加: 注意文（アップロード欄の直前に表示）
         gr.HTML(
             "<div class='notice'>"
             "注意：他者が作成した画像のアップロードはご遠慮ください。"
@@ -185,6 +204,13 @@ with gr.Blocks(css=css) as demo:
             lines=2
         )
 
+        # NEW: 出力解像度プリセットのプルダウン
+        preset_dropdown = gr.Dropdown(
+            label="出力解像度プリセット",
+            choices=list(RESOLUTIONS.keys()),
+            value="nealy 9:16"
+        )
+
         run_button = gr.Button("🎨 生成開始", variant="primary")
         status_text = gr.Textbox(label="ステータス", interactive=False)
 
@@ -204,9 +230,10 @@ with gr.Blocks(css=css) as demo:
             true_guidance_scale = gr.Slider(label="True guidance scale", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
             num_inference_steps = gr.Slider(label="生成ステップ数", minimum=1, maximum=40, step=1, value=4)
 
+    # NEW: クリック時に preset_dropdown を引数として渡す
     run_button.click(
         fn=generate_turnaround,
-        inputs=[input_image, extra_prompt, seed, randomize_seed, true_guidance_scale, num_inference_steps],
+        inputs=[input_image, extra_prompt, preset_dropdown, seed, randomize_seed, true_guidance_scale, num_inference_steps],
         outputs=[result_front, result_back, result_left, result_right, result_concat, seed, status_text],
     )