import gradio as gr from apscheduler.schedulers.background import BackgroundScheduler from src.static.env import API, REPO_ID, HF_TOKEN from src.static.about import TITLE, INTRO, ABOUT, DOCUMENTATION from src.leaderboards.get_from_hub import get_leaderboard_info from src.static.tag_info import * from src.static.display import make_clickable def restart_space(): API.restart_space(repo_id=REPO_ID, token=HF_TOKEN) LEADERBOARDS_TO_INFO, INFO_TO_LEADERBOARDS = get_leaderboard_info() def update_leaderboards(show_all, modality_tags, submission_tags, test_set_tags, evaluation_tags, language_tags): spaces_of_interest = [] if show_all: spaces_of_interest = INFO_TO_LEADERBOARDS["all"] else: for tag in modality_tags: spaces_of_interest.extend(INFO_TO_LEADERBOARDS["modality"][tag.lower()]) for tag in submission_tags: spaces_of_interest.extend(INFO_TO_LEADERBOARDS["submission"][tag.lower()]) for tag in test_set_tags: spaces_of_interest.extend(INFO_TO_LEADERBOARDS["test"][tag.lower()]) for tag in evaluation_tags: spaces_of_interest.extend(INFO_TO_LEADERBOARDS["modality"][tag.lower()]) for tag in language_tags: spaces_of_interest.extend(INFO_TO_LEADERBOARDS["language"][tag.lower()]) return "- " + "\n - ".join([ make_clickable(space) + f"\n*Tags: {', '.join(LEADERBOARDS_TO_INFO[space])}*" for space in spaces_of_interest ]) demo = gr.Blocks() with demo: gr.Markdown(TITLE) gr.Markdown(INTRO, elem_classes="markdown-text") with gr.Tabs(elem_classes="tab-buttons") as tabs: with gr.TabItem("Search"): gr.Markdown("Let's look for leaderboards relevant for you! Select the categories of your choice") with gr.Row(): with gr.Column(): show_all = gr.Checkbox( value=False, label="Show all leaderboards" ) modality_tags = gr.CheckboxGroup( choices=[tag.name for tag in Modality], value=[], label="Modality of choice" ) submission_tags = gr.CheckboxGroup( choices=[tag.name for tag in SubmissionType], value=[], label="Submission type" ) test_set_tags = gr.CheckboxGroup( choices=[tag.name for tag in TestSetStatus], value=[], label="Test set status" ) with gr.Column(): evaluation_tags = gr.CheckboxGroup( choices=[tag.name for tag in EvaluationCategory], value=[], label="Specific evaluation categories" ) language_tags = gr.CheckboxGroup( choices=[tag.capitalize() for tag in sorted(list(INFO_TO_LEADERBOARDS["language"].keys()))], value=[], label="Specific languages" ) with gr.Row(): leaderboards = gr.Markdown( value="", ) for selector in [show_all, modality_tags, submission_tags, test_set_tags, evaluation_tags, language_tags]: selector.change( update_leaderboards, [show_all, modality_tags, submission_tags, test_set_tags, evaluation_tags, language_tags], leaderboards, queue=True, ) with gr.TabItem("About"): gr.Markdown(ABOUT, elem_classes="markdown-text") with gr.TabItem("Documentation"): gr.Markdown(DOCUMENTATION, elem_classes="markdown-text") scheduler = BackgroundScheduler() scheduler.add_job(restart_space, "interval", seconds=10800) # restarted every 3h scheduler.start() demo.queue(default_concurrency_limit=40).launch()