Extract generation profile config policy

prompt_builder.py

@@ -26,6 +26,7 @@ try:
     from . import camera_config as camera_policy
     from . import category_cast_config as category_cast_policy
     from . import generate_prompt_batches as g
+    from . import generation_profile_config as generation_profile_policy
     from . import location_config as location_policy
     from . import pair_clothing
     from . import pair_camera
@@ -65,6 +66,7 @@ except ImportError:  # Allows local smoke tests with `python -c`.
     import camera_config as camera_policy
     import category_cast_config as category_cast_policy
     import generate_prompt_batches as g
+    import generation_profile_config as generation_profile_policy
     import location_config as location_policy
     import pair_clothing
     import pair_camera
@@ -1035,74 +1037,7 @@ def seed_mode_choices() -> list[str]:
 CATEGORY_PRESETS = category_cast_policy.CATEGORY_PRESETS
 CAST_PRESETS = category_cast_policy.CAST_PRESETS
 
-GENERATION_PROFILE_PRESETS = {
-    "balanced": {
-        "clothing": "full",
-        "poses": "standard",
-        "expression_enabled": True,
-        "expression_intensity": 0.5,
-        "backside_bias": 0.0,
-        "minimal_clothing_ratio": -1.0,
-        "standard_pose_ratio": -1.0,
-        "trigger": "sxcpinup_coloredpencil",
-        "prepend_trigger_to_prompt": True,
-    },
-    "casual_clean": {
-        "clothing": "full",
-        "poses": "standard",
-        "expression_enabled": True,
-        "expression_intensity": 0.35,
-        "backside_bias": 0.0,
-        "minimal_clothing_ratio": -1.0,
-        "standard_pose_ratio": -1.0,
-        "trigger": "sxcpinup_coloredpencil",
-        "prepend_trigger_to_prompt": True,
-    },
-    "evocative_softcore": {
-        "clothing": "minimal",
-        "poses": "evocative",
-        "expression_enabled": True,
-        "expression_intensity": 0.65,
-        "backside_bias": 0.2,
-        "minimal_clothing_ratio": -1.0,
-        "standard_pose_ratio": -1.0,
-        "trigger": "sxcpinup_coloredpencil",
-        "prepend_trigger_to_prompt": True,
-    },
-    "hardcore_intense": {
-        "clothing": "minimal",
-        "poses": "evocative",
-        "expression_enabled": True,
-        "expression_intensity": 0.9,
-        "backside_bias": 0.0,
-        "minimal_clothing_ratio": -1.0,
-        "standard_pose_ratio": -1.0,
-        "trigger": "sxcpinup_coloredpencil",
-        "prepend_trigger_to_prompt": True,
-    },
-    "krea2_friendly": {
-        "clothing": "full",
-        "poses": "standard",
-        "expression_enabled": True,
-        "expression_intensity": 0.55,
-        "backside_bias": 0.0,
-        "minimal_clothing_ratio": -1.0,
-        "standard_pose_ratio": -1.0,
-        "trigger": "sxcpinup_coloredpencil",
-        "prepend_trigger_to_prompt": False,
-    },
-    "flux_original": {
-        "clothing": "full",
-        "poses": "standard",
-        "expression_enabled": True,
-        "expression_intensity": 0.5,
-        "backside_bias": 0.0,
-        "minimal_clothing_ratio": -1.0,
-        "standard_pose_ratio": -1.0,
-        "trigger": "sxcpinup_coloredpencil",
-        "prepend_trigger_to_prompt": True,
-    },
-}
+GENERATION_PROFILE_PRESETS = generation_profile_policy.GENERATION_PROFILE_PRESETS
 
 
 def category_preset_choices() -> list[str]:
@@ -1114,7 +1049,7 @@ def cast_preset_choices() -> list[str]:
 
 
 def generation_profile_choices() -> list[str]:
-    return list(GENERATION_PROFILE_PRESETS)
+    return generation_profile_policy.generation_profile_choices()
 
 
 def build_category_config_json(preset: str = "auto_weighted", subcategory: str = RANDOM_SUBCATEGORY) -> str:
@@ -1145,60 +1080,22 @@ def build_generation_profile_json(
     trigger_policy: str = "profile_default",
     expression_enabled: bool = True,
 ) -> str:
-    profile = profile if profile in GENERATION_PROFILE_PRESETS else "balanced"
-    config = dict(GENERATION_PROFILE_PRESETS[profile])
-    if clothing_override in ("full", "minimal", "random"):
-        config["clothing"] = clothing_override
-    if poses_override in ("standard", "evocative", "random"):
-        config["poses"] = poses_override
-    config["expression_enabled"] = not _is_false(expression_enabled)
-    if expression_intensity_mode == "random":
-        config["expression_intensity"] = -1.0
-    elif expression_intensity_mode == "fixed" and float(expression_intensity) >= 0:
-        config["expression_intensity"] = _clamped_float(expression_intensity, config["expression_intensity"])
-    if float(backside_bias) >= 0:
-        config["backside_bias"] = _clamped_float(backside_bias, config["backside_bias"])
-    if float(minimal_clothing_ratio) >= 0:
-        config["minimal_clothing_ratio"] = _clamped_float(minimal_clothing_ratio, config["minimal_clothing_ratio"])
-    if float(standard_pose_ratio) >= 0:
-        config["standard_pose_ratio"] = _clamped_float(standard_pose_ratio, config["standard_pose_ratio"])
-    if trigger_policy == "prepend_trigger":
-        config["prepend_trigger_to_prompt"] = True
-    elif trigger_policy == "do_not_prepend":
-        config["prepend_trigger_to_prompt"] = False
-    config["profile"] = profile
-    return json.dumps(config, ensure_ascii=True, sort_keys=True)
+    return generation_profile_policy.build_generation_profile_json(
+        profile=profile,
+        clothing_override=clothing_override,
+        poses_override=poses_override,
+        expression_intensity_mode=expression_intensity_mode,
+        expression_intensity=expression_intensity,
+        backside_bias=backside_bias,
+        minimal_clothing_ratio=minimal_clothing_ratio,
+        standard_pose_ratio=standard_pose_ratio,
+        trigger_policy=trigger_policy,
+        expression_enabled=expression_enabled,
+    )
 
 
 def _parse_generation_profile(profile_config: str | dict[str, Any] | None) -> dict[str, Any]:
-    if not profile_config:
-        return dict(GENERATION_PROFILE_PRESETS["balanced"])
-    if isinstance(profile_config, dict):
-        raw = profile_config
-    else:
-        try:
-            raw = json.loads(str(profile_config))
-        except json.JSONDecodeError as exc:
-            raise ValueError(f"Invalid generation_profile JSON: {exc}") from exc
-    if not isinstance(raw, dict):
-        raise ValueError("generation_profile must be a JSON object")
-    profile = str(raw.get("profile") or "balanced")
-    parsed = dict(GENERATION_PROFILE_PRESETS.get(profile, GENERATION_PROFILE_PRESETS["balanced"]))
-    parsed.update(raw)
-    parsed["clothing"] = parsed["clothing"] if parsed.get("clothing") in ("full", "minimal", "random") else "full"
-    parsed["poses"] = parsed["poses"] if parsed.get("poses") in ("standard", "evocative", "random") else "standard"
-    parsed["expression_enabled"] = not _is_false(parsed.get("expression_enabled", True))
-    try:
-        raw_expression_intensity = float(parsed.get("expression_intensity"))
-    except (TypeError, ValueError):
-        raw_expression_intensity = 0.5
-    parsed["expression_intensity"] = -1.0 if raw_expression_intensity < 0 else _clamped_float(raw_expression_intensity, 0.5)
-    parsed["backside_bias"] = _clamped_float(parsed.get("backside_bias"), 0.0)
-    parsed["minimal_clothing_ratio"] = _clamped_float(parsed.get("minimal_clothing_ratio"), -1.0, -1.0, 1.0)
-    parsed["standard_pose_ratio"] = _clamped_float(parsed.get("standard_pose_ratio"), -1.0, -1.0, 1.0)
-    parsed["trigger"] = str(parsed.get("trigger") or "sxcpinup_coloredpencil")
-    parsed["prepend_trigger_to_prompt"] = bool(parsed.get("prepend_trigger_to_prompt"))
-    return parsed
+    return generation_profile_policy.parse_generation_profile(profile_config)
 
 
 def build_filter_config_json(