Update app.py

app.py

@@ -383,6 +383,7 @@ def reload(chosenmodel1=None, chosenmodel2=None, userid=None, chose_a=False, cho
     else:
         out.append(gr.update(value=f'{chosenmodel1}', interactive=False, visible=True))
         out.append(gr.update(value=f'Your vote: {chosenmodel2}', interactive=False, visible=True))
+    out.append(gr.update(visible=True))
     return out
 
 with gr.Blocks() as leaderboard:
@@ -447,7 +448,6 @@ def synthandreturn(text):
         raise gr.Error(f'You did not enter any text')
     # Get two random models
     mdl1, mdl2 = random.sample(list(AVAILABLE_MODELS.keys()), 2)
-    print("Using models:", mdl1, mdl2)
     return (
         text,
         "Synthesize",
@@ -470,7 +470,10 @@ def synthandreturn(text):
         gr.update(visible=True, interactive=True),
         gr.update(visible=False),
         gr.update(visible=False),
+        gr.update(visible=False), #nxt round btn
     )
+def clear_stuff():
+    return "", "Synthesize", gr.update(visible=False), '', '', gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 with gr.Blocks() as vote:
     useridstate = gr.State()
     gr.Markdown(INSTR)
@@ -490,12 +493,14 @@ with gr.Blocks() as vote:
                 aud2 = gr.Audio(interactive=False, show_label=False, show_download_button=False, show_share_button=False, waveform_options={'waveform_progress_color': '#3C82F6'})
                 bbetter = gr.Button("B is better", variant='primary')
                 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)
+    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]
+    outputs = [text, btn, r2, model1, model2, aud1, aud2, abetter, bbetter, prevmodel1, prevmodel2, nxtroundbtn]
     btn.click(synthandreturn, inputs=[text], outputs=outputs)
+    nxtroundbtn.click(clear_stuff, outputs=outputs)
 
     # nxt_outputs = [prevmodel1, prevmodel2, abetter, bbetter]
-    nxt_outputs = [abetter, bbetter, prevmodel1, prevmodel2]
+    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])
     # skipbtn.click(b_is_better, outputs=outputs, inputs=[model1, model2, useridstate])