Spaces:
Running
Running
| import os | |
| import gradio as gr | |
| import pandas as pd | |
| from gradio.themes.utils.sizes import text_md | |
| from content import (HEADER_MARKDOWN, LEADERBOARD_TAB_TITLE_MARKDOWN, SUBMISSION_TAB_TITLE_MARKDOWN) | |
| from server import LeaderboardServer | |
| leaderboard_server = LeaderboardServer() | |
| def on_submit_pressed(): | |
| return gr.update(value='Processing submission...', interactive=False) | |
| def validate_submission_inputs(team_name, submission_id, link_to_model, submission_file): | |
| if not team_name or not submission_id or not link_to_model or not submission_file: | |
| raise ValueError('Please fill in all fields') | |
| if not os.path.exists(submission_file): | |
| raise ValueError('File does not exist') | |
| def process_submission(team_name, submission_id, description, link_to_model, submission_file): | |
| try: | |
| validate_submission_inputs(team_name, submission_id, link_to_model, submission_file) | |
| metadata = { | |
| "team_name": team_name, | |
| "submission_id": submission_id, | |
| "description": description, | |
| "link_to_model": link_to_model, | |
| } | |
| gr.Info('Submission valid, running local tournament...') | |
| leaderboard_server.prepare_model_for_submission(submission_file, metadata) | |
| except ValueError as err: | |
| gr.Warning(str(err)) | |
| return gr.update(visible=False), gr.update(visible=True), gr.update(interactive=True, | |
| visible=True), gr.update( | |
| interactive=True, visible=True), gr.update(visible=True), gr.update( | |
| value=leaderboard_server.get_leaderboard(leaderboard_server.pre_submit[0]), visible=True) | |
| def submit_results(): | |
| leaderboard_server.save_pre_submit() | |
| leaderboard_server.update_leaderboard() | |
| gr.Info('Submission successful!') | |
| return gr.update(value='Pre-submit model', visible=True, interactive=True), gr.update( | |
| visible=False), gr.update(visible=False), gr.update(visible=False), gr.update( | |
| visible=False), gr.update(visible=False), gr.DataFrame( | |
| value=leaderboard_server.get_leaderboard(), visible=True) | |
| def erase_presubmit(): | |
| leaderboard_server.pre_submit = None | |
| return gr.update(value='Pre-submit model', visible=True, interactive=True), gr.update( | |
| visible=False), gr.update(visible=False), gr.update(visible=False), gr.update( | |
| visible=False), gr.update(visible=False) | |
| with (gr.Blocks(theme=gr.themes.Soft(text_size=text_md), css="footer {visibility: hidden}") as main): | |
| app_state = gr.State({}) | |
| with gr.Row(): | |
| with gr.Row(): | |
| gr.Markdown(HEADER_MARKDOWN) | |
| with gr.Row(): | |
| with gr.Tab('Leaderboard') as leaderboards_tab: | |
| gr.Markdown(LEADERBOARD_TAB_TITLE_MARKDOWN) | |
| results_table = gr.DataFrame(leaderboard_server.get_leaderboard(), interactive=False, label=None, | |
| visible=True) | |
| with gr.Tab('Submission'): | |
| with gr.Column(): | |
| gr.Markdown(SUBMISSION_TAB_TITLE_MARKDOWN) | |
| with gr.Row(): | |
| submission_team_name_tb = gr.Textbox(label='Team Name') | |
| submission_id_tb = gr.Textbox(label='Submission ID') | |
| with gr.Row(): | |
| description_tb = gr.Textbox(label='Description', type='text') | |
| link_to_model_tb = gr.Textbox(label='Link to model', type='text') | |
| submission_file_path = gr.File(label='Upload your results', type='filepath') | |
| pre_submission_btn = gr.Button(value='Pre-submit model', interactive=True) | |
| submit_prompt = gr.Markdown( | |
| """ | |
| Do you really want to submit a model? This action is irreversible. | |
| """, | |
| visible=False | |
| ) | |
| pre_submit_info = gr.Markdown( | |
| """ | |
| This is how will ranking look like after your submission: | |
| """, | |
| visible=False | |
| ) | |
| pre_submit_table = gr.DataFrame(pd.DataFrame(), interactive=False, label=None, visible=False) | |
| submission_btn_yes = gr.Button(value='Submit model', interactive=False, visible=False) | |
| submission_btn_no = gr.Button(value='Reverse process', interactive=False, visible=False) | |
| pre_submission_btn.click( | |
| fn=on_submit_pressed, | |
| outputs=[pre_submission_btn] | |
| ).then( | |
| fn=process_submission, | |
| inputs=[submission_team_name_tb, submission_id_tb, description_tb, link_to_model_tb, | |
| submission_file_path], | |
| outputs=[pre_submission_btn, submit_prompt, submission_btn_yes, submission_btn_no, pre_submit_info, | |
| pre_submit_table] | |
| ) | |
| submission_btn_yes.click( | |
| fn=submit_results, | |
| outputs=[pre_submission_btn, submission_btn_yes, submission_btn_no, submit_prompt, pre_submit_info, | |
| pre_submit_table, results_table] | |
| ) | |
| submission_btn_no.click( | |
| fn=erase_presubmit, | |
| outputs=[pre_submission_btn, submission_btn_yes, submission_btn_no, submit_prompt, pre_submit_info, | |
| pre_submit_table] | |
| ) | |
| main.launch() | |