3DGen-Arena / serve /gradio_web_t2i_named.py
ZhangYuhan's picture
update server
f8f1d3f
raw
history blame
8.51 kB
import json
from functools import partial
from .utils import *
from .vote_utils import (
upvote_last_response_t2s as upvote_last_response,
downvote_last_response_t2s as downvote_last_response,
flag_last_response_t2s as flag_last_response,
leftvote_last_response_t2s_multi as leftvote_last_response,
rightvote_last_response_t2s_multi as rightvote_last_response,
tievote_last_response_t2s_multi as tievote_last_response,
bothbad_vote_last_response_t2s_multi as bothbad_vote_last_response,
share_click_t2s_multi as share_click,
share_js
)
from .inference import(
sample_prompt,
generate_t2s_multi
)
from constants import TEXT_PROMPT_PATH
with open(TEXT_PROMPT_PATH, 'r') as f:
prompt_list = json.load(f)
def build_side_by_side_ui_named(models):
notice_markdown = """
# βš”οΈ GenAI-Arena βš”οΈ : Benchmarking Text-to-3D generative models
## πŸ“œ Rules
- Generate with any two selected models side-by-side and vote!
- Sample or Input prompt you want to generate.
- Click "Send" to submit the prompt.
- Click "Clear" to start a new round.
## πŸ† Arena Elo
Find out who is the πŸ₯‡conditional image generation models! More models are going to be supported.
## πŸ‘‡ Generating now!
"""
model_list = models.get_t2s_models()
gen_func = partial(generate_t2s_multi, models.inference_parallel, models.render_parallel)
state_0 = gr.State()
state_1 = gr.State()
gr.Markdown(notice_markdown, elem_id="notice_markdown")
with gr.Group(elem_id="share-region-named"):
with gr.Row():
with gr.Column():
model_selector_left = gr.Dropdown(
choices=model_list,
value=model_list[0] if len(model_list) > 0 else "",
interactive=True,
show_label=False,
container=False,
)
with gr.Column():
model_selector_right = gr.Dropdown(
choices=model_list,
value=model_list[1] if len(model_list) > 1 else "",
interactive=True,
show_label=False,
container=False,
)
with gr.Accordion("πŸ” Expand to see all Arena players", open=False):
model_description_md = get_model_description_md(model_list)
gr.Markdown(model_description_md, elem_id="model_description_markdown")
with gr.Row():
with gr.Column():
normal_left = gr.Image(width=512, label = "Model A", show_copy_button=True)
rgb_left = gr.Image(width=512, label = "Model A", show_copy_button=True)
with gr.Column():
normal_right = gr.Image(width=512, label = "Model B", show_copy_button=True,)
rgb_right = gr.Image(width=512, label = "Model B", show_copy_button=True,)
with gr.Row():
slow_warning = gr.Markdown("", elem_id="notice_markdown")
with gr.Row(elem_id="Geometry Quality"):
geo_leftvote_btn = gr.Button(
value="πŸ‘ˆ A is better", visible=False, interactive=False
)
geo_rightvote_btn = gr.Button(
value="πŸ‘‰ B is better", visible=False, interactive=False
)
geo_tie_btn = gr.Button(value="🀝 Tie", visible=False, interactive=False)
geo_bothbad_btn = gr.Button(
value="πŸ‘Ž Both are bad", visible=False, interactive=False
)
with gr.Row(elem_id="Texture Quality"):
text_leftvote_btn = gr.Button(
value="πŸ‘ˆ A is better", visible=False, interactive=False
)
text_rightvote_btn = gr.Button(
value="πŸ‘‰ B is better", visible=False, interactive=False
)
text_tie_btn = gr.Button(value="🀝 Tie", visible=False, interactive=False)
text_bothbad_btn = gr.Button(
value="πŸ‘Ž Both are bad", visible=False, interactive=False
)
with gr.Row(elem_id="Alignment Quality"):
align_leftvote_btn = gr.Button(
value="πŸ‘ˆ A is better", visible=False, interactive=False
)
align_rightvote_btn = gr.Button(
value="πŸ‘‰ B is better", visible=False, interactive=False
)
align_tie_btn = gr.Button(value="🀝 Tie", visible=False, interactive=False)
align_bothbad_btn = gr.Button(
value="πŸ‘Ž Both are bad", visible=False, interactive=False
)
with gr.Row():
textbox = gr.Textbox(
show_label=False,
placeholder="πŸ‘‰ Enter your prompt or Sample a random prompt, and press ENTER",
container=True,
elem_id="input_box",
)
sample_btn = gr.Button(value="🎲 Sample", variant="primary", scale=0)
send_btn = gr.Button(value="πŸ“€ Send", variant="primary", scale=0)
with gr.Row():
clear_btn = gr.Button(value="πŸ—‘οΈ Clear", interactive=False)
regenerate_btn = gr.Button(value="πŸ”„ Regenerate", interactive=False)
share_btn = gr.Button(value="πŸ“· Share")
gr.Markdown(acknowledgment_md, elem_id="ack_markdown")
geo_btn_list = [geo_leftvote_btn, geo_rightvote_btn, geo_tie_btn, geo_bothbad_btn]
text_btn_list = [text_leftvote_btn, text_rightvote_btn, text_tie_btn, text_bothbad_btn]
align_btn_list = [align_leftvote_btn, align_rightvote_btn, align_tie_btn, align_bothbad_btn]
states = [state_0, state_1]
model_selectors = [model_selector_left, model_selector_right]
results = [normal_left, rgb_left, normal_right, rgb_right]
model_selector_left.change(
clear_history_side_by_side,
None,
states + [textbox] + results,
api_name="model_selector_left"
)
model_selector_right.change(
clear_history_side_by_side,
None,
states + [textbox] + results,
api_name="model_selector_right"
)
for btn_list in [geo_btn_list, text_btn_list, align_btn_list]:
leftvote_btn, rightvote_btn, tie_btn, bothbad_btn = btn_list
leftvote_btn.click(
leftvote_last_response,
states + model_selectors,
[textbox] + btn_list + model_selectors
)
rightvote_btn.click(
rightvote_last_response,
states + model_selectors,
[textbox] + btn_list + model_selectors
)
tie_btn.click(
tievote_last_response,
states + model_selectors,
[textbox] + btn_list + model_selectors
)
bothbad_btn.click(
bothbad_vote_last_response,
states + model_selectors,
[textbox] + btn_list + model_selectors
)
sample_btn.click(
sample_prompt,
states + model_selectors + [prompt_list],
states + [textbox],
api_name="sample_btn_named"
)
textbox.then(
gen_func,
states + [textbox] + model_selectors + [prompt_list],
states + results + model_selectors,
api_name="submit_btn_named"
).then(
enable_buttons_side_by_side,
None,
geo_btn_list + text_btn_list + align_btn_list + [regenerate_btn, clear_btn]
)
send_btn.click(
gen_func,
states + [textbox] + model_selectors + [prompt_list],
states + results + model_selectors,
api_name="send_btn_named"
).then(
enable_buttons_side_by_side,
None,
geo_btn_list + text_btn_list + align_btn_list + [regenerate_btn, clear_btn]
)
clear_btn.click(
clear_history_side_by_side_anony,
None,
states + [textbox] + results + model_selectors,
api_name="clear_btn_named"
).then(
disable_buttons_side_by_side,
None,
geo_btn_list + text_btn_list + align_btn_list + [regenerate_btn, clear_btn]
)
regenerate_btn.click(
gen_func,
states + [textbox] + model_selectors + [prompt_list],
states + results + model_selectors,
api_name="regenerate_btn_named"
).then(
enable_buttons_side_by_side,
None,
geo_btn_list + text_btn_list + align_btn_list + [regenerate_btn, clear_btn]
)
share_btn.click(
share_click,
states + model_selectors,
[],
js=share_js
)