Update app.py

app.py

@@ -9,6 +9,11 @@ enable_btn = gr.Button.update(interactive=True)
 # Load chatbot URLs and model names from a JSON file
 with open('chatbot_urls.json', 'r') as file:
     chatbots = json.load(file)
+def clear_chat(state):
+    if state is not None:
+        state = {}
+    return state, None,None,gr.Button.update(interactive=False),gr.Button.update(interactive=False)
+
 
 # Initialize or get user-specific ELO ratings
 def get_user_elo_ratings(state):
@@ -79,7 +84,7 @@ def chat_with_bots(user_input, state):
     return bot1_response, bot2_response
 
 def update_ratings(state, winner_index):
-    elo_ratings = get_user_elo_ratings()
+    elo_ratings = get_user_elo_ratings(state)
     bot_names = list(chatbots.keys())
     winner = state['last_bots'][winner_index]
     loser = state['last_bots'][1 - winner_index]
@@ -91,12 +96,8 @@ def update_ratings(state, winner_index):
 def vote_up_model(state, chatbot):
     update_message = update_ratings(state, 0)
     chatbot.append(update_message)
-    return chatbot
+    return chatbot, gr.Button.update(interactive=False),gr.Button.update(interactive=False)  # Disable voting buttons
 
-def vote_down_model(state, chatbot):
-    update_message = update_ratings(state, 1)
-    chatbot.append(update_message)
-    return chatbot
 def user_ask(state, chatbot1, chatbot2, textbox):
     global enable_btn
     user_input = textbox
@@ -144,11 +145,13 @@ with gr.Blocks() as demo:
             upvote_btn_b = gr.Button(value="👍 Upvote B",interactive=False)        
 
     textbox = gr.Textbox(placeholder="Enter your prompt (up to 200 characters)", max_chars=200)
-    submit_btn = gr.Button(value="Send")
-
+    with gr.Row():
+        submit_btn = gr.Button(value="Send")
+        reset_btn = gr.Button(value="Reset")
+    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_b.click(vote_down_model, inputs=[state, chatbot2], outputs=[chatbot2])
+    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])
 
     demo.launch()