[#34] Block submission during response generation (#35)

* [#34] Block submission during response generation

Changes
- Users cannot click the Run button while the response is being generated
- Component activation logic is moved from the response generation function to the Run button click handler

Screenshot: https://screen.yanolja.in/OkPOwEtERIM4NQro.png

* Remove unnecessary comments

* Add TODO

---------

Co-authored-by: suhyun.kang <suhyun.kang@yanolja.group>

app.py

@@ -139,9 +139,21 @@ with gr.Blocks(title="Arena") as app:
   instruction_state = gr.State("")
 
   submit.click(
-      get_responses, [prompt, category_radio, source_language, target_language],
-      response_boxes + model_names + vote_buttons +
-      [instruction_state, model_name_row, vote_row])
+      fn=get_responses,
+      inputs=[prompt, category_radio, source_language, target_language],
+      outputs=response_boxes + model_names + [instruction_state]).then(
+          fn=lambda: [gr.Button(interactive=True),
+                      gr.Row(visible=True)
+                     ] + [gr.Button(interactive=True) for _ in range(3)],
+          outputs=[submit, vote_row] + vote_buttons)
+
+  # TODO(#42): Hide vote buttons until response generation is successful.
+  submit.click(fn=lambda: [
+      gr.Button(interactive=False),
+      gr.Row(visible=False),
+      gr.Row(visible=False)
+  ],
+               outputs=[submit, vote_row, model_name_row])
 
   common_inputs = response_boxes + model_names + [
       prompt, instruction_state, category_radio, source_language,

response.py

@@ -51,8 +51,6 @@ def get_responses(user_prompt, category, source_lang, target_lang):
 
   models = sample(list(supported_models), 2)
   instruction = get_instruction(category, source_lang, target_lang)
-  activated_vote_buttons = [gr.Button(interactive=True) for _ in range(3)]
-  deactivated_vote_buttons = [gr.Button(interactive=False) for _ in range(3)]
 
   responses = []
   for model in models:
@@ -85,16 +83,6 @@ def get_responses(user_prompt, category, source_lang, target_lang):
   # It simulates concurrent stream response generation.
   max_response_length = max(len(response) for response in responses)
   for i in range(max_response_length):
-    yield [response[:i + 1] for response in responses
-          ] + models + deactivated_vote_buttons + [
-              instruction,
-              gr.Row(visible=False),
-              gr.Row(visible=False)
-          ]
-
-  # After generating the response, the vote_row should become visible,
-  # while the model_name_row should remain hidden.
-  yield responses + models + activated_vote_buttons + [
-      instruction, gr.Row(visible=False),
-      gr.Row(visible=True)
-  ]
+    yield [response[:i + 1] for response in responses] + models + [instruction]
+
+  yield responses + models + [instruction]