Spaces:
Running
on
Zero
Running
on
Zero
| import gradio as gr | |
| import os | |
| from serve.gradio_web import * | |
| from serve.gradio_web_image_editing import * | |
| from serve.gradio_web_video_generation import * | |
| from serve.leaderboard import build_leaderboard_tab | |
| from model.model_manager import ModelManager | |
| from pathlib import Path | |
| from serve.constants import SERVER_PORT, ROOT_PATH, ELO_RESULTS_DIR | |
| def build_combine_demo(models, elo_results_file, leaderboard_table_file): | |
| with gr.Blocks( | |
| title="Play with Open Vision Models", | |
| theme=gr.themes.Default(), | |
| css=block_css, | |
| ) as demo: | |
| with gr.Tabs() as tabs_combine: | |
| with gr.Tab("Image Generation", id=0): | |
| with gr.Tabs() as tabs_ig: | |
| with gr.Tab("Generation Arena (battle)", id=0): | |
| build_side_by_side_ui_anony(models) | |
| with gr.Tab("Generation Arena (side-by-side)", id=1): | |
| build_side_by_side_ui_named(models) | |
| with gr.Tab("Generation Playground", id=2): #Direct Chat | |
| build_single_model_ui(models, add_promotion_links=True) | |
| if elo_results_file: | |
| with gr.Tab("Generation Leaderboard", id=3): | |
| build_leaderboard_tab(elo_results_file['t2i_generation'], leaderboard_table_file['t2i_generation']) | |
| with gr.Tab("Image Edition", id=5): | |
| with gr.Tabs() as tabs_ie: | |
| with gr.Tab("Edition Arena (battle)", id=5): | |
| build_side_by_side_ui_anony_ie(models) | |
| with gr.Tab("Edition Arena (side-by-side)", id=6): | |
| build_side_by_side_ui_named_ie(models) | |
| with gr.Tab("Edition Playground", id=7): #Direct Chat | |
| build_single_model_ui_ie(models, add_promotion_links=True) | |
| if elo_results_file: | |
| with gr.Tab("Edition Leaderboard", id=8): | |
| build_leaderboard_tab(elo_results_file['image_editing'], leaderboard_table_file['image_editing']) | |
| with gr.Tab("Video Generation", id=10): | |
| with gr.Tabs() as tabs_vg: | |
| with gr.Tab("Video Generation Arena (battle)", id=10): | |
| build_side_by_side_ui_anony_vg(models) | |
| with gr.Tab("Video Generation Arena (side-by-side)", id=11): | |
| build_side_by_side_ui_named_vg(models) | |
| with gr.Tab("Video Generation Playground", id=12): #Direct Chat | |
| build_single_model_ui_vg(models, add_promotion_links=True) | |
| if elo_results_file and 'video_generation' in elo_results_file: | |
| with gr.Tab("Video Generation Leaderboard", id=13): | |
| build_leaderboard_tab(elo_results_file['video_generation'], leaderboard_table_file['video_generation']) | |
| with gr.Tab("About Us", id=4): | |
| build_about() | |
| return demo | |
| def load_elo_results(elo_results_dir): | |
| from collections import defaultdict | |
| elo_results_file = defaultdict(lambda: None) | |
| leaderboard_table_file = defaultdict(lambda: None) | |
| if elo_results_dir is not None: | |
| elo_results_dir = Path(elo_results_dir) | |
| elo_results_file = {} | |
| leaderboard_table_file = {} | |
| for file in elo_results_dir.glob('elo_results_*.pkl'): | |
| if 't2i_generation' in file.name: | |
| elo_results_file['t2i_generation'] = file | |
| elif 'image_editing' in file.name: | |
| elo_results_file['image_editing'] = file | |
| elif 'video_generation' in file.name: | |
| elo_results_file['video_generation'] = file | |
| else: | |
| raise ValueError(f"Unknown file name: {file.name}") | |
| for file in elo_results_dir.glob('*_leaderboard.csv'): | |
| if 't2i_generation' in file.name: | |
| leaderboard_table_file['t2i_generation'] = file | |
| elif 'image_editing' in file.name: | |
| leaderboard_table_file['image_editing'] = file | |
| elif 'video_generation' in file.name: | |
| leaderboard_table_file['video_generation'] = file | |
| else: | |
| raise ValueError(f"Unknown file name: {file.name}") | |
| return elo_results_file, leaderboard_table_file | |
| if __name__ == "__main__": | |
| server_port = int(SERVER_PORT) | |
| root_path = ROOT_PATH | |
| elo_results_dir = ELO_RESULTS_DIR | |
| models = ModelManager() | |
| elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir) | |
| demo = build_combine_demo(models, elo_results_file, leaderboard_table_file) | |
| demo.queue(max_size=20).launch(server_port=server_port, root_path=ROOT_PATH) |