Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 5,579 Bytes
d40e945 6e03e6f d40e945 6e03e6f d40e945 6e03e6f d40e945 6e03e6f d40e945 6e03e6f d40e945 6e03e6f 95b1085 6e03e6f 95b1085 6e03e6f 95b1085 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
import gradio as gr
from .config import *
from .synth import *
from .vote import *
from .messages import *
def disable():
return [gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)]
def enable():
return [gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)]
with gr.Blocks() as vote:
# sample played
#aplayed = gr.State(value=False)
#bplayed = gr.State(value=False)
# voter ID
useridstate = gr.State()
gr.Markdown(INSTR)
with gr.Group():
with gr.Row():
text = gr.Textbox(container=False, show_label=False, placeholder="Enter text to synthesize", lines=1, max_lines=1, scale=9999999, min_width=0)
randomt = gr.Button('🎲', scale=0, min_width=0, variant='tool', elem_id="vote-random-button")
randomt.click(randomsent, outputs=[text, randomt])
btn = gr.Button("Synthesize", variant='primary', elem_id="vote-synth-button")
model1 = gr.Textbox(interactive=False, lines=1, max_lines=1, visible=False)
#model1 = gr.Textbox(interactive=False, lines=1, max_lines=1, visible=True)
model2 = gr.Textbox(interactive=False, lines=1, max_lines=1, visible=False)
#model2 = gr.Textbox(interactive=False, lines=1, max_lines=1, visible=True)
with gr.Row(visible=False) as r2:
with gr.Column():
with gr.Group():
aud1 = gr.Audio(interactive=False, show_label=False, show_download_button=False, show_share_button=False, autoplay=True, elem_id="vote-a-audio")
abetter = gr.Button("A is better", variant='primary', elem_id="vote-a-button")
prevmodel1 = gr.Textbox(interactive=False, show_label=False, container=False, value="Vote to reveal model A", text_align="center", lines=1, max_lines=1, visible=False)
with gr.Column():
with gr.Group():
aud2 = gr.Audio(interactive=False, show_label=False, show_download_button=False, show_share_button=False, elem_id="vote-b-audio")
bbetter = gr.Button("B is better", variant='primary', elem_id="vote-b-button")
prevmodel2 = gr.Textbox(interactive=False, show_label=False, container=False, value="Vote to reveal model B", text_align="center", lines=1, max_lines=1, visible=False)
aud1.stop(None, None, js="""function() {
document.getElementById('vote-b-audio')?.querySelector('button.play-pause-button')?.click();
}""")
nxtroundbtn = gr.Button('Next round', visible=False)
# outputs = [text, btn, r2, model1, model2, prevmodel1, aud1, prevmodel2, aud2, abetter, bbetter]
outputs = [
text,
btn,
r2,
model1,
model2,
aud1,
aud2,
abetter,
bbetter,
prevmodel1,
prevmodel2,
nxtroundbtn
]
"""
text,
"Synthesize",
gr.update(visible=True), # r2
mdl1, # model1
mdl2, # model2
gr.update(visible=True, value=results[mdl1]), # aud1
gr.update(visible=True, value=results[mdl2]), # aud2
gr.update(visible=True, interactive=False), #abetter
gr.update(visible=True, interactive=False), #bbetter
gr.update(visible=False), #prevmodel1
gr.update(visible=False), #prevmodel2
gr.update(visible=False), #nxt round btn"""
btn.click(disable, outputs=[btn, abetter, bbetter]).then(synthandreturn, inputs=[text], outputs=outputs).then(enable, outputs=[btn, abetter, bbetter])
nxtroundbtn.click(clear_stuff, outputs=outputs)
# Allow interaction with the vote buttons only when both audio samples have finished playing
#aud1.stop(unlock_vote, outputs=[abetter, bbetter, aplayed, bplayed], inputs=[gr.State(value=0), aplayed, bplayed])
#aud2.stop(unlock_vote, outputs=[abetter, bbetter, aplayed, bplayed], inputs=[gr.State(value=1), aplayed, bplayed])
# nxt_outputs = [prevmodel1, prevmodel2, abetter, bbetter]
nxt_outputs = [abetter, bbetter, prevmodel1, prevmodel2, nxtroundbtn]
abetter.click(a_is_better, outputs=nxt_outputs, inputs=[model1, model2, useridstate])
bbetter.click(b_is_better, outputs=nxt_outputs, inputs=[model1, model2, useridstate])
# Add custom JS for keyboard shortcuts
# vote.load(None, None, js="""function() {
# document.addEventListener('keydown', function(e) {
# // Only handle shortcuts if we're on the vote tab
# if (!document.querySelector('#vote-a-button')?.offsetParent) {
# // Check if random button is visible
# if (document.querySelector('#vote-random-button')?.offsetParent) {
# if (e.key === 'r') {
# document.getElementById('vote-random-button')?.click();
# } else if (e.key === 'Enter' && (e.ctrlKey || e.metaKey)) {
# document.getElementById('vote-synth-button')?.click();
# }
# }
# return;
# }
# // Only handle A and B keys when not typing in an input
# if (document.activeElement.tagName === 'INPUT' ||
# document.activeElement.tagName === 'TEXTAREA') {
# return;
# }
# if (e.key.toLowerCase() === 'a') {
# document.getElementById('vote-a-button')?.click();
# } else if (e.key.toLowerCase() === 'b') {
# document.getElementById('vote-b-button')?.click();
# }
# });
# }""")
|