Address comments

Author: BabyChouSr

fastchat/serve/gradio_block_arena_named.py

@@ -29,6 +29,8 @@
     get_ip,
     get_model_description_md,
     _write_to_json,
+    show_vote_button,
+    dont_show_vote_button,
 )
 from fastchat.serve.moderation.moderator import AzureAndOpenAIContentModerator
 from fastchat.serve.remote_logger import get_remote_logger
@@ -178,7 +180,7 @@ def add_text(
                 no_change_btn,
             ]
             * 6
-            + [True]
+            + [dont_show_vote_button]
         )
 
     model_list = [states[i].model_name for i in range(num_sides)]
@@ -197,7 +199,7 @@ def add_text(
                 no_change_btn,
             ]
             * 6
-            + [True]
+            + [dont_show_vote_button]
         )
 
     conv = states[0].conv
@@ -213,7 +215,7 @@ def add_text(
                 no_change_btn,
             ]
             * 6
-            + [True]
+            + [dont_show_vote_button]
         )
 
     text = text[:INPUT_CHAR_LEN_LIMIT]  # Hard cut-off
@@ -230,7 +232,7 @@ def add_text(
             disable_btn,
         ]
         * 6
-        + [False]
+        + [show_vote_button]
     )
 
 
@@ -378,7 +380,7 @@ def build_side_by_side_ui_named(models):
     states = [gr.State() for _ in range(num_sides)]
     model_selectors = [None] * num_sides
     chatbots = [None] * num_sides
-    dont_show_vote_buttons = gr.State(False)
+    show_vote_buttons = gr.State(True)
 
     notice = gr.Markdown(notice_markdown, elem_id="notice_markdown")
 
@@ -540,24 +542,24 @@ def build_side_by_side_ui_named(models):
     textbox.submit(
         add_text,
         states + model_selectors + [textbox],
-        states + chatbots + [textbox] + btn_list + [dont_show_vote_buttons],
+        states + chatbots + [textbox] + btn_list + [show_vote_buttons],
     ).then(
         bot_response_multi,
         states + [temperature, top_p, max_output_tokens],
         states + chatbots + btn_list,
     ).then(
-        flash_buttons, [dont_show_vote_buttons], btn_list
+        flash_buttons, [show_vote_buttons], btn_list
     )
     send_btn.click(
         add_text,
         states + model_selectors + [textbox],
-        states + chatbots + [textbox] + btn_list + [dont_show_vote_buttons],
+        states + chatbots + [textbox] + btn_list + [show_vote_buttons],
     ).then(
         bot_response_multi,
         states + [temperature, top_p, max_output_tokens],
         states + chatbots + btn_list,
     ).then(
-        flash_buttons, [dont_show_vote_buttons], btn_list
+        flash_buttons, [show_vote_buttons], btn_list
     )
 
     return states + model_selectors