Update app.py

app.py

@@ -88,6 +88,34 @@ def generate_single_view(input_images, prompt, seed, num_inference_steps, true_g
     
     return result[0]
 
+# --- NEW: 横連結ユーティリティ ---
+def concat_images_horizontally(images, bg_color=(255, 255, 255)):
+    """
+    複数のPIL画像を横に連結して1枚のPIL画像を返す。
+    すべて同じ高さにスケールしてアスペクト比は維持。
+    """
+    images = [img.convert("RGB") for img in images if img is not None]
+    if not images:
+        return None
+
+    # 連結の基準高さ（最大の高さ）に合わせて横幅を等比リサイズ
+    target_h = max(img.height for img in images)
+    resized = []
+    for img in images:
+        if img.height != target_h:
+            new_w = int(img.width * (target_h / img.height))
+            img = img.resize((new_w, target_h), Image.LANCZOS)
+        resized.append(img)
+
+    total_w = sum(img.width for img in resized)
+    canvas = Image.new("RGB", (total_w, target_h), bg_color)
+
+    x = 0
+    for img in resized:
+        canvas.paste(img, (x, 0))
+        x += img.width
+    return canvas
+
 # --- Main Inference Function ---
 @spaces.GPU(duration=300)
 def generate_turnaround(
@@ -100,6 +128,7 @@ def generate_turnaround(
 ):
     """
     入力画像から4つの視点（正面、背面、左側面、右側面）の立ち絵を生成
+    さらに「正面 → 右向き → 背面 → 左向き」を横並びで連結した画像も返す
     """
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -134,26 +163,11 @@ def generate_turnaround(
     # 4. 正面立ち絵を入力として右側面を生成
     progress(1.0, desc="右側面立ち絵を生成中...")
     right_image = generate_single_view([front_image], PROMPTS["right"], seed+3, num_inference_steps, true_guidance_scale)
-    
-    # 5. 4つの画像を連結（正面、右向き、背面、左向きの順）
-    images = [front_image, right_image, back_image, left_image]
-    widths = [img.width for img in images]
-    heights = [img.height for img in images]
-    
-    # 連結画像のサイズを計算
-    total_width = sum(widths)
-    max_height = max(heights)
-    
-    # 新しい画像を作成
-    combined_image = Image.new('RGB', (total_width, max_height))
-    
-    # 画像を左から順に貼り付け
-    x_offset = 0
-    for img in images:
-        combined_image.paste(img, (x_offset, 0))
-        x_offset += img.width
-    
-    return front_image, back_image, left_image, right_image, combined_image, seed, "✅ 4視点の立ち絵生成が完了しました"
+
+    # --- NEW: 指定順で横連結（正面 → 右向き → 背面 → 左向き） ---
+    concatenated = concat_images_horizontally([front_image, right_image, back_image, left_image])
+
+    return front_image, back_image, left_image, right_image, concatenated, seed, "✅ 4視点の立ち絵生成と横連結画像の出力が完了しました"
 
 # --- UI Layout ---
 css = """
@@ -177,7 +191,7 @@ css = """
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# キャラクター4視点立ち絵自動生成")
     gr.Markdown("キャラクター画像をアップロードすると、正面・背面・左側面・右側面の4つの立ち絵を自動生成します")
-    
+
     with gr.Column(elem_id="col-container"):
         with gr.Row():
             input_image = gr.Image(
@@ -203,9 +217,9 @@ with gr.Blocks(css=css) as demo:
                 result_left = gr.Image(label="左側面", type="pil", height=500, show_download_button=True, container=True, image_mode="RGB")
             with gr.Column():
                 result_right = gr.Image(label="右側面", type="pil", height=500, show_download_button=True, container=True, image_mode="RGB")
-        
-        gr.Markdown("### 連結画像（正面→右向き→背面→左向き）")
-        result_combined = gr.Image(label="4視点連結画像", type="pil", height=400, show_download_button=True, container=True, image_mode="RGB")
+
+        # --- NEW: 横連結画像の出力欄 ---
+        result_concat = gr.Image(label="連結（正面 → 右向き → 背面 → 左向き）", type="pil", height=500, show_download_button=True, container=True, image_mode="RGB")
 
         with gr.Accordion("⚙️ 詳細設定", open=False):
             seed = gr.Slider(
@@ -244,7 +258,8 @@ with gr.Blocks(css=css) as demo:
             true_guidance_scale,
             num_inference_steps,
         ],
-        outputs=[result_front, result_back, result_left, result_right, result_combined, seed, status_text],
+        # --- NEW: 5つ目の出力として連結画像を追加 ---
+        outputs=[result_front, result_back, result_left, result_right, result_concat, seed, status_text],
     )
 
 if __name__ == "__main__":