Update app.py

app.py

@@ -45,7 +45,7 @@ pipe.load_lora_weights(
         "2vXpSwA7/iroiro-lora", 
         weight_name="qwen_lora/Qwen-Image-Edit-2509-Lightning-4steps-V1.0-bf16_dim1.safetensors"
     )
-pipe.fuse_lora()
+pipe.fuse_lora(lora_scale=0.8)
 
 # Apply the same optimizations from the first version
 pipe.transformer.__class__ = QwenImageTransformer2DModel
@@ -106,7 +106,7 @@ def generate_turnaround(
     
     # 入力画像の確認
     if image is None:
-        return None, None, None, None, seed, "エラー: 入力画像をアップロードしてください"
+        return None, None, None, None, None, seed, "エラー: 入力画像をアップロードしてください"
     
     # PIL画像として処理
     if isinstance(image, Image.Image):
@@ -115,7 +115,7 @@ def generate_turnaround(
         try:
             input_image = Image.open(image).convert("RGB")
         except:
-            return None, None, None, None, seed, "エラー: 画像の読み込みに失敗しました"
+            return None, None, None, None, None, seed, "エラー: 画像の読み込みに失敗しました"
     
     pil_images = [input_image]
     
@@ -135,7 +135,25 @@ def generate_turnaround(
     progress(1.0, desc="右側面立ち絵を生成中...")
     right_image = generate_single_view([front_image], PROMPTS["right"], seed+3, num_inference_steps, true_guidance_scale)
     
-    return front_image, back_image, left_image, right_image, seed, "✅ 4視点の立ち絵生成が完了しました"
+    # 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視点の立ち絵生成が完了しました"
 
 # --- UI Layout ---
 css = """
@@ -185,6 +203,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")
 
         with gr.Accordion("⚙️ 詳細設定", open=False):
             seed = gr.Slider(
@@ -223,7 +244,7 @@ with gr.Blocks(css=css) as demo:
             true_guidance_scale,
             num_inference_steps,
         ],
-        outputs=[result_front, result_back, result_left, result_right, seed, status_text],
+        outputs=[result_front, result_back, result_left, result_right, result_combined, seed, status_text],
     )
 
 if __name__ == "__main__":