Add Quick Start scenario pickers (Norma Jean + Marilyn) with per-scenario input-requirement hints

app.py

@@ -342,25 +342,117 @@ css = """
 }
 """
 
-IDENTITY_PRESETS = [
-    ("Face Close-up", "Same person, face close-up portrait, studio lighting, sharp focus, looking slightly left, clean solid background"),
-    ("3/4 Angle", "Same person, three-quarter angle view, upper body, soft directional light from left, slight smile"),
-    ("Full Body", "Same person, full body standing, neutral pose, studio lighting, solid gray background"),
-    ("Over Shoulder", "Same person from behind, looking back over shoulder, natural light, upper body and face visible"),
-    ("Shirtless", "Same person, shirtless, upper body, warm studio lighting, confident relaxed expression, simple background"),
-    ("Golden Hour", "Same person, golden hour sunlight, outdoor, warm tones, half body, candid natural feel"),
-    ("Dramatic Lit", "Same person, dramatic chiaroscuro side lighting, moody shadows, face and torso, high contrast, cinematic"),
-    ("Mirror Shot", "Same person, casual mirror selfie, bathroom or bedroom, natural smartphone lighting, authentic and candid"),
-]
+# ============ Quick Start Scenarios (one-click: sets prompt + every setting + tells
+# you what the input should be) ============
+# Identity (Norma Jean / Kontext) scenarios. Kontext re-poses from a reference, so the
+# input hint is about what reference photo gives the cleanest result.
+IDENTITY_SCENARIOS = {
+    "Face close-up (studio)": {
+        "prompt": "Same person, face close-up portrait, sharp focus, looking toward camera, even studio lighting, clean solid background",
+        "candid": False, "nsfw": 0.3, "realism": 1.0, "guidance": 2.5, "steps": 28,
+        "hint": "a clear, well-lit photo where the FACE is visible. Front-on or a slight angle is best.",
+    },
+    "3/4 angle portrait": {
+        "prompt": "Same person, three-quarter angle view, upper body, soft directional light from the left, subtle natural expression",
+        "candid": False, "nsfw": 0.3, "realism": 1.0, "guidance": 2.5, "steps": 28,
+        "hint": "any clear face photo. Head-and-shoulders works great.",
+    },
+    "Full body (training staple)": {
+        "prompt": "Same person, full body standing, neutral relaxed pose, even studio lighting, solid gray background",
+        "candid": False, "nsfw": 0.4, "realism": 1.0, "guidance": 2.5, "steps": 28,
+        "hint": "a photo that shows his build if you have one. Face-only still works — the body is inferred.",
+    },
+    "Shirtless torso": {
+        "prompt": "Same person, shirtless, upper body and torso, warm studio light, relaxed confident expression, simple background",
+        "candid": False, "nsfw": 1.0, "realism": 1.0, "guidance": 2.5, "steps": 28,
+        "hint": "face + upper body if possible. Kontext keeps the face and generates the torso.",
+    },
+    "Candid mirror selfie": {
+        "prompt": "Same person, casual mirror selfie, phone in hand, bedroom or bathroom, natural window light, authentic",
+        "candid": True, "nsfw": 0.5, "realism": 1.0, "guidance": 2.5, "steps": 28,
+        "hint": "any clear face photo. This one leans hardest on candid realism.",
+    },
+    "Golden hour outdoor": {
+        "prompt": "Same person, outdoors at golden hour, warm low sunlight, half body, relaxed candid feel",
+        "candid": True, "nsfw": 0.4, "realism": 1.0, "guidance": 2.5, "steps": 28,
+        "hint": "a clear face photo — the lighting gets replaced with golden hour.",
+    },
+    "Dramatic cinematic": {
+        "prompt": "Same person, dramatic side lighting, moody shadows, face and torso, high contrast, cinematic",
+        "candid": False, "nsfw": 0.5, "realism": 1.0, "guidance": 3.0, "steps": 30,
+        "hint": "a sharp, well-exposed face photo. Dramatic lighting needs clean features to work from.",
+    },
+    "Over-the-shoulder / back": {
+        "prompt": "Same person seen from behind, looking back over the shoulder, upper body and face visible, natural light",
+        "candid": False, "nsfw": 0.6, "realism": 1.0, "guidance": 2.5, "steps": 28,
+        "hint": "a front-facing face photo is fine — the model turns the pose for you.",
+    },
+}
 
-EXPLICIT_PRESETS = [
-    ("Full Nude Stand", "same person, fully nude, full body standing, anatomically correct male, penis visible, studio lighting, solid neutral background, realistic, high quality"),
-    ("Nude Reclining", "same person, nude, reclining on bed, relaxed natural pose, full body visible, anatomically detailed male, warm soft bedroom light, intimate, realistic"),
-    ("Shower", "same person, nude, wet skin, standing in shower, water droplets, steam, full body, anatomically correct, natural lighting, realistic"),
-    ("Nude Back", "same person, nude from behind, athletic back and buttocks visible, looking over shoulder, studio lighting, artistic, realistic"),
-    ("Seated Nude", "same person, seated nude on chair, legs apart, confident relaxed pose, full body, anatomically detailed, studio lighting, realistic"),
-    ("Morning Stretch", "same person, nude, stretching in morning light, full body, anatomically correct male, golden hour through window, natural, realistic"),
-]
+# Explicit (Marilyn / FLUX.1-dev img2img) scenarios. This is img2img: it KEEPS the
+# composition of the starting image, so the input hint really matters here.
+EXPLICIT_SCENARIOS = {
+    "Full nude — standing": {
+        "prompt": "same person, fully nude, full body standing, anatomically correct male, penis visible, studio lighting, solid neutral background, realistic, high quality",
+        "strength": 0.80, "guidance": 3.5, "steps": 28, "nsfw": 1.0, "realism": 1.0, "candid": False,
+        "hint": "a FULL-BODY STANDING shot. Img2img keeps the pose — start standing to get a standing nude.",
+    },
+    "Nude — reclining / lying down": {
+        "prompt": "same person, nude, reclining on bed, relaxed, full body visible, anatomically detailed male, warm soft bedroom light, intimate, realistic",
+        "strength": 0.80, "guidance": 3.5, "steps": 28, "nsfw": 1.0, "realism": 1.0, "candid": False,
+        "hint": "a shot where he is ALREADY LYING DOWN or reclining. Img2img won't cleanly turn a standing photo into a lying-down one.",
+    },
+    "Shower / wet": {
+        "prompt": "same person, nude, wet skin, standing in shower, water droplets, steam, full body, anatomically correct, natural lighting, realistic",
+        "strength": 0.80, "guidance": 3.5, "steps": 28, "nsfw": 1.0, "realism": 1.0, "candid": False,
+        "hint": "a STANDING full-body shot.",
+    },
+    "Nude from behind": {
+        "prompt": "same person, nude from behind, athletic back and buttocks visible, looking over shoulder, studio lighting, realistic",
+        "strength": 0.80, "guidance": 3.5, "steps": 28, "nsfw": 1.0, "realism": 1.0, "candid": False,
+        "hint": "a standing shot; a back or 3/4 view works best.",
+    },
+    "Keep-the-face light nude": {
+        "prompt": "same person, nude, natural relaxed pose, anatomically correct male, soft lighting, realistic",
+        "strength": 0.60, "guidance": 3.5, "steps": 28, "nsfw": 1.0, "realism": 1.0, "candid": False,
+        "hint": "your BEST identity shot. Low strength (0.60) preserves the face — also fill in the Character Description box.",
+    },
+    "Explicit — max freedom": {
+        "prompt": "same person, nude, explicit, anatomically detailed male, realistic, high quality",
+        "strength": 0.85, "guidance": 3.5, "steps": 28, "nsfw": 1.0, "realism": 1.0, "candid": False,
+        "hint": "a full-body shot. High strength (0.85) changes the body most — fill in the Character Description box to hold the face.",
+    },
+}
+
+
+def apply_identity_scenario(name):
+    s = IDENTITY_SCENARIOS.get(name)
+    if not s:
+        return (gr.update(),) * 7
+    return (
+        gr.update(value=s["prompt"]),
+        gr.update(value=s["candid"]),
+        gr.update(value=s["nsfw"]),
+        gr.update(value=s["realism"]),
+        gr.update(value=s["guidance"]),
+        gr.update(value=s["steps"]),
+        gr.update(value=f"**Best input for this scenario:** {s['hint']}"),
+    )
+
+
+def apply_explicit_scenario(name):
+    s = EXPLICIT_SCENARIOS.get(name)
+    if not s:
+        return (gr.update(),) * 7
+    return (
+        gr.update(value=s["prompt"]),
+        gr.update(value=s["strength"]),
+        gr.update(value=s["nsfw"]),
+        gr.update(value=s["realism"]),
+        gr.update(value=s["guidance"]),
+        gr.update(value=s["steps"]),
+        gr.update(value=f"**Best input for this scenario:** {s['hint']}"),
+    )
 
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     gallery_state = gr.State([])
@@ -399,6 +491,17 @@ FLUX.1-dev has weaker safety training, so the NSFW LoRAs actually work here. Bes
                             type="pil",
                             height=300,
                         )
+
+                        with gr.Group():
+                            gr.Markdown("**🚀 Quick Start — pick a scenario, everything gets set up for you**")
+                            with gr.Row():
+                                id_scenario_dd = gr.Dropdown(
+                                    choices=list(IDENTITY_SCENARIOS.keys()),
+                                    label="Scenario", value=None, scale=3,
+                                )
+                                id_scenario_btn = gr.Button("Apply", variant="secondary", scale=1)
+                            id_scenario_hint = gr.Markdown("")
+
                         id_prompt = gr.Textbox(
                             label="Variation Prompt",
                             placeholder="Same person, shirtless, different angle...",
@@ -408,15 +511,6 @@ FLUX.1-dev has weaker safety training, so the NSFW LoRAs actually work here. Bes
                             id_gen_btn = gr.Button("Generate", variant="primary", scale=2)
                             id_gen_add_btn = gr.Button("Generate & Add", variant="secondary", scale=2)
 
-                        with gr.Accordion("Presets", open=True):
-                            for i in range(0, len(IDENTITY_PRESETS), 2):
-                                with gr.Row():
-                                    for j in range(2):
-                                        if i + j < len(IDENTITY_PRESETS):
-                                            label, text = IDENTITY_PRESETS[i + j]
-                                            btn = gr.Button(label, size="sm", min_width=120)
-                                            btn.click(fn=make_preset_fn(text), outputs=[id_prompt])
-
                         id_candid = gr.Checkbox(
                             label="Candid realism mode (kills the AI gloss)",
                             value=True,
@@ -472,6 +566,16 @@ FLUX.1-dev has weaker safety training, so the NSFW LoRAs actually work here. Bes
                             lines=1,
                             info="Gets prepended to every prompt. The more specific, the more the output looks like your guy.",
                         )
+                        with gr.Group():
+                            gr.Markdown("**🚀 Quick Start — pick a scenario, everything gets set up for you**")
+                            with gr.Row():
+                                ex_scenario_dd = gr.Dropdown(
+                                    choices=list(EXPLICIT_SCENARIOS.keys()),
+                                    label="Scenario", value=None, scale=3,
+                                )
+                                ex_scenario_btn = gr.Button("Apply", variant="secondary", scale=1)
+                            ex_scenario_hint = gr.Markdown("")
+
                         ex_prompt = gr.Textbox(
                             label="Explicit Prompt",
                             placeholder="same person, fully nude, anatomically correct...",
@@ -481,15 +585,6 @@ FLUX.1-dev has weaker safety training, so the NSFW LoRAs actually work here. Bes
                             ex_gen_btn = gr.Button("Generate", variant="primary", scale=2)
                             ex_gen_add_btn = gr.Button("Generate & Add", variant="secondary", scale=2)
 
-                        with gr.Accordion("Presets", open=True):
-                            for i in range(0, len(EXPLICIT_PRESETS), 2):
-                                with gr.Row():
-                                    for j in range(2):
-                                        if i + j < len(EXPLICIT_PRESETS):
-                                            label, text = EXPLICIT_PRESETS[i + j]
-                                            btn = gr.Button(label, size="sm", min_width=120)
-                                            btn.click(fn=make_preset_fn(text), outputs=[ex_prompt])
-
                         with gr.Accordion("Advanced", open=False):
                             ex_seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
                             ex_randomize = gr.Checkbox(label="Randomize seed", value=True)
@@ -551,6 +646,18 @@ FLUX.1-dev has weaker safety training, so the NSFW LoRAs actually work here. Bes
             "Train your character LoRAs here, animate them there.</p>"
         )
 
+    # --- Quick Start scenario events ---
+    id_scenario_btn.click(
+        fn=apply_identity_scenario,
+        inputs=[id_scenario_dd],
+        outputs=[id_prompt, id_candid, id_nsfw, id_realism, id_guidance, id_steps, id_scenario_hint],
+    )
+    ex_scenario_btn.click(
+        fn=apply_explicit_scenario,
+        inputs=[ex_scenario_dd],
+        outputs=[ex_prompt, ex_strength, ex_nsfw, ex_realism, ex_guidance, ex_steps, ex_scenario_hint],
+    )
+
     # --- Identity tab events ---
     id_gen_btn.click(
         fn=generate_identity,