Compact exact Krea2 atlas prompts

This commit is contained in:
2026-06-30 21:37:39 +02:00
parent 4689cc7942
commit ff484aa27c
5 changed files with 40 additions and 9 deletions
+25
View File
@@ -6847,9 +6847,34 @@ def smoke_krea2_pov_atlas_variant_prompt_routes() -> None:
"framed as " not in prompt and "the image is framed as " not in prompt,
f"{key} final Krea prompt kept generic composition text after atlas route: {prompt}",
)
for forbidden in (
"pov foreground clothing cue",
"body is fully exposed",
"explicit consensual",
"use clear adult anatomy",
):
_expect(forbidden not in prompt, f"{key} final Krea prompt kept generic formatter layer {forbidden!r}: {prompt}")
atlas_geometry = " ".join([str(variant.get("canonical_geometry") or ""), *[str(cue) for cue in variant.get("prompt_cues") or []]]).lower()
if any(term in atlas_geometry for term in ("top-down", "top view", "top-view", "nadir", "overhead")):
_expect("eye-level shot" not in prompt, f"{key} final Krea prompt kept contradictory eye-level camera text: {prompt}")
if key == "pov_blowjob_top_down_vertical_shaft":
_expect(
prompt.count("nadir-angle standing male pov top-view oral") == 1,
f"{key} final Krea prompt repeated top-view atlas header instead of staying compact: {prompt}",
)
for forbidden in (
"the woman takes the viewer's penis",
"visible saliva",
"cum on",
"hands holding hips",
"eyes looking up",
):
_expect(forbidden not in prompt, f"{key} final Krea prompt kept generic detail fragment {forbidden!r}: {prompt}")
for required in (
"mouth seals around the centered shaft",
"one hand wraps the base",
):
_expect(required in prompt, f"{key} final Krea prompt lost compact contact cue {required!r}: {prompt}")
for avoid in variant.get("avoid_cues") or []:
avoid_text = _expect_text(f"{key}.avoid_cue", avoid, 4).lower()
_expect(avoid_text not in prompt, f"{key} final Krea prompt leaked avoid cue {avoid_text!r}: {prompt}")