Update app.py

app.py

@@ -91,11 +91,17 @@ def update_ratings(state, winner_index):
     
     elo_ratings = update_elo_ratings(elo_ratings, winner, loser)
     write_elo_ratings(elo_ratings)
-    return f"Updated ELO ratings:\n{winner}: {elo_ratings[winner]}\n{loser}: {elo_ratings[loser]}"
+    return ['Winner: '+state['last_bots'][winner_index],'Loser: '+state['last_bots'][1 - winner_index]]
 
-def vote_up_model(state, chatbot):
+def vote_up_model(state, chatbot,chatbot2):
     update_message = update_ratings(state, 0)
-    chatbot.append(update_message)
+    chatbot.append(update_message[0])
+    chatbot2.append(update_message[1])
+    return chatbot, gr.Button.update(interactive=False),gr.Button.update(interactive=False)  # Disable voting buttons
+def vote_down_model(state, chatbot,chatbot2):
+    update_message = update_ratings(state, 1)
+    chatbot2.append(update_message[0])
+    chatbot.append(update_message[1])
     return chatbot, gr.Button.update(interactive=False),gr.Button.update(interactive=False)  # Disable voting buttons
 
 def user_ask(state, chatbot1, chatbot2, textbox):
@@ -159,8 +165,8 @@ with gr.Blocks() as demo:
         reset_btn.click(clear_chat, inputs=[state], outputs=[state, chatbot1, chatbot2, upvote_btn_a, upvote_btn_b])
         textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox,upvote_btn_a,upvote_btn_b])
         submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox,upvote_btn_a,upvote_btn_b])
-        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1], outputs=[chatbot1,upvote_btn_a,upvote_btn_b])
-        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot2], outputs=[chatbot2,upvote_btn_a,upvote_btn_b])
+        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1,chatbot2], outputs=[chatbot1,chatbot2,upvote_btn_a,upvote_btn_b])
+        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1,chatbot2], outputs=[chatbot1,chatbot2,upvote_btn_a,upvote_btn_b])
     with gr.Tab("Leaderboard"):
         leaderboard = gr.Dataframe()
         refresh_btn = gr.Button("Refresh Leaderboard")