from dataclasses import dataclass
from enum import Enum

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


# Init: to update with your specific keys
class Tasks(Enum):
    # task_key in the json file, metric_key in the json file, name to display in the leaderboard
    task0 = Task("asr_eval1", "EN_LibriSpeech", "EN_LibriSpeech CER")
    task1 = Task("asr_eval2", "ML_SUPERB", "ML_SUPERB CER")
    task2 = Task("asr_eval3", "Bitrate", "Bitrate")


# Your leaderboard name
TITLE = """<h1 align="center" id="space-title">Discrete-Speech based ASR Leaderboard</h1>"""

# What does your leaderboard evaluate?
INTRODUCTION_TEXT = """
The leaderboard for discrete speech challenge (ASR Track) at Interspeech 2024. Challenge details can be found at https://www.wavlab.org/activities/2024/Interspeech2024-Discrete-Speech-Unit-Challenge/
"""

# Which evaluations are you running? how can people reproduce what you have?
LLM_BENCHMARKS_TEXT = f"""
## How it works

The evaluation (static version) are conducted by the organizers only.

We will accept submissions from the google form (see rules in the challenge website).

## Reproducibility
To reproduce our results, please refer to the baseline scripts at https://github.com/espnet/espnet/tree/master/egs2/interspeech2024_dsu_challenge/asr2

"""

EVALUATION_QUEUE_TEXT = """
"""

CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results and the challenge (coming soon, for now, please cite the challenge website)"
CITATION_BUTTON_TEXT = r"""
"""