Update app.py

app.py

@@ -134,24 +134,44 @@ def user_ask(state, chatbot1, chatbot2, textbox):
 # Gradio interface setup
 with gr.Blocks() as demo:
     state = gr.State({})
-
-    with gr.Row():
-        with gr.Column():
-            chatbot1 = gr.Chatbot(label='Model A').style(height=600)
-            upvote_btn_a = gr.Button(value="👍 Upvote A",interactive=False)
-        
-        with gr.Column():
-            chatbot2 = gr.Chatbot(label='Model B').style(height=600)
-            upvote_btn_b = gr.Button(value="👍 Upvote B",interactive=False)        
-
-    textbox = gr.Textbox(placeholder="Enter your prompt (up to 200 characters)", max_chars=200)
-    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_a,upvote_btn_b])
-    upvote_btn_b.click(vote_down_model, inputs=[state, chatbot2], outputs=[chatbot2,upvote_btn_a,upvote_btn_b])
-
-    demo.launch()
+    with gr.Tab("Chatbot Arena"):
+        with gr.Row():
+            with gr.Column():
+                chatbot1 = gr.Chatbot(label='Model A').style(height=600)
+                upvote_btn_a = gr.Button(value="👍 Upvote A",interactive=False)
+            
+            with gr.Column():
+                chatbot2 = gr.Chatbot(label='Model B').style(height=600)
+                upvote_btn_b = gr.Button(value="👍 Upvote B",interactive=False)        
+    
+        textbox = gr.Textbox(placeholder="Enter your prompt (up to 200 characters)", max_chars=200)
+        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_a,upvote_btn_b])
+        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot2], outputs=[chatbot2,upvote_btn_a,upvote_btn_b])
+    with gr.Tab("Leaderboard"):
+        leaderboard = gr.Dataframe()
+        refresh_btn = gr.Button("Refresh Leaderboard")
+
+    # Function to refresh leaderboard
+    def refresh_leaderboard():
+        return generate_leaderboard()
+
+    # Event handler for the refresh button
+    refresh_btn.click(refresh_leaderboard, inputs=[], outputs=[leaderboard])
+
+    # Launch the Gradio interface
+
+    demo.launch()
+import pandas as pd
+
+# Function to generate leaderboard data
+def generate_leaderboard():
+    elo_ratings = read_elo_ratings()  # Assuming this function returns a dict of {bot_name: elo_score}
+    leaderboard_data = pd.DataFrame(list(elo_ratings.items()), columns=['Chatbot', 'ELO Score'])
+    leaderboard_data = leaderboard_data.sort_values('ELO Score', ascending=False)
+    return leaderboard_data