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(15)
]


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",
                # "str",
                # "number",
                # "str",
                # "number",
                # "str",
                # "str",
            ],
            value=table_vals,
            elem_id="arena_leaderboard_dataframe",
            height=500,
            column_widths=[
                50
                # , 200, 100, 100, 100, 150, 150
            ],
            wrap=True,
        )

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