Split into 2 tabs: Tab1 Identity (close-up face builder + send-to-scene as FaceID), Tab2 = existing scene builder (#39)

- Split into 2 tabs: Tab1 Identity (close-up face builder + send-to-scene as FaceID), Tab2 = existing scene builder (899f185fd8e1bd5177663d04e88316021b2ad93f)


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

app.py

@@ -219,6 +219,63 @@ def build_prompt(subject, age, ethnicity, skin, face, body, hair, eyes, outfit,
 # ---------------------------------------------------------------------------
 # Layout (mirrors the FLUX LoRA DLC reference UI)
 # ---------------------------------------------------------------------------
+def build_face_prompt(gender, age, ethnicity, faceshape, skin, eyecolor,
+                      eyes, brow, nose, mouth, hair, expr):
+    """Assemble Tab-1 face fields into an identity-focused prompt (Thai ok)."""
+    who = _field(gender) or "ผู้หญิง"
+    e, a = _field(ethnicity), _field(age)
+    if e:
+        who = f"{e} {who}"
+    if a:
+        who = f"{who} อายุ {a} ปี"
+    parts = [who]
+    for v in (faceshape, skin, eyecolor, eyes, brow, nose, mouth, hair, expr):
+        s = _field(v)
+        if s:
+            parts.append(s)
+    return ", ".join(parts) + ", detailed face, beautiful detailed eyes, detailed skin"
+
+
+@spaces.GPU(duration=120)
+def generate_face(model_id, prompt, translator):
+    """Tab 1: generate a close-up identity portrait to lock the face first."""
+    models = load_models()
+    cfg = pm.get_model(models, model_id)
+    if cfg is None:
+        raise gr.Error("ไม่พบโมเดล — กด Reload models ในแถบ 2 / model not found")
+    if not (prompt and str(prompt).strip()):
+        raise gr.Error("กรอกช่องใบหน้าแล้วกด '✨ สร้าง prompt ใบหน้า' ก่อน")
+    import random as _r
+    seed = _r.randint(0, MAX_SEED)
+    p = pm.translate_prompt(prompt, translator)
+    # force a close-up identity framing regardless of translator wording
+    p = "RAW photo, close-up portrait, face focus, headshot, looking at viewer, " + p
+    neg = cfg.get("negative_prompt", "")
+    try:
+        img = pm.run_generation(
+            cfg=cfg, mode="txt2img", prompt=p, negative_prompt=neg, ref_image=None,
+            steps=int(cfg.get("default_steps", 30)),
+            guidance=float(cfg.get("default_guidance", 5.0)),
+            denoise=0.4, ip_scale=0.7, width=512, height=640, seed=seed,
+        )
+    except Exception as e:  # noqa
+        traceback.print_exc()
+        raise gr.Error(str(e))
+    return img, f"✅ ใบหน้า · {cfg['label']} · seed {seed}"
+
+
+def send_face_to_scene(face_img):
+    """Copy the Tab-1 face into Tab-2's reference image + switch to FaceID."""
+    if face_img is None:
+        return gr.update(), gr.update(), "⚠️ ยังไม่มีใบหน้า — กด 'สร้างใบหน้า' ก่อน"
+    return (
+        gr.update(value=face_img),
+        gr.update(value="face_id"),
+        "✅ ส่งใบหน้าไปแถบ 2 แล้ว (ตั้งเป็น FaceID) — ไปแถบ '🎬 ฉาก/ท่าทาง' ตั้งค่าฉาก/ชุด/ท่า แล้วกด Generate",
+    )
+
+
+
 CSS = """
 #gen-btn {height: 100%; font-size: 1.3rem; font-weight: 700;}
 .card {border-radius: 14px;}
@@ -228,157 +285,188 @@ footer {visibility: hidden;}
 with gr.Blocks(css=CSS, theme=gr.themes.Soft(primary_hue="blue"),
                title="Character Studio") as demo:
     gr.Markdown("## 🎭 Character Studio — multi-model character generator (ZeroGPU)")
-
-    with gr.Row():
-        prompt = gr.Textbox(
-            label="Edit Prompt", lines=2, scale=4,
-            placeholder="✦ เลือกโมเดลแล้วพิมพ์ prompt / Choose a model and type the prompt",
-        )
-        gen_btn = gr.Button("Generate", variant="primary", scale=1, elem_id="gen-btn")
-
-    # ---- Prompt Builder: fill blanks instead of writing sentences ----
-    with gr.Accordion("📝 ตัวช่วยสร้าง prompt / Prompt Builder — กรอกช่อง ไม่ต้องเขียนประโยค",
-                      open=False):
-        gr.Markdown("กรอกเฉพาะช่องที่ต้องการ (เป็นภาษาไทยได้) แล้วกดปุ่มด้านล่าง "
-                    "ระบบจะประกอบเป็น prompt ให้ในช่องด้านบน · เว้นว่างช่องที่ไม่ใช้ได้")
-        with gr.Row():
-            b_subject = gr.Textbox(label="ใคร / ตัวละคร", value="ผู้หญิง")
-            b_age = gr.Textbox(label="อายุ", placeholder="เช่น 25")
-            b_ethnicity = gr.Textbox(label="เชื้อชาติ / สัญชาติ", placeholder="เช่น เกาหลี, ไทย")
-        with gr.Row():
-            b_body = gr.Dropdown(
-                label="รูปร่าง / สัดส่วน (เลือกได้หลายอัน · พิมพ์เองได้)",
-                choices=[
-                    ("ผอมเพรียว / slim", "slim, slender figure"),
-                    ("ต���วเล็กบอบบาง / petite", "petite, small frame"),
-                    ("ทรงนาฬิกาทราย / hourglass", "hourglass figure, narrow waist"),
-                    ("อวบอั๋นเว้าโค้ง / curvy", "curvy, voluptuous"),
-                    ("ฟิตมีกล้าม / athletic", "athletic, toned, fit body"),
-                    ("อึ๋มอกใหญ่ / busty", "busty, large breasts"),
-                    ("ท้วมอวบ / chubby", "chubby, plump"),
-                    ("สูงขายาวทรงนางแบบ / tall model", "tall, long legs, model body"),
-                    ("สะโพกผาย / wide hips", "wide hips"),
-                    ("ต้นขาหนา / thick thighs", "thick thighs"),
-                    ("เอวเล็ก / slim waist", "slim waist"),
-                    ("อกเล็ก / small breasts", "small breasts"),
-                    ("ทรงผู้ใหญ่ / mature", "mature body"),
-                ],
-                value=None, multiselect=True, allow_custom_value=True)
-            b_hair = gr.Textbox(label="ทรงผม / สีผม", placeholder="เช่น ผมยาวสีดำ, ผมบลอนด์")
-            b_eyes = gr.Textbox(label="สีตา", placeholder="เช่น ตาสีน้ำตาล")
-        with gr.Row():
-            b_skin = gr.Textbox(label="สีผิว", placeholder="เช่น ผิวขาว, ผิวสองสี, ผิวแทนทอง, ผิวเนียน")
-            b_face = gr.Textbox(label="โครงหน้า", placeholder="เช่น หน้าเรียว, หน้ารูปไข่, หน้ากลม, คางแหลม")
-        with gr.Row():
-            b_outfit = gr.Textbox(label="ชุด / เสื้อผ้า", placeholder="เช่น เดรสสีขาว, ชุดนักเรียน")
-            b_pose = gr.Dropdown(
-                label="ท่าโพส (เลือกผสมได้ · พิมพ์เองได้)",
-                choices=[
-                    ("ยืนตรง / standing", "standing"),
-                    ("ยืนมือเท้าเอว / hands on hips", "standing, hands on hips"),
-                    ("ยืนเอียงสะโพก / contrapposto", "standing, contrapposto pose"),
-                    ("ยืนพิงกำแพง / leaning on wall", "leaning against wall"),
-                    ("นั่งเก้าอี้ / sitting (chair)", "sitting on chair"),
-                    ("นั่งพื้น / sitting (floor)", "sitting on floor"),
-                    ("นั่งไขว่ห้าง / crossed legs", "sitting, crossed legs"),
-                    ("นั่งชันเข่า / knees up", "sitting, knees up, hugging knees"),
-                    ("คุกเข่า / kneeling", "kneeling"),
-                    ("นอนหงาย / lying on back", "lying on back"),
-                    ("นอนคว่ำ / on stomach", "lying on stomach"),
-                    ("นอนตะแคง / on side", "lying on side"),
-                    ("เดิน / walking", "walking"),
-                    ("วิ่ง / running", "running"),
-                    ("มือรวบผม / hand in hair", "hand in own hair, arm up"),
-                    ("กอดอก / arms crossed", "crossed arms"),
-                    ("โบกมือ / waving", "waving hand"),
-                ],
-                value=None, multiselect=True, allow_custom_value=True)
-            b_expr = gr.Textbox(label="สีหน้า / อารมณ์", placeholder="เช่น ยิ้มอ่อนๆ")
-        with gr.Row():
-            b_scene = gr.Textbox(label="สถานที่ / ฉาก", placeholder="เช่น คาเฟ่, ริมทะเล, ในสวน")
-            b_light = gr.Textbox(label="แสง", placeholder="เช่น แสงเช้านุ่มๆ, แสงสตูดิโอ")
-            b_shot = gr.Dropdown(
-                label="มุมกล้อง / ช็อต (เลือกได้หลายอัน · พิมพ์เองได้)",
-                choices=[
-                    ("โคลสอัพใบหน้า / close-up", "close-up portrait, face focus, headshot"),
-                    ("ครึ่งตัวบน / upper body", "upper body shot, bust"),
-                    ("ครึ่งตัว เอวขึ้นไป / half body", "half body shot, waist up"),
-                    ("เต็มตัว หัวถึงเท้า / full body", "full body shot, head to toe, full body visible"),
-                    ("ระยะไกล เห็นรอบตัว / wide shot", "wide shot, full body, distant, environmental"),
-                    ("มุมเงย / low angle", "from below, low angle shot"),
-                    ("มุมก้ม / high angle", "from above, high angle shot"),
-                    ("มุมข้าง โปรไฟล์ / side", "from side, profile view"),
-                    ("มุมหลัง / from behind", "from behind, back view"),
-                    ("เซลฟี่ / 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.
-    ids_state = gr.State(model_ids(MODELS))
-    selected_id = gr.State(MODELS[0]["id"] if MODELS else None)
-
-    with gr.Row(equal_height=False):
-        # ---- left: model picker ----
-        with gr.Column(scale=1):
-            with gr.Group():
-                gr.Markdown("### 🧩 เลือกโมเดล / Models")
-                model_gallery = gr.Gallery(
-                    value=gallery_items(MODELS),
-                    label=None, show_label=False, columns=2, height="auto",
-                    object_fit="cover", allow_preview=False, container=False,
-                    elem_classes="card",
+    with gr.Tabs():
+        with gr.Tab("👤 แถบ 1 · สร้างใบหน้า / Identity"):
+            gr.Markdown("### 👤 แถบ 1 · สร้างใบหน้า/อัตลักษณ์ก่อน — เน้นโคลสอัพให้ได้หน้าที่ต้องการ แล้วส่งไปแถบ 2")
+            gr.Markdown("กรอกลักษณะใบหน้า (ไทยได้) → กด สร้าง prompt → กด สร้างใบหน้า · ใช้โมเดลที่เลือกจากแถบ 2")
+            with gr.Row():
+                f_gender = gr.Textbox(label="เพศ", value="ผู้หญิง")
+                f_age = gr.Textbox(label="อายุ", placeholder="เช่น 22")
+                f_ethnicity = gr.Textbox(label="เชื้อชาติ", placeholder="เช่น ไทย, เกาหลี")
+            with gr.Row():
+                f_faceshape = gr.Textbox(label="โครงหน้า", placeholder="เช่น หน้าเรียว, รูปไข่, กลม")
+                f_skin = gr.Textbox(label="สีผิว", placeholder="เช่น ผิวขาว, สองสี, แทน")
+                f_eyecolor = gr.Textbox(label="สีตา", placeholder="เช่น น้ำตาล, ดำ")
+            with gr.Row():
+                f_eyes = gr.Textbox(label="ดวงตา", placeholder="เช่น ตากลมโต, ตาชั้นเดียว")
+                f_brow = gr.Textbox(label="คิ้ว", placeholder="เช่น คิ้วเข้ม, คิ้วโก่ง")
+                f_nose = gr.Textbox(label="จมูก", placeholder="เช่น จมูกโด่ง, เล็ก")
+            with gr.Row():
+                f_mouth = gr.Textbox(label="ปาก", placeholder="เช่น ปากอิ่ม, ยิ้มมุมปาก")
+                f_hair = gr.Textbox(label="ทรงผม / สีผม", placeholder="เช่น ผมยาวสีดำ")
+                f_expr = gr.Textbox(label="สีหน้า / อารมณ์", placeholder="เช่น ยิ้มอ่อนๆ, นิ่ง")
+            face_build_btn = gr.Button("✨ สร้าง prompt ใบหน้า", variant="secondary")
+            with gr.Row():
+                with gr.Column(scale=3):
+                    face_prompt = gr.Textbox(label="Prompt ใบหน้า (แก้ได้)", lines=2, placeholder="กดปุ่มด้านบน หรือพิมพ์เอง")
+                with gr.Column(scale=1):
+                    face_gen_btn = gr.Button("🎨 สร้างใบหน้า (close-up)", variant="primary")
+            face_output = gr.Image(label="ใบหน้าที่ได้", height=420, type="pil", elem_classes="card")
+            face_status = gr.Markdown("")
+            send_face_btn = gr.Button("➡️ ใช้ใบหน้านี้ในแถบ 2 (FaceID)", variant="secondary")
+            send_note = gr.Markdown("")
+        with gr.Tab("🎬 แถบ 2 · สร้างฉาก/ท่าทาง / Scene"):
+
+            with gr.Row():
+                prompt = gr.Textbox(
+                    label="Edit Prompt", lines=2, scale=4,
+                    placeholder="✦ เลือกโมเดลแล้วพิมพ์ prompt / Choose a model and type the prompt",
                 )
-                selected_md = gr.Markdown(
-                    f"**เลือก / Selected:** {MODELS[0]['label']}" if MODELS else ""
-                )
-                reload_btn = gr.Button("🔄 Reload models", size="sm")
-                reload_status = gr.Markdown("")
-
-            mode_radio = gr.Radio(
-                choices=modes_for(MODELS, MODELS[0]["id"]) if MODELS else [],
-                value="txt2img",
-                label="โหมดรูปต้นแบบ / Input mode",
-            )
-
-            translator = gr.Radio(
-                choices=[("ปิด / Off", "off"),
-                         ("NLLB-200 (เร็ว)", "nllb"),
-                         ("Typhoon 2 (ไทยแน่น)", "typhoon")],
-                value="typhoon",
-                label="แปลไทย→อังกฤษ / Auto-translate (พิมพ์ไทยได้เลย)",
-            )
-
-        # ---- right: output ----
-        with gr.Column(scale=1):
-            output = gr.Image(label="Generated Image", height=560, elem_classes="card")
-            status = gr.Markdown("")
-
-    # ---- advanced ----
-    with gr.Accordion("Advanced Settings", open=False):
-        with gr.Row():
-            with gr.Column():
-                ref_image = gr.Image(label="Input image (รูปต้นแบบ)", type="pil", height=240)
-                ip_scale = gr.Slider(0.0, 1.5, value=0.7, step=0.05,
-                                     label="Reference strength (IP-Adapter / FaceID)")
-                denoise = gr.Slider(0.1, 1.0, value=0.65, step=0.01,
-                                    label="Denoise strength (img2img · ต่ำ = อิงรูปมาก)")
-            with gr.Column():
-                negative_prompt = gr.Textbox(label="Negative prompt", lines=2)
+                gen_btn = gr.Button("Generate", variant="primary", scale=1, elem_id="gen-btn")
+
+            # ---- Prompt Builder: fill blanks instead of writing sentences ----
+            with gr.Accordion("📝 ตัวช่วยสร้าง prompt / Prompt Builder — กรอกช่อง ไม่ต้องเขียนประโยค",
+                              open=False):
+                gr.Markdown("กรอกเฉพาะช่องที่ต้องการ (เป็นภาษาไทยได้) แล้วกดปุ่มด้านล่าง "
+                            "ระบบจะประกอบเป็น prompt ให้ในช่องด้านบน · เว้นว่างช่องที่ไม่ใช้ได้")
+                with gr.Row():
+                    b_subject = gr.Textbox(label="ใคร / ตัวละคร", value="ผู้หญิง")
+                    b_age = gr.Textbox(label="อายุ", placeholder="เช่น 25")
+                    b_ethnicity = gr.Textbox(label="เชื้อชาติ / สัญชาติ", placeholder="เช่น เกาหลี, ไทย")
+                with gr.Row():
+                    b_body = gr.Dropdown(
+                        label="รูปร่าง / สัดส่วน (เลือกได้หลายอัน · พิมพ์เองได้)",
+                        choices=[
+                            ("ผอมเพรียว / slim", "slim, slender figure"),
+                            ("ตัวเล็กบอบบาง / petite", "petite, small frame"),
+                            ("ทรงนาฬิกาทราย / hourglass", "hourglass figure, narrow waist"),
+                            ("อวบอั๋นเว้าโค้ง / curvy", "curvy, voluptuous"),
+                            ("ฟิตมีกล้าม / athletic", "athletic, toned, fit body"),
+                            ("อึ๋มอกใหญ่ / busty", "busty, large breasts"),
+                            ("ท้วมอวบ / chubby", "chubby, plump"),
+                            ("สูงขายาวทรงนางแบบ / tall model", "tall, long legs, model body"),
+                            ("สะโพกผาย / wide hips", "wide hips"),
+                            ("ต้นขาหนา / thick thighs", "thick thighs"),
+                            ("เอวเล็ก / slim waist", "slim waist"),
+                            ("อกเล็ก / small breasts", "small breasts"),
+                            ("ทรงผู้ใหญ่ / mature", "mature body"),
+                        ],
+                        value=None, multiselect=True, allow_custom_value=True)
+                    b_hair = gr.Textbox(label="ทรงผม / สีผม", placeholder="เช่น ผม��าวสีดำ, ผมบลอนด์")
+                    b_eyes = gr.Textbox(label="สีตา", placeholder="เช่น ตาสีน้ำตาล")
                 with gr.Row():
-                    steps = gr.Slider(1, 50, value=28, step=1, label="Steps")
-                    guidance = gr.Slider(0.0, 15.0, value=6.5, step=0.1, label="Guidance (CFG)")
+                    b_skin = gr.Textbox(label="สีผิว", placeholder="เช่น ผิวขาว, ผิวสองสี, ผิวแทนทอง, ผิวเนียน")
+                    b_face = gr.Textbox(label="โครงหน้า", placeholder="เช่น หน้าเรียว, หน้ารูปไข่, หน้ากลม, คางแหลม")
                 with gr.Row():
-                    width = gr.Slider(384, 1280, value=512, step=64, label="Width")
-                    height = gr.Slider(384, 1280, value=768, step=64, label="Height")
+                    b_outfit = gr.Textbox(label="ชุด / เสื้อผ้า", placeholder="เช่น เดรสสีขาว, ชุดนักเรียน")
+                    b_pose = gr.Dropdown(
+                        label="ท่าโพส (เลือกผสมได้ · พิมพ์เองได้)",
+                        choices=[
+                            ("ยืนตรง / standing", "standing"),
+                            ("ยืนมือเท้าเอว / hands on hips", "standing, hands on hips"),
+                            ("ยืนเอียงสะโพก / contrapposto", "standing, contrapposto pose"),
+                            ("ยืนพิงกำแพง / leaning on wall", "leaning against wall"),
+                            ("นั่งเก้าอี้ / sitting (chair)", "sitting on chair"),
+                            ("นั่งพื้น / sitting (floor)", "sitting on floor"),
+                            ("นั่งไขว่ห้าง / crossed legs", "sitting, crossed legs"),
+                            ("นั่งชันเข่า / knees up", "sitting, knees up, hugging knees"),
+                            ("คุกเข่า / kneeling", "kneeling"),
+                            ("นอนหงาย / lying on back", "lying on back"),
+                            ("นอนคว่ำ / on stomach", "lying on stomach"),
+                            ("นอนตะแคง / on side", "lying on side"),
+                            ("เดิน / walking", "walking"),
+                            ("วิ่ง / running", "running"),
+                            ("มือรวบผม / hand in hair", "hand in own hair, arm up"),
+                            ("กอดอก / arms crossed", "crossed arms"),
+                            ("โบกมือ / waving", "waving hand"),
+                        ],
+                        value=None, multiselect=True, allow_custom_value=True)
+                    b_expr = gr.Textbox(label="สีหน้า / อารมณ์", placeholder="เช่น ยิ้มอ่อนๆ")
                 with gr.Row():
-                    seed = gr.Number(value=-1, label="Seed (-1 = random)", precision=0)
-                    randomize = gr.Checkbox(value=True, label="Randomize seed")
+                    b_scene = gr.Textbox(label="สถานที่ / ฉาก", placeholder="เช่น คาเฟ่, ริมทะเล, ในสวน")
+                    b_light = gr.Textbox(label="แสง", placeholder="เช่น แสงเช้านุ่มๆ, แสงสตูดิโอ")
+                    b_shot = gr.Dropdown(
+                        label="มุมกล้อง / ช็อต (เลือกได้หลายอัน · พิมพ์เองได้)",
+                        choices=[
+                            ("โคลสอัพใบหน้า / close-up", "close-up portrait, face focus, headshot"),
+                            ("ครึ่งตัวบน / upper body", "upper body shot, bust"),
+                            ("ครึ่งตัว เอวขึ้นไป / half body", "half body shot, waist up"),
+                            ("เต็มตัว หัวถึงเท้า / full body", "full body shot, head to toe, full body visible"),
+                            ("ระยะไกล เห็นรอบตัว / wide shot", "wide shot, full body, distant, environmental"),
+                            ("มุมเงย / low angle", "from below, low angle shot"),
+                            ("มุมก้ม / high angle", "from above, high angle shot"),
+                            ("มุมข้าง โปรไฟล์ / side", "from side, profile view"),
+                            ("มุมหลัง / from behind", "from behind, back view"),
+                            ("เซลฟี่ / 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.
+            ids_state = gr.State(model_ids(MODELS))
+            selected_id = gr.State(MODELS[0]["id"] if MODELS else None)
+
+            with gr.Row(equal_height=False):
+                # ---- left: model picker ----
+                with gr.Column(scale=1):
+                    with gr.Group():
+                        gr.Markdown("### 🧩 เลือกโมเดล / Models")
+                        model_gallery = gr.Gallery(
+                            value=gallery_items(MODELS),
+                            label=None, show_label=False, columns=2, height="auto",
+                            object_fit="cover", allow_preview=False, container=False,
+                            elem_classes="card",
+                        )
+                        selected_md = gr.Markdown(
+                            f"**เลือก / Selected:** {MODELS[0]['label']}" if MODELS else ""
+                        )
+                        reload_btn = gr.Button("🔄 Reload models", size="sm")
+                        reload_status = gr.Markdown("")
+
+                    mode_radio = gr.Radio(
+                        choices=modes_for(MODELS, MODELS[0]["id"]) if MODELS else [],
+                        value="txt2img",
+                        label="โหมดรูปต้นแบบ / Input mode",
+                    )
+
+                    translator = gr.Radio(
+                        choices=[("ปิด / Off", "off"),
+                                 ("NLLB-200 (เร็ว)", "nllb"),
+                                 ("Typhoon 2 (ไทยแน่น)", "typhoon")],
+                        value="typhoon",
+                        label="แปลไทย→อังกฤษ / Auto-translate (พิมพ์ไทยได้เลย)",
+                    )
+
+                # ---- right: output ----
+                with gr.Column(scale=1):
+                    output = gr.Image(label="Generated Image", height=560, elem_classes="card")
+                    status = gr.Markdown("")
+
+            # ---- advanced ----
+            with gr.Accordion("Advanced Settings", open=False):
+                with gr.Row():
+                    with gr.Column():
+                        ref_image = gr.Image(label="Input image (รูปต้นแบบ)", type="pil", height=240)
+                        ip_scale = gr.Slider(0.0, 1.5, value=0.7, step=0.05,
+                                             label="Reference strength (IP-Adapter / FaceID)")
+                        denoise = gr.Slider(0.1, 1.0, value=0.65, step=0.01,
+                                            label="Denoise strength (img2img · ต่ำ = อิงรูปมาก)")
+                    with gr.Column():
+                        negative_prompt = gr.Textbox(label="Negative prompt", lines=2)
+                        with gr.Row():
+                            steps = gr.Slider(1, 50, value=28, step=1, label="Steps")
+                            guidance = gr.Slider(0.0, 15.0, value=6.5, step=0.1, label="Guidance (CFG)")
+                        with gr.Row():
+                            width = gr.Slider(384, 1280, value=512, step=64, label="Width")
+                            height = gr.Slider(384, 1280, value=768, step=64, label="Height")
+                        with gr.Row():
+                            seed = gr.Number(value=-1, label="Seed (-1 = random)", precision=0)
+                            randomize = gr.Checkbox(value=True, label="Randomize seed")
 
     # ---- wiring ----
     model_gallery.select(
@@ -404,6 +492,26 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft(primary_hue="blue"),
     prompt.submit(generate, inputs=gen_inputs, outputs=[output, seed, status])
 
 
+
+    # ---- Tab 1 (identity) wiring ----
+    face_build_btn.click(
+        build_face_prompt,
+        inputs=[f_gender, f_age, f_ethnicity, f_faceshape, f_skin, f_eyecolor,
+                f_eyes, f_brow, f_nose, f_mouth, f_hair, f_expr],
+        outputs=face_prompt,
+    )
+    face_gen_btn.click(
+        generate_face,
+        inputs=[selected_id, face_prompt, translator],
+        outputs=[face_output, face_status],
+    )
+    send_face_btn.click(
+        send_face_to_scene,
+        inputs=[face_output],
+        outputs=[ref_image, mode_radio, send_note],
+    )
+
+
 if __name__ == "__main__":
     # allowed_paths lets Gradio serve the local model preview thumbnails.
     demo.queue(max_size=12).launch(allowed_paths=["previews"])