Ethanfel/ComfyUI-Ethanfel-Prompt-Builder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clean atlas Krea prompt output

Browse Source
This commit is contained in:
Ethanfel
2026-06-30 23:42:20 +02:00
parent caeafa0714
commit f8f2fb43df
5 changed files with 68 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/prompt_smoke.py
+24 -3
View File
@@ -6762,6 +6762,12 @@ def smoke_krea2_pov_pose_variant_catalog() -> None:
refs = variant.get("reference_images")
_expect(isinstance(prompt_cues, list) and prompt_cues, f"{key} has no prompt cues")
_expect(isinstance(avoid_cues, list) and avoid_cues, f"{key} has no avoid cues")
for cue in prompt_cues:
cue_text = _expect_text(f"{key}.prompt_cue", cue, 8)
_expect(
not re.search(r"\b(?:may|optionally|either|or)\b", cue_text, flags=re.IGNORECASE),
f"{key} prompt cue should be a direct model instruction, not an option list: {cue_text!r}",
)
_expect(isinstance(refs, list) and refs, f"{key} has no reference images")
hook = variant.get("generator_hook") or {}
_expect_text(f"{key}.generator_hook.module", hook.get("module"), 6)
@@ -6844,10 +6850,25 @@ def smoke_krea2_pov_atlas_variant_prompt_routes() -> None:
cue_text = _expect_text(f"{key}.prompt_cue", cue, 8).lower()
_expect(cue_text in prompt, f"{key} final Krea prompt lost atlas cue {cue_text!r}: {prompt}")
atlas_action_prompt = prompt.split(" camera is ", 1)[0]
_expect(
not re.search(r"\b(?:may|optionally|either|or)\b", atlas_action_prompt, flags=re.IGNORECASE),
f"{key} final Krea prompt kept optional atlas action wording: {prompt}",
)
_expect(
"; " not in atlas_action_prompt,
f"{key} final Krea prompt kept semicolon-delimited atlas cue formatting: {prompt}",
)
_expect(
"camera layout" not in prompt
and "keep the visible partner" not in prompt
and "context stays beside" not in prompt
and "pov body or hand cues stay" not in prompt,
f"{key} final Krea prompt kept generic camera-scene layout prose: {prompt}",
)
_expect(
"hands or body cues" not in prompt and "perspective or foreground cues" not in prompt,
f"{key} final Krea prompt kept option wording in POV camera phrase: {prompt}",
)
_expect(
"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}",
@@ -7016,7 +7037,7 @@ def smoke_krea2_pose_variant_catalog_policy() -> None:
ready = krea2_pose_variant_catalog.get_variant("pov_ejaculation_aftermath_open_thigh_candidate")
_expect(ready.get("status") == "candidate", "Ready aftermath variant should remain a candidate until fixed-seed evidence exists")
_expect(
any("thick semen or fluid is visible around the exposed pussy or anal opening" in str(cue) for cue in ready.get("prompt_cues", [])),
any("thick semen and clear fluid are visible around the exposed pussy and anal opening" in str(cue) for cue in ready.get("prompt_cues", [])),
"Ready aftermath variant lost explicit post-ejaculation fluid/opening cue",
)
_expect(
@@ -7038,7 +7059,7 @@ def smoke_krea2_pose_variant_catalog_policy() -> None:
spread = krea2_pose_variant_catalog.get_variant("pov_spread_open_thigh_presentation")
_expect(spread.get("status") == "candidate", "Spread variant should remain a candidate until fixed-seed evidence exists")
_expect(
any("legs raised or knees held wide" in str(cue) for cue in spread.get("prompt_cues", [])),
any("legs raised and knees held wide" in str(cue) for cue in spread.get("prompt_cues", [])),
"Spread variant lost open-thigh cue",
)
oral_top = krea2_pose_variant_catalog.get_variant("pov_blowjob_top_down_vertical_shaft")
@@ -7158,7 +7179,7 @@ def smoke_krea2_pose_variant_catalog_policy() -> None:
"Missionary variant lost open-leg reclined cue",
)
_expect(
any("viewer stands or braces at the foot edge" in str(cue) for cue in missionary.get("prompt_cues", [])),
any("viewer braced at the foot edge" in str(cue) for cue in missionary.get("prompt_cues", [])),
"Missionary variant lost elevated-support edge cue",
)
missionary_evidence = missionary.get("evidence") or {}