import random
from pathlib import Path

import gradio as gr
import pandas as pd
from gradio_leaderboard import Leaderboard, SelectColumns, SearchColumns

abs_path = Path(__file__).parent

df = pd.read_csv("opensci-ref-table.csv")

df.drop("#Tokens", axis=1, inplace=True)
df.drop("AVG", axis=1, inplace=True)

benchmarks = df.columns[1:]
df["Average ⬆️"] = df.loc[:, benchmarks].mean(axis=1)
# df.set_index("Model", inplace=True)

with gr.Blocks() as demo:
    gr.Markdown(
        """
    # 🥇 OpenEuroLLM Leaderboard 🇪🇺
    """
    )
    Leaderboard(
        value=df.round(2),
        select_columns=SelectColumns(
            default_selection=list(df.columns),
            cant_deselect=["Model"],
            label="Select Columns to Display:",
        ),
        search_columns=SearchColumns(
            primary_column="Model",
            label="Filter a model",
            secondary_columns=[],
        ),
    )

if __name__ == "__main__":
    demo.launch()