#!/usr/bin/env python
# -*- coding: utf-8 -*-
# flake8: noqa E501

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):
    # task_key, metric_key, title
    task00 = Task("naive_judge", "score", "NaïveJudge")
    task01 = Task("human_eval_solidity_pass_1", "score", "HumanEval for Solidity (pass@1)")
    task02 = Task("human_eval_solidity_pass_3", "score", "HumanEval for Solidity (pass@3)")
    task03 = Task("rouge1", "score", "ROUGE-unigrams")
    task04 = Task("rouge2", "score", "ROUGE-bigrams")
    task05 = Task("rougeL", "score", "ROUGE-Longest Common Subsequence")
    task06 = Task("rougeLsum", "score", "ROUGE-Lsum")
    task07 = Task("bleu", "score", "Bleu")
    task08 = Task("brevity_penalty", "score", "Brevity Penalty")
# ---------------------------------------------------

# Your leaderboard name
TITLE = """<br><img src="file/images/solbench.svg" width="500"  style="display: block; margin-left: auto; margin-right: auto;">
<h2 align="center" id="space-title">Solidity Leaderboard | Powered by IQ</h2>"""

# What does your leaderboard evaluate?
INTRODUCTION_TEXT = ""

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

## Reproducibility
To reproduce our results, here is the commands you can run:

"""

EVALUATION_REQUESTS_TEXT = """
## Make sure you can load your model and tokenizer using AutoClasses

```python
from transformers import AutoConfig, AutoModel, AutoTokenizer
config = AutoConfig.from_pretrained("your model name", revision=revision)
model = AutoModel.from_pretrained("your model name", revision=revision)
tokenizer = AutoTokenizer.from_pretrained("your model name", revision=revision)
```
If this step fails, follow the error messages to debug your model before submitting it. It's likely your model has been improperly uploaded.

Note: make sure your model is public.
"""
EVALUATION_SCRIPT = ''
CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
CITATION_BUTTON_TEXT = ''