"""Frontend metadata for the Motion Transfer task."""
from __future__ import annotations

from typing import Any, Dict

from src.tasks.base import TaskPlugin


def validate_scene(_scene_dir: str) -> None:
    return None


def evaluate_scene(_scene_dir: str, _gt_scene: Any) -> Dict[str, float]:
    return {"psnr": 0.0, "ssim": 0.0, "lpips": 0.0, "pmf": 0.0}


def load_gt_scene(scene_id: str, gt_dir: str) -> Any:
    return {"scene_id": scene_id, "gt_dir": gt_dir}


TASK = TaskPlugin(
    name="motionTransfer",
    track_id="motionTransfer",
    display_name="Motion Transfer",
    description=(
        "Transfer motion dynamics from a source video to a target image while preserving "
        "the target's visual appearance. Evaluated with PSNR, SSIM, LPIPS, and PMF."
    ),
    expected_scene_layout=(
        "```\n"
        "<scene_id>.zip\n"
        "`-- CineCamera_n/\n"
        "    `-- rgb/\n"
        "        |-- 0000.jpg\n"
        "        |-- ...\n"
        "        `-- <last_frame>.jpg\n"
        "```"
    ),
    validate_scene_fn=validate_scene,
    evaluate_scene_fn=evaluate_scene,
    load_gt_scene_fn=load_gt_scene,
    primary_metric="pmf",
    higher_is_better=True,
    leaderboard_columns=["psnr", "ssim", "lpips", "pmf"],
)

TASK.column_display_names = {
    "psnr": "PSNR",
    "ssim": "SSIM",
    "lpips": "LPIPS",
    "pmf": "PMF",
}

TASKS = {
    TASK.name: TASK,
}