Update app.py

app.py

@@ -62,6 +62,11 @@ 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."
 }
 
+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)
@@ -99,13 +104,14 @@ def concat_images_horizontally(images, bg_color=(255, 255, 255)):
 @spaces.GPU(duration=300)
 def generate_turnaround(
     image,
+    extra_prompt="",                 # ← 追加: ユーザー追記用プロンプト
     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:
@@ -118,17 +124,23 @@ def generate_turnaround(
 
     pil_images = [input_image]
 
+    # 各プロンプト末尾に追記
+    p_front = _append_prompt(PROMPTS["front"], extra_prompt)
+    p_back  = _append_prompt(PROMPTS["back"],  extra_prompt)
+    p_left  = _append_prompt(PROMPTS["left"],  extra_prompt)
+    p_right = _append_prompt(PROMPTS["right"], extra_prompt)
+
     progress(0.25, desc="正面生成中...")
-    front = generate_single_view(pil_images, PROMPTS["front"], seed, num_inference_steps, true_guidance_scale)
+    front = generate_single_view(pil_images, p_front, seed, num_inference_steps, true_guidance_scale)
 
     progress(0.5, desc="背面生成中...")
-    back = generate_single_view([front], PROMPTS["back"], seed+1, num_inference_steps, true_guidance_scale)
+    back = generate_single_view([front], p_back, seed+1, num_inference_steps, true_guidance_scale)
 
     progress(0.75, desc="左側面生成中...")
-    left = generate_single_view([front], PROMPTS["left"], seed+2, num_inference_steps, true_guidance_scale)
+    left = generate_single_view([front], p_left, seed+2, num_inference_steps, true_guidance_scale)
 
     progress(1.0, desc="右側面生成中...")
-    right = generate_single_view([front], PROMPTS["right"], seed+3, num_inference_steps, true_guidance_scale)
+    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視点＋連結画像を生成しました"
@@ -145,6 +157,13 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Column(elem_id="col-container"):
         input_image = gr.Image(label="入力画像", type="pil", height=500)
+        # ← 追加: 追記プロンプト欄
+        extra_prompt = gr.Textbox(
+            label="追加プロンプト（各視点プロンプトの末尾に追記）",
+            placeholder="例: high detail, anime style, soft lighting, 4k, pastel colors",
+            lines=2
+        )
+
         run_button = gr.Button("🎨 生成開始", variant="primary")
         status_text = gr.Textbox(label="ステータス", interactive=False)
 
@@ -164,9 +183,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)
 
+    # 入力の並びに extra_prompt を追加
     run_button.click(
         fn=generate_turnaround,
-        inputs=[input_image, seed, randomize_seed, true_guidance_scale, num_inference_steps],
+        inputs=[input_image, extra_prompt, seed, randomize_seed, true_guidance_scale, num_inference_steps],
         outputs=[result_front, result_back, result_left, result_right, result_concat, seed, status_text],
     )