import gradio as gr
import pandas as pd

dataset_files = {
    "XTREME Overview": "xtreme-overview-merged.csv",
}

def load_dataset(name):
    df = pd.read_csv(dataset_files[name])
    # df.reset_index(inplace=True)
    return df

with gr.Blocks() as demo:
    gr.Markdown("# TLP matrix: Languages x Tasks")
    gr.Markdown("Counts of dataset sizes per language and task")

    dataset_dropdown = gr.Dropdown(choices=list(dataset_files.keys()), label="Select dataset")

    # Set initial dataframe value here
    initial_dataset = list(dataset_files.keys())[0]
    initial_df = load_dataset(initial_dataset)
    dataframe = gr.Dataframe(value=initial_df, type="pandas", interactive=False)

    dataset_dropdown.change(fn=load_dataset, inputs=dataset_dropdown, outputs=dataframe)

demo.launch()