"""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",
    "long_horizon_next_action": "Long-Horizon Next-Action Forecasting",
    "next_subtask_forecast": "Long-Horizon Next-Subtask Forecasting",
    "interaction_text_prediction": "Interaction Text Prediction",
    "action_object_relation": "Action-Object Relation Prediction",
    "object_set_forecast": "Future Object-Set Forecasting",
    "imu_to_hand_pose": "IMU-to-Hand Pose Reconstruction",
    "camera_view_sync_retrieval": "Camera-View Synchronization Retrieval",
    "time_to_transition": "Time-to-Next-Transition Regression",
}


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