Clémentine
hides all when selecting anything
fd5a922
raw
history blame
4.48 kB
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"*Tags: {', '.join(LEADERBOARDS_TO_INFO[space]) if len(LEADERBOARDS_TO_INFO[space]) > 0 else 'None. Please fill the tags!'}*"
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 [modality_tags, submission_tags, test_set_tags, evaluation_tags, language_tags]:
selector.change(
lambda _: False,
outputs=show_all
)
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()