"""Canonical reader-facing names for public task-suite artifacts."""

from __future__ import annotations

from typing import Any


TASK_DISPLAY_NAMES = {
    "timeline_action": "Action Recognition",
    "timeline_subtask": "Procedure Step Recognition",
    "transition_detection": "Action Boundary Detection",
    "next_action": "Next-Action Prediction",
    "hand_trajectory_forecast": "Hand Trajectory Forecasting",
    "contact_prediction": "Contact State Prediction",
    "object_relevance": "Object Relevance Prediction",
    "caption_grounding": "Language Grounding",
    "cross_modal_retrieval": "Cross-Modal Retrieval",
    "modality_reconstruction": "Cross-Modal Reconstruction",
    "temporal_order": "Temporal Order Verification",
    "misalignment_detection": "Multimodal Synchronization Detection",
}


def task_display_name(task_id: str) -> str:
    """Return the public label while preserving unknown ids for debugging."""
    return TASK_DISPLAY_NAMES.get(task_id, task_id.replace("_", " ").title())


def with_task_display(task_id: str, payload: dict[str, Any]) -> dict[str, Any]:
    payload.setdefault("task", task_id)
    payload["task_display_name"] = task_display_name(task_id)
    return payload