Clear inputs on model change

app.py

@@ -43,7 +43,6 @@ def model_gen(content, chat_history, model_type: str):
     if content is None:
         return None, None
     gen = MODEL_LIB[model_type]
-    print(MODEL_LIB[model_type])
     return gen(content, chat_history)
 
 def clear_chat():
@@ -53,7 +52,7 @@ def model_regen(content, chat_history, model_type: str):
     if content is None:
         return None, None
     gen = MODEL_LIB[model_type]
-    print(MODEL_LIB[model_type])
+    print(chat_history[:-1])
     return gen(content, chat_history[:-1])
 
 MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
@@ -67,6 +66,9 @@ def tab_arena():
             model_right = gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], value= "RUBASE", interactive=True, multiselect=False, label="Right model")
             chatbot_right = gradio.Chatbot()
 
+    with gradio.Row():
+        msg = gradio.Textbox(label='Prompt', placeholder='Put your prompt here')
+
     with gradio.Row():
         gradio.Button('Both Good')
         gradio.Button('Left Better')
@@ -80,15 +82,12 @@ def tab_arena():
             temp = gradio.Slider(label='Temperature', minimum=0, maximum=1, value=0.7, step=0.05, interactive=True)
             max_tokens = gradio.Slider(label='Max ouput tokens', minimum=1, maximum=2048, value=512, step=1, interactive=True)
 
-    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(model_regen, [msg, chatbot_left, model_left], [msg, chatbot_left])
-        regen_right.click(model_regen, [msg, chatbot_right, model_right], [msg, chatbot_left])
+        regen_right.click(model_regen, [msg, chatbot_right, model_right], [msg, chatbot_right])
     
     with gradio.Blocks():
         model_left.change(clear_chat, [], [msg, chatbot_left])