Update app.py

app.py

@@ -53,6 +53,7 @@ def get_bot_response(url, prompt):
     }
     response = requests.post(url, json=payload, headers=headers)
     return response.json()
+
 def chat_with_bots(user_input):
     bot_names = list(chatbots.keys())
     random.shuffle(bot_names)
@@ -63,33 +64,53 @@ def chat_with_bots(user_input):
 
     return bot1_response, bot2_response
 
-def update_ratings(bot_index):
+def user_ask(state, chatbot, textbox):
+    user_input = textbox.value
+    bot1_response, bot2_response = chat_with_bots(user_input)
+    chatbot.append("User: " + user_input)
+    chatbot.append("Bot 1: " + bot1_response['output'])
+    chatbot.append("Bot 2: " + bot2_response['output'])
+    state['last_bots'] = [bot1_response['model_name'], bot2_response['model_name']]
+    return state, chatbot, ""
+
+def update_ratings(state, winner_index):
     elo_ratings = get_user_elo_ratings()
     bot_names = list(chatbots.keys())
-    winner = bot_names[bot_index]
-    loser = bot_names[1 - bot_index]
+    winner = state['last_bots'][winner_index]
+    loser = state['last_bots'][1 - winner_index]
     
     elo_ratings = update_elo_ratings(elo_ratings, winner, loser)
     write_elo_ratings(elo_ratings)
-    
-    return f"Bot 1 is {bot_names[0]} with ELO {elo_ratings[bot_names[0]]}\nBot 2 is {bot_names[1]} with ELO {elo_ratings[bot_names[1]]}"
-
-# Gradio Interface
-iface = gr.Interface(
-    fn=chat_with_bots,
-    inputs=[
-        gr.components.Textbox(label="Your message")
-    ],
-    outputs=[
-        gr.components.Textbox(label="Bot 1 Response"),
-        gr.components.Textbox(label="Bot 2 Response")
-    ],
-    live=True
-)
-
-iface.add_component(gr.components.Radio(["Bot 1", "Bot 2"], label="Vote for the Best Response"), "vote")
-iface.add_component(gr.components.Textbox(label="Voting Result"), "vote_result")
-
-iface.update(value=update_ratings, component_name="vote", outputs=["vote_result"])
-
-iface.launch()
+    return f"Updated ELO ratings:\n{winner}: {elo_ratings[winner]}\n{loser}: {elo_ratings[loser]}"
+
+def vote_up_model(state, chatbot):
+    update_message = update_ratings(state, 0)
+    chatbot.append(update_message)
+    return chatbot
+
+def vote_down_model(state, chatbot):
+    update_message = update_ratings(state, 1)
+    chatbot.append(update_message)
+    return chatbot
+
+with gr.Blocks() as demo:
+    state = gr.State({})
+
+    with gr.Row():
+        with gr.Column(scale=0.5):
+            chatbot = gr.Chatbot(label='ChatBox')
+            with gr.Row():
+                textbox = gr.Textbox(placeholder="Enter text and press ENTER")
+                submit_btn = gr.Button(value="Submit")
+        with gr.Column():
+            upvote_btn = gr.Button(value="👍 Upvote Bot 1")
+            downvote_btn = gr.Button(value="👎 Upvote Bot 2")
+            clear_btn = gr.Button(value="🗑️ Clear Chat")
+
+    textbox.submit(user_ask, [state, chatbot, textbox], [state, chatbot, textbox])
+    submit_btn.click(user_ask, [state, chatbot, textbox], [state, chatbot, textbox])
+    upvote_btn.click(vote_up_model, [state, chatbot], [chatbot])
+    downvote_btn.click(vote_down_model, [state, chatbot], [chatbot])
+    clear_btn.click(lambda _: chatbot.clear(), inputs=[], outputs=[chatbot])
+
+demo.launch(share=True, enable_queue=True, server_name='0.0.0.0', server_port=7860)