0.34 pass choice

app.py

@@ -180,7 +180,7 @@ def generate_both(system_prompt, input_text, chatbot_a, chatbot_b, max_new_token
 def clear():
     return [], []
 
-def reveal_bot(chatbot_a, chatbot_b):
+def reveal_bot(selection, chatbot_a, chatbot_b):
     chatbot_a.append(["User", "I am Bot Arno"])
     chatbot_b.append(["User", "I am Bot Bernd"])
     return chatbot_a, chatbot_b
@@ -212,7 +212,7 @@ with gr.Blocks() as demo:
             top_p = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, label="Top-p", step=0.01)
             repetition_penalty = gr.Slider(minimum=0.1, maximum=2.0, value=1.1, label="Repetition Penalty", step=0.1)
 
-    better_bot.select(reveal_bot, inputs=[chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) #fckp outputs=[chatbot_a, chatbot_b]
+    better_bot.select(reveal_bot, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) #fckp outputs=[chatbot_a, chatbot_b]
     input_text.submit(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])
     submit_btn.click(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])
     clear_btn.click(clear, outputs=[chatbot_a, chatbot_b])