Prompt Builder: add pose reference image upload -> auto ControlNet OpenPose lock (reuses pose path), prompt drives face/clothes/scene (#38)

- Prompt Builder: add pose reference image upload -> auto ControlNet OpenPose lock (reuses pose path), prompt drives face/clothes/scene (1a001728297c731b0bdee85f9ee389dc57d4a148)


Co-authored-by: pormungtailaw <pormungtai@users.noreply.huggingface.co>

app.py

@@ -80,12 +80,23 @@ def modes_for(models, model_id):
 @spaces.GPU(duration=120)
 def generate(model_id, mode, prompt, negative_prompt, ref_image,
              steps, guidance, denoise, ip_scale, width, height, seed, randomize,
-             translator):
+             translator, pose_ref_image=None):
     models = load_models()
     cfg = pm.get_model(models, model_id)
     if cfg is None:
         raise gr.Error("ไม่พบโมเดลที่เลือก โปรด Reload models / Selected model not found.")
 
+    # Prompt Builder pose reference: if the user uploaded a pose image, lock the pose
+    # to it via ControlNet OpenPose (reuses the tested 'pose' path) regardless of the
+    # mode radio — face/clothes/scene still come from the text prompt.
+    if pose_ref_image is not None:
+        if "pose" in pm.SUPPORTED_MODES.get(cfg["base"], []):
+            mode = "pose"
+            ref_image = pose_ref_image
+        else:
+            raise gr.Error("โมเดลนี้ไม่รองรับ Pose lock (ใช้ได้กับ SD1.5 / SDXL) — "
+                           "เลือกโมเดล SD1.5 เช่น majicMIX แล้วลองใหม่")
+
     if randomize or seed is None or int(seed) < 0:
         seed = random.randint(0, MAX_SEED)
 
@@ -301,6 +312,11 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft(primary_hue="blue"),
                     ("เซลฟี่ / selfie", "selfie, pov, arm extended"),
                 ],
                 value=None, multiselect=True, allow_custom_value=True)
+        with gr.Row():
+            b_pose_img = gr.Image(
+                label="📷 รูปอ้างอิงท่าโพส (ใส่รูป → ล็อกท่าตามรูปด้วย ControlNet OpenPose · "
+                      "หน้า/ชุด/ฉาก ยังคุมด้วยช่องด้านบน · เว้นว่าง = ใช้ข้อความท่าโพส)",
+                type="pil", sources=["upload"], height=240)
         build_btn = gr.Button("✨ สร้าง prompt → ใส่ในช่องด้านบน", variant="secondary")
 
     # State: ordered list of model ids (matches gallery order) + current selection.
@@ -383,7 +399,7 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft(primary_hue="blue"),
 
     gen_inputs = [selected_id, mode_radio, prompt, negative_prompt, ref_image,
                   steps, guidance, denoise, ip_scale, width, height, seed, randomize,
-                  translator]
+                  translator, b_pose_img]
     gen_btn.click(generate, inputs=gen_inputs, outputs=[output, seed, status])
     prompt.submit(generate, inputs=gen_inputs, outputs=[output, seed, status])