Move pair cast styling policy

pair_cast.py

@@ -2,11 +2,75 @@ from __future__ import annotations
 
 from typing import Any, Callable
 
+try:
+    from . import pair_clothing
+    from . import pair_options
+except ImportError:  # Allows local smoke tests with top-level imports.
+    import pair_clothing
+    import pair_options
+
 
 DescriptorFromRow = Callable[[dict[str, Any]], str]
 CastDescriptors = Callable[..., list[str]]
-PartnerStyling = Callable[..., dict[str, Any]]
-CastPhrase = Callable[[int, int], str]
+AxisRng = Callable[[dict[str, int], str, int, int], Any]
+Choose = Callable[[Any, list[str]], str]
+SlotSoftcoreOutfit = Callable[[dict[str, Any] | None, Any], str]
+
+
+def cast_summary_phrase(women_count: int, men_count: int) -> str:
+    women_count = max(0, int(women_count))
+    men_count = max(0, int(men_count))
+    if women_count + men_count == 0:
+        women_count = 1
+    person_count = women_count + men_count
+    women_label = "woman" if women_count == 1 else "women"
+    men_label = "man" if men_count == 1 else "men"
+    return f"{women_count} {women_label}, {men_count} {men_label}, {person_count} total adults"
+
+
+def softcore_partner_styling(
+    *,
+    seed_config: dict[str, int],
+    seed: int,
+    row_number: int,
+    women_count: int,
+    men_count: int,
+    pov_labels: list[str] | None,
+    label_map: dict[str, dict[str, Any]] | None,
+    axis_rng: AxisRng,
+    choose: Choose,
+    slot_softcore_outfit: SlotSoftcoreOutfit,
+) -> dict[str, Any]:
+    content_rng = axis_rng(seed_config, "content", seed, row_number + 421)
+    pose_rng = axis_rng(seed_config, "pose", seed, row_number + 421)
+    pov_set = set(pov_labels or [])
+    outfits: list[str] = []
+    for index in range(max(0, women_count - 1)):
+        label = chr(ord("B") + index)
+        full_label = f"Woman {label}"
+        outfit = slot_softcore_outfit((label_map or {}).get(full_label), content_rng) or choose(
+            content_rng,
+            pair_options.INSTA_OF_SOFTCORE_PARTNER_WOMEN_OUTFITS,
+        )
+        sentence = pair_clothing.softcore_outfit_sentence(full_label, outfit)
+        if sentence:
+            outfits.append(sentence)
+    for index in range(max(0, men_count)):
+        label = chr(ord("A") + index)
+        full_label = f"Man {label}"
+        if full_label in pov_set:
+            continue
+        outfit = slot_softcore_outfit((label_map or {}).get(full_label), content_rng) or choose(
+            content_rng,
+            pair_options.INSTA_OF_SOFTCORE_PARTNER_MEN_OUTFITS,
+        )
+        sentence = pair_clothing.softcore_outfit_sentence(full_label, outfit)
+        if sentence:
+            outfits.append(sentence)
+    return {
+        "outfits": outfits,
+        "pose": choose(pose_rng, pair_options.SOFTCORE_CAST_POSES),
+    }
 
 
 def resolve_insta_pair_cast_context(
@@ -31,8 +95,9 @@ def resolve_insta_pair_cast_context(
     descriptor_from_row: DescriptorFromRow,
     build_cast_descriptors: CastDescriptors,
     prompt_cast_descriptors: Callable[[str], str],
-    partner_styling: PartnerStyling,
-    cast_phrase: CastPhrase,
+    axis_rng: AxisRng,
+    choose: Choose,
+    slot_softcore_outfit: SlotSoftcoreOutfit,
 ) -> dict[str, Any]:
     descriptor = descriptor_from_row(soft_row)
     cast_descriptors = build_cast_descriptors(
@@ -51,14 +116,17 @@ def resolve_insta_pair_cast_context(
     cast_descriptor_text = prompt_cast_descriptors("; ".join(cast_descriptors))
     same_softcore_cast = options["softcore_cast"] == "same_as_hardcore"
     soft_cast_descriptor_text = cast_descriptor_text if same_softcore_cast else f"Woman A: {descriptor}"
-    soft_partner_styling = partner_styling(
-        parsed_seed_config,
-        seed,
-        row_number,
-        hard_women_count if same_softcore_cast else 1,
-        hard_men_count if same_softcore_cast else 0,
-        pov_character_labels if same_softcore_cast else [],
-        character_slot_map,
+    soft_partner_styling = softcore_partner_styling(
+        seed_config=parsed_seed_config,
+        seed=seed,
+        row_number=row_number,
+        women_count=hard_women_count if same_softcore_cast else 1,
+        men_count=hard_men_count if same_softcore_cast else 0,
+        pov_labels=pov_character_labels if same_softcore_cast else [],
+        label_map=character_slot_map,
+        axis_rng=axis_rng,
+        choose=choose,
+        slot_softcore_outfit=slot_softcore_outfit,
     )
     if not same_softcore_cast:
         soft_partner_styling = {"outfits": [], "pose": ""}
@@ -67,7 +135,7 @@ def resolve_insta_pair_cast_context(
     soft_cast = (
         "solo creator setup with Woman A alone"
         if options["softcore_cast"] == "solo"
-        else f"soft creator-teaser setup with {cast_phrase(hard_women_count, hard_men_count)}"
+        else f"soft creator-teaser setup with {cast_summary_phrase(hard_women_count, hard_men_count)}"
     )
     soft_cast_presence = (
         (
@@ -86,7 +154,7 @@ def resolve_insta_pair_cast_context(
         if same_softcore_cast and soft_partner_outfit_text
         else ""
     )
-    hard_cast = cast_phrase(hard_women_count, hard_men_count)
+    hard_cast = cast_summary_phrase(hard_women_count, hard_men_count)
     soft_descriptor_sentence = (
         f"Cast descriptors: {soft_cast_descriptor_text}. "
         if same_softcore_cast