Clear inputs on model change

app.py

@@ -46,8 +46,11 @@ def model_gen(content, model_type: str, chat_history):
     print(MODEL_LIB[model_type])
     return gen(content, chat_history)
 
-def clear_chat(chatbot, msg):
-    return [], ''
+def clear_chat():
+    return '', []
+
+def rerun(msg, chatbot):
+    return msg, chatbot[:-1]
 
 MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 
@@ -66,15 +69,26 @@ def tab_arena():
         gradio.Button('Right Better')
         gradio.Button('Both Bad')
 
-    with gradio.Accordion("See Details"):
-        context = gradio.Checkbox(label="No context", value=False)
+    with gradio.Row():
+        with gradio.Accordion("Parameters", open=False):
+            context = gradio.Checkbox(label="No context", value=False)
+            top_p = gradio.Slider(label='Top P', minimum=0, maximum=1, value=1, step=0.5)
+            temp = gradio.Slider(label='Temperature', minimum=0, maximum=1, value=0.7, step=0.5)
+            max_tokens = gradio.Slider(label='Max ouput tokens', minimum=1, maximum=2048, value=512, step=1)
 
-    msg = gradio.Textbox()
-    clear = gradio.ClearButton([msg, chatbot_left, chatbot_right])
+    with gradio.Row():
+        msg = gradio.Textbox()
+
+    with gradio.Row():
+        clear = gradio.ClearButton([msg, chatbot_left, chatbot_right], value='Clear history')
+        regen_left = gradio.Button(value='Regenerate left answer')
+        regen_right = gradio.Button(value='Regenerate right answer')
+        regen_left.click(rerun, [msg, chatbot_left], [msg, chatbot_left])
+        regen_right.click(rerun, [msg, chatbot_right], [msg, chatbot_left])
     
     with gradio.Blocks():
-        model_left.change(clear_chat, [chatbot_left, msg], [chatbot_left, msg])
-        model_right.change(clear_chat, [chatbot_right, msg], [chatbot_right, msg])
+        model_left.change(clear_chat, [], [msg, chatbot_left])
+        model_right.change(clear_chat, [], [msg, chatbot_right])
         msg.submit(model_gen, [msg, model_left, chatbot_left], [msg, chatbot_left])
         msg.submit(model_gen, [msg, model_right, chatbot_right], [msg, chatbot_right])