Add men and couple casual outfit categories

caption_naturalizer.py

@@ -356,6 +356,18 @@ def possessive_pronoun(subject: str) -> str:
     return "Her" if subject == "woman" else "His"
 
 
+def _couple_clothing_sentence(clothing: str) -> str:
+    clothing = _clean_text(clothing)
+    lower = clothing.lower()
+    partner_text = re.sub(r"\bPartner ([AB]) wears\b", r"Partner \1 wearing", clothing)
+    partner_text = re.sub(r"\bPartner ([AB]) has\b", r"Partner \1 with", partner_text)
+    if lower.startswith("partner a "):
+        return f"The outfits show {partner_text}"
+    if lower.startswith(("two ", "paired ", "coordinated ")):
+        return f"The outfits are {partner_text}"
+    return f"They wear {clothing}"
+
+
 def _couple_from_row(row: dict[str, Any], detail_level: str, keep_style: bool) -> tuple[str, str] | None:
     subject = _clean_text(row.get("subject_phrase") or row.get("primary_subject"))
     primary = _clean_text(row.get("primary_subject"))
@@ -381,7 +393,7 @@ def _couple_from_row(row: dict[str, Any], detail_level: str, keep_style: bool) -
     if body:
         parts.append(f"Their body types are {body}")
     if clothing:
-        parts.append(f"They wear {clothing}")
+        parts.append(_couple_clothing_sentence(clothing))
     if pose:
         parts.append(f"The pose is {pose}")
     if scene:
@@ -396,7 +408,7 @@ def _couple_from_row(row: dict[str, Any], detail_level: str, keep_style: bool) -
 
 
 def _configured_cast_from_row(row: dict[str, Any], detail_level: str, keep_style: bool) -> tuple[str, str] | None:
-    if _clean_text(row.get("subject_type")) != "configured_cast" and not _clean_text(row.get("women_count")):
+    if _clean_text(row.get("subject_type")) != "configured_cast":
         if "hardcore sexual poses" not in _clean_text(row.get("main_category")).lower():
             return None