from dataclasses import dataclass
from enum import Enum

@dataclass
class Task:
    benchmark: str
    metric: str
    col_name: str


# Select your tasks here
# ---------------------------------------------------
class Tasks(Enum):
    # benchmark key in metric_results, value key inside each metric, display name
    asr_i2p = Task("asr_i2p", "value", "ASR (I2P)")
    asr_ring_a_bell = Task("asr_ring_a_bell", "value", "ASR (RingABell)")
    asr_mma_diffusion = Task("asr_mma_diffusion", "value", "ASR (MMA)")
    err = Task("err", "value", "ERR")
    fid = Task("fid", "value", "FID")
    clip_score = Task("clip_score", "value", "CLIP Score")
    ua_ira = Task("ua_ira", "value", "UA-IRA")
    tifa = Task("tifa", "value", "TIFA")
# ---------------------------------------------------



# Your leaderboard name
TITLE = """<h1 align="center" id="space-title">Demo leaderboard</h1>"""

# What does your leaderboard evaluate?
INTRODUCTION_TEXT = """
Intro text
"""

CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
CITATION_BUTTON_TEXT = r"""
"""