# Add this at the top of your script
import warnings

warnings.filterwarnings("ignore")

import gradio as gr
from data_loader import (
    load_data,
    CATEGORIES,
    METHODOLOGY,
    HEADER_CONTENT,
    CARDS
)
from tabs.leaderboard_v1 import create_leaderboard_tab, filter_leaderboard
from tabs.leaderboard_v2 import create_leaderboard_v2_interface


def create_app():
    df = load_data()

    with gr.Blocks(
        theme=gr.themes.Default(primary_hue=gr.themes.colors.red)
    ) as app:
        with gr.Tabs():

            # Create v2 tab
            with gr.Tab("Leaderboard v2"):
                create_leaderboard_v2_interface()
                
            # Create v1 tab
            with gr.Tab("Leaderboard v1"):
                lb_output, lb_plot1, lb_plot2 = create_leaderboard_tab(
                    df, CATEGORIES, METHODOLOGY, HEADER_CONTENT, CARDS
                )


        # Initial loads
        app.load(
            fn=lambda: filter_leaderboard(
                df, "All", list(CATEGORIES.keys())[0], "Performance"
            ),
            outputs=[lb_output, lb_plot1, lb_plot2],
        )

    return app


demo = create_app()
demo.launch()