Expand route simulation coverage

This commit is contained in:
2026-06-27 18:49:01 +02:00
parent cac4fe47cd
commit f91953f12b
3 changed files with 215 additions and 15 deletions
+9 -1
View File
@@ -96,6 +96,14 @@ def _row_explicit_signal_text(
return " ".join(deps.clean(value) for value in values if deps.clean(value))
def _uses_hardcore_action_route(row: dict[str, Any]) -> bool:
return (
str(row.get("category_slug") or "").strip() == "hardcore_sexual_poses"
or bool(str(row.get("action_family") or "").strip())
or bool(str(row.get("position_family") or "").strip())
)
def row_core_tags_result(request: SDXLRowTagRequest, deps: SDXLTagRouteDependencies) -> SDXLTagRoute:
row = request.row
tags: list[str] = []
@@ -117,7 +125,7 @@ def row_core_tags_result(request: SDXLRowTagRequest, deps: SDXLTagRouteDependenc
item = deps.row_value(row, "item", ("Sexual scene", "Sexual pose", "Erotic outfit", "Clothing")) or deps.clean(
row.get("custom_item")
)
pose = deps.row_value(row, "pose", ("Sexual pose", "Pose"))
pose = "" if _uses_hardcore_action_route(row) else deps.row_value(row, "pose", ("Sexual pose", "Pose"))
role_graph = deps.clean(row.get("source_role_graph") or row.get("role_graph"))
scene = deps.row_value(row, "scene_text", ("Setting", "Scene")) or deps.clean(row.get("scene"))
expression = deps.row_value(row, "character_expression_text") or deps.row_value(