Add caption naturalizer profiles

2026-06-27 01:43:48 +02:00
parent 36ce394462
commit 21da2949c6
8 changed files with 101 additions and 4 deletions
@@ -627,10 +627,16 @@ def naturalize_caption(
    include_trigger: bool = True,
    detail_level: str = "balanced",
    style_policy: str = "drop_style_tail",
+    caption_profile: str = caption_policy.CAPTION_PROFILE_DEFAULT,
 ) -> tuple[str, str]:
    """Rewrite tag-style prompt/caption text into compact natural language."""
    input_hint = input_hint if input_hint in ("auto", "metadata_json", "caption_or_prompt") else "auto"
-    detail_level = caption_policy.normalize_detail_level(detail_level)
+    detail_level, style_policy, include_trigger = caption_policy.apply_caption_profile(
+        caption_profile,
+        detail_level=detail_level,
+        style_policy=style_policy,
+        include_trigger=include_trigger,
+    )
    keep_style = caption_policy.keep_style_terms(style_policy)
    row, row_method = _row_from_inputs(source_text, metadata_json, input_hint)
    if row is not None: