98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Any, Callable
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class CaptionFormatRequest:
|
|
source_text: str
|
|
metadata_json: str = ""
|
|
input_hint: str = "auto"
|
|
trigger: str = ""
|
|
include_trigger: bool = True
|
|
detail_level: str = "balanced"
|
|
style_policy: str = "drop_style_tail"
|
|
caption_profile: str = "manual_controls"
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class CaptionFormatRoute:
|
|
caption: str
|
|
method: str
|
|
branch: str
|
|
input_hint: str
|
|
detail_level: str
|
|
style_policy: str
|
|
include_trigger: bool
|
|
keep_style: bool
|
|
|
|
def as_tuple(self) -> tuple[str, str]:
|
|
return self.caption, self.method
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class CaptionFormatDependencies:
|
|
apply_caption_profile: Callable[[str, str, str, bool], tuple[str, str, bool]]
|
|
keep_style_terms: Callable[[str], bool]
|
|
row_from_inputs: Callable[[str, str, str], tuple[dict[str, Any] | None, str]]
|
|
metadata_to_prose: Callable[[dict[str, Any], str, bool], tuple[str, str]]
|
|
text_to_prose: Callable[[str, str, bool], tuple[str, str]]
|
|
with_trigger: Callable[[str, str, bool], str]
|
|
sanitize_prose_text: Callable[..., str]
|
|
|
|
|
|
def naturalize_caption_result(
|
|
request: CaptionFormatRequest,
|
|
deps: CaptionFormatDependencies,
|
|
) -> CaptionFormatRoute:
|
|
input_hint = request.input_hint if request.input_hint in ("auto", "metadata_json", "caption_or_prompt") else "auto"
|
|
detail_level, style_policy, include_trigger = deps.apply_caption_profile(
|
|
request.caption_profile,
|
|
request.detail_level,
|
|
request.style_policy,
|
|
request.include_trigger,
|
|
)
|
|
keep_style = deps.keep_style_terms(style_policy)
|
|
row, row_method = deps.row_from_inputs(request.source_text, request.metadata_json, input_hint)
|
|
if row is not None:
|
|
prose, method = deps.metadata_to_prose(row, detail_level, keep_style)
|
|
caption = deps.sanitize_prose_text(
|
|
deps.with_trigger(prose, request.trigger, include_trigger),
|
|
triggers=(request.trigger,),
|
|
)
|
|
full_method = f"{row_method}:{method}"
|
|
return CaptionFormatRoute(
|
|
caption=caption,
|
|
method=full_method,
|
|
branch="metadata",
|
|
input_hint=input_hint,
|
|
detail_level=detail_level,
|
|
style_policy=style_policy,
|
|
include_trigger=include_trigger,
|
|
keep_style=keep_style,
|
|
)
|
|
|
|
prose, method = deps.text_to_prose(request.source_text, detail_level, keep_style)
|
|
caption = deps.sanitize_prose_text(
|
|
deps.with_trigger(prose, request.trigger, include_trigger),
|
|
triggers=(request.trigger,),
|
|
)
|
|
return CaptionFormatRoute(
|
|
caption=caption,
|
|
method=method,
|
|
branch="text",
|
|
input_hint=input_hint,
|
|
detail_level=detail_level,
|
|
style_policy=style_policy,
|
|
include_trigger=include_trigger,
|
|
keep_style=keep_style,
|
|
)
|
|
|
|
|
|
def naturalize_caption(
|
|
request: CaptionFormatRequest,
|
|
deps: CaptionFormatDependencies,
|
|
) -> tuple[str, str]:
|
|
return naturalize_caption_result(request, deps).as_tuple()
|