updates

app.py

@@ -172,6 +172,10 @@ def get_bot_profile(bot_config):
 
 
 with gr.Blocks() as demo:
+    gr.Markdown("""
+    # Chai: Model Evaluation
+    Visit each tab for details ⬇️
+    """)
     with gr.Tabs():
         with gr.TabItem("Playground"):
             get_tab_playground(download_bot_config, get_bot_profile, model_mapping)

tabs/arena_battle.py

@@ -6,6 +6,14 @@ from conversation import Conversation
 
 
 def get_tab_arena_battle(download_bot_config, get_bot_profile, model_mapping, client):
+    gr.Markdown("""
+    # ⚔️ Chatbot Arena (battle) ⚔️
+    ## Rules
+    * Chat with two anonymous models side-by-side and vote for which one is better!
+    * You can do multiple rounds of conversations before voting or vote for each message.
+    * The names of the models will be revealed of the top after your voted and pressed "Show models".
+    * Click “Restart” to start a new round with new models.
+    """)
     default_bot_id = "_bot_e21de304-6151-4a04-b025-4c553ae8cbca"
     bot_config = download_bot_config(default_bot_id)
     user_state = gr.State(
@@ -16,7 +24,7 @@ def get_tab_arena_battle(download_bot_config, get_bot_profile, model_mapping, cl
         reload_bot_button = gr.Button("Reload bot")
     bot_profile = gr.HTML(get_bot_profile(bot_config))
     with gr.Accordion("Bot config:", open=False):
-        gr.Markdown(f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}\n")
+        bot_config_text = gr.Markdown(f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}\n")
 
     with gr.Row():
         values = list(model_mapping.keys())
@@ -81,7 +89,8 @@ def get_tab_arena_battle(download_bot_config, get_bot_profile, model_mapping, cl
     def reload_bot(bot_id):
         bot_config = download_bot_config(bot_id)
         bot_profile = get_bot_profile(bot_config)
-        return bot_profile, [(None, bot_config["firstMessage"])], [(None, bot_config["firstMessage"])], bot_config
+        return bot_profile, [(None, bot_config["firstMessage"])], [(None, bot_config[
+            "firstMessage"])], bot_config, f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}"
 
     def get_generation_args(model_tag):
         model = model_mapping[model_tag]
@@ -212,7 +221,7 @@ def get_tab_arena_battle(download_bot_config, get_bot_profile, model_mapping, cl
     # model_a.change(enable_voting, None, [vote_a, vote_b, vote_tie, vote_bad], queue=False)
     # model_b.change(enable_voting, None, [vote_a, vote_b, vote_tie, vote_bad], queue=False)
     reload_bot_button.click(disable_voting, None, [vote_a, vote_b, vote_tie, vote_bad], queue=False)
-    reload_bot_button.click(reload_bot, [bot_id], [bot_profile, chatbot_a, chatbot_b, user_state],
+    reload_bot_button.click(reload_bot, [bot_id], [bot_profile, chatbot_a, chatbot_b, user_state, bot_config_text],
                             queue=False)
     send.click(enable_voting, None, [vote_a, vote_b, vote_tie, vote_bad], queue=False)
     clear.click(disable_voting, None, [vote_a, vote_b, vote_tie, vote_bad], queue=False)

tabs/arena_side_by_side.py

@@ -6,6 +6,13 @@ from conversation import Conversation
 
 
 def get_tab_arena_side_by_side(download_bot_config, get_bot_profile, model_mapping, client):
+    gr.Markdown("""
+    # ⚔️ Chatbot Arena (side-by-side) ⚔️
+    ## Rules
+    * Chat with two models side-by-side and vote for which one is better!
+    * You pick the models you want to chat with.
+    * You can continue chatting and voting or click “Clear” to start a new round.
+    """)
     default_bot_id = "_bot_e21de304-6151-4a04-b025-4c553ae8cbca"
     bot_config = download_bot_config(default_bot_id)
     user_state = gr.State(
@@ -16,7 +23,7 @@ def get_tab_arena_side_by_side(download_bot_config, get_bot_profile, model_mappi
         reload_bot_button = gr.Button("Reload bot")
     bot_profile = gr.HTML(get_bot_profile(bot_config))
     with gr.Accordion("Bot config:", open=False):
-        gr.Markdown(f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}\n")
+        bot_config_text = gr.Markdown(f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}\n")
 
     with gr.Row():
         values = list(model_mapping.keys())
@@ -80,7 +87,8 @@ def get_tab_arena_side_by_side(download_bot_config, get_bot_profile, model_mappi
     def reload_bot(bot_id):
         bot_config = download_bot_config(bot_id)
         bot_profile = get_bot_profile(bot_config)
-        return bot_profile, [(None, bot_config["firstMessage"])], [(None, bot_config["firstMessage"])], bot_config
+        return bot_profile, [(None, bot_config["firstMessage"])], [(None, bot_config[
+            "firstMessage"])], bot_config, f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}"
 
     def get_generation_args(model_tag):
         model = model_mapping[model_tag]
@@ -184,7 +192,7 @@ def get_tab_arena_side_by_side(download_bot_config, get_bot_profile, model_mappi
     model_b.change(get_generation_args, [model_b],
                    [temperature_model_b, repetition_penalty_model_b, max_new_tokens_model_b, top_k_model_b,
                     top_p_model_b], queue=False)
-    reload_bot_button.click(reload_bot, [bot_id], [bot_profile, chatbot_a, chatbot_b, user_state],
+    reload_bot_button.click(reload_bot, [bot_id], [bot_profile, chatbot_a, chatbot_b, user_state, bot_config_text],
                             queue=False)
     clear.click(clear_chat, [user_state], [msg, chatbot_a, chatbot_b], queue=False)
     model_a.change(clear_chat, [user_state], [msg, chatbot_a, chatbot_b], queue=False)

tabs/playground.py

@@ -3,6 +3,12 @@ from conversation import Conversation
 
 
 def get_tab_playground(download_bot_config, get_bot_profile, model_mapping):
+    gr.Markdown("""
+    # 🎢.️ Playground 🎢.️
+    ## Rules
+    * Chat with any model you would like with any bot from the Chai app.
+    * Click “Clear” to start a new conversation.
+    """)
     default_bot_id = "_bot_e21de304-6151-4a04-b025-4c553ae8cbca"
     bot_config = download_bot_config(default_bot_id)
     user_state = gr.State(
@@ -14,7 +20,7 @@ def get_tab_playground(download_bot_config, get_bot_profile, model_mapping):
 
     bot_profile = gr.HTML(get_bot_profile(bot_config))
     with gr.Accordion("Bot config:", open=False):
-        gr.Markdown(f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}\n")
+        bot_config_text = gr.Markdown(f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}")
 
     first_message = (None, bot_config["firstMessage"])
     chatbot = gr.Chatbot([first_message])
@@ -85,7 +91,8 @@ def get_tab_playground(download_bot_config, get_bot_profile, model_mapping):
     def reload_bot(bot_id, bot_profile, chat_history):
         bot_config = download_bot_config(bot_id)
         bot_profile = get_bot_profile(bot_config)
-        return bot_profile, [(None, bot_config["firstMessage"])], bot_config
+        return bot_profile, [(None, bot_config[
+            "firstMessage"])], bot_config, f"# Memory\n{bot_config['memory']}\n# Prompt\n{bot_config['prompt']}"
 
     def get_generation_args(model_tag):
         model = model_mapping[model_tag]
@@ -111,5 +118,6 @@ def get_tab_playground(download_bot_config, get_bot_profile, model_mapping):
     regenerate.click(regenerate_response,
                      [chatbot, user_state, model_tag, temperature, repetition_penalty, max_new_tokens, top_k,
                       top_p], [chatbot], queue=False)
-    reload_bot_button.click(reload_bot, [bot_id, bot_profile, chatbot], [bot_profile, chatbot, user_state],
+    reload_bot_button.click(reload_bot, [bot_id, bot_profile, chatbot],
+                            [bot_profile, chatbot, user_state, bot_config_text],
                             queue=False)