Expand route simulation coverage
This commit is contained in:
+9
-1
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user