app.py

import gradio as gr
import pandas as pd
from utils import load_all_results


def sort_leaderboard(sort_by="COCO", descending=True):
    df = load_all_results()
    return df.sort_values(by=sort_by, ascending=not descending).reset_index(drop=True)


def build_interface():
    with gr.Blocks() as demo:
        gr.Markdown("## 📊 LVU VLM Leaderboard")
        gr.Markdown("Benchmarks: COCO, VQAv2, OKVQA, GQA")

        with gr.Row():
            sort_metric = gr.Dropdown(choices=["COCO", "VQAv2", "OKVQA", "GQA"], label="Sort by", value="COCO")
            descending = gr.Checkbox(label="Descending", value=True)

        leaderboard = gr.Dataframe(label="Leaderboard")

        sort_metric.change(fn=sort_leaderboard, inputs=[sort_metric, descending], outputs=leaderboard)
        descending.change(fn=sort_leaderboard, inputs=[sort_metric, descending], outputs=leaderboard)

        leaderboard.render()
    return demo


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