Qwen-Image-2509-CharacterSheet

Running on Zero

App Files Files Community

tori29umai commited on 13 days ago

Commit

9d7a47a

verified ·

1 Parent(s): 35a6f3d

Update app.py

Browse files

Files changed (1) hide show

app.py +35 -8

app.py CHANGED Viewed

@@ -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."
 }
 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
 @spaces.GPU()
 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生成（ユーザー追記プロンプト対応）"""
  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視点＋連結画像を生成しました"
 # --- 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
  )
  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)
  run_button.click(
  fn=generate_turnaround,
- 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],
  )

  "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,
  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
  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生成（ユーザー追記プロンプト対応 & 出力解像度プリセット対応）"""
  if randomize_seed:
  seed = random.randint(0, MAX_SEED)
  if image is None:
  progress(1.0, desc="右側面生成中...")
  right = generate_single_view([front], p_right, seed+3, num_inference_steps, true_guidance_scale)
+ # 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 = """
  gr.Markdown("アップロードしたキャラクター画像から正面・背面・左右側面、さらに4枚連結のPNG画像を出力します。")
  with gr.Column(elem_id="col-container"):
  gr.HTML(
  "<div class='notice'>"
  "注意：他者が作成した画像のアップロードはご遠慮ください。"
  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)
  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, 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],
  )