import pandas as pd
import gradio as gr

df = pd.DataFrame(
    {
        "A": [14, 4, 5, 4, 1],
        "B": [5, 2, 54, 3, 2],
        "C": [20, 20, 7, 3, 8],
        "D": [14, 3, 6, 2, 6],
        "E": [23, 45, 64, 32, 23],
    }
)

t = df.style.highlight_max(color="lightgreen", axis=0)

models = ["OpenAI GPT-3", "GPT-2", "Bert", "GPT-3", "T5"]
orgs = ["OpenAI", "Anthropic", "Google", "Mistral", "Microsoft"]
licenses = [
    "OpenAI License",
    "Apache License",
    "Proprietary",
    "Llama License",
    "MIT License",
    "CC-BY-NC",
]
# function ot get random number between a range
import random


def get_random_number(min_value, max_value):
    """Get a random number between min_value and max_value."""
    return random.randint(min_value, max_value)


table_vals = [
    [
        i + 1,
        f"[{models[get_random_number(0, 4)]}](https://www.google.com)",
        get_random_number(2000, 3000),
        f"+{get_random_number(1, 10)} / -{get_random_number(1, 10)}",
        get_random_number(3000, 20000),
        orgs[get_random_number(0, 4)],
        licenses[get_random_number(0, 5)],
    ]
    for i in range(60)
]


with gr.Blocks() as demo:
    # with gr.Tab():
    #     gr.Dataframe(
    #         headers=["A", "B", "C", "D", "E"],
    #         datatype=["number", "number", "number", "number", "number"],
    #         value=df,
    #         elem_id="dataframe",
    #         height=700,
    #         column_widths=[100, 100, 100, 100, 100],
    #         wrap=True,
    #     )
    # with gr.Tab():
    gr.Dataframe(
        headers=[
            "Rank",
            "🤖 Model",
            "⭐ Arena Elo",
            "📊 95% CI",
            "🗳️ Votes",
            "Organization",
            "License",
        ],
        datatype=[
            "str",
            "markdown",
            "number",
            "str",
            "number",
            "str",
            "str",
        ],
        value=table_vals,
        elem_id="arena_leaderboard_dataframe",
        height=700,
        column_widths=[50, 200, 100, 100, 100, 150, 150],
        wrap=True,
    )

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