src/app.py

import gradio as gr

from chain_data import sync_metagraph
from leaderboard import create_leaderboard, create_dropdown
from validator_states import create_validator_states
from validator_weights import create_weights
from wandb_data import sync


def main():
    sync_metagraph()
    sync()
    with gr.Blocks(css=".typewriter {font-family: 'JMH Typewriter', sans-serif;}", fill_height=True, fill_width=True) as app:
        with gr.Tab("Leaderboard") as leaderboard_tab:
            dropdown = gr.Dropdown()
            dropdown.attach_load_event(lambda: create_dropdown(), None)

            leaderboard_dataframe = gr.Dataframe()
            leaderboard_dataframe.attach_load_event(lambda uid: create_leaderboard(uid), None, inputs=[dropdown])
            leaderboard_tab.select(lambda uid: create_leaderboard(uid), inputs=[dropdown], outputs=[leaderboard_dataframe])

            dropdown.change(lambda uid: create_leaderboard(uid), inputs=[dropdown], outputs=[leaderboard_dataframe])

        with gr.Tab("Validator States") as validator_states_tab:
            validator_states_dataframe = gr.Dataframe()
            validator_states_dataframe.attach_load_event(lambda: create_validator_states(), None)
            validator_states_tab.select(lambda: create_validator_states(), [], [validator_states_dataframe])

        with gr.Tab("Validator Weights") as validator_weights_tab:
            validator_weights_dataframe = gr.Dataframe()
            validator_weights_dataframe.attach_load_event(lambda: create_weights(), None)
            validator_weights_tab.select(lambda: create_weights(), [], [validator_weights_dataframe])

        with gr.Tab("Model Demo"):
            gr.Label("Coming soon!", show_label=False)
    app.launch()


if __name__ == "__main__":
    main()