from datasets import load_dataset
import gradio as gr

# all levels, validation split (public)
ds_val = load_dataset(
    "gaia-benchmark/GAIA",
    "2023_all",
    split="validation",
    token=True,                 # use_auth_token=True on older installs also works
    trust_remote_code=True,     # needed because GAIA uses a loading script
)

ds_val.set_format(type="pandas")
df_val = ds_val[:]

temp_id = df_val.at[0, 'task_id']

with gr.Blocks(title="Display task_id") as demo:
    gr.Markdown( "Task_id: {}".format(temp_id))

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