Update app.py

app.py

@@ -6,7 +6,15 @@ import logging
 import arena_config
 import plotly.graph_objects as go
 from typing import Dict
-from leaderboard import get_current_leaderboard, update_leaderboard, start_backup_thread, get_leaderboard
+from leaderboard import (
+    get_current_leaderboard,
+    update_leaderboard, 
+    start_backup_thread, 
+    get_leaderboard, 
+    get_elo_leaderboard,
+    ensure_elo_ratings_initialized
+)
+import sys
 
 
 # Initialize logging for errors only
@@ -122,6 +130,7 @@ def record_vote(prompt, left_response, right_response, left_model, right_model,
     return (
         gr.update(value=result_message, visible=True),  # Show result as Markdown
         get_leaderboard(),                              # Update leaderboard
+        get_elo_leaderboard(),  # Add this line
         gr.update(interactive=False),                   # Disable left vote button
         gr.update(interactive=False),                   # Disable right vote button
         gr.update(interactive=False),                   # Disable tie button
@@ -200,6 +209,8 @@ def get_leaderboard_chart():
         )
     )
 
+    chart_data = fig.to_json()
+    print(f"Chart size: {sys.getsizeof(chart_data)} bytes")
     return fig
 
 def new_battle():
@@ -315,6 +326,10 @@ with gr.Blocks(css="""
     with gr.Tab("Performance Chart"):
         leaderboard_chart = gr.Plot(label="Model Performance Chart")
     
+    # ELO Leaderboard Tab
+    with gr.Tab("ELO Leaderboard"):
+        elo_leaderboard = gr.HTML(label="ELO Leaderboard")
+    
     # Define interactions
     submit_btn.click(
         battle_arena,
@@ -327,14 +342,14 @@ with gr.Blocks(css="""
     left_vote_btn.click(
         lambda *args: record_vote(*args, "Left is better"),
         inputs=[prompt_input, left_output, right_output, left_model, right_model],
-        outputs=[result, leaderboard, left_vote_btn, 
+        outputs=[result, leaderboard, elo_leaderboard, left_vote_btn, 
                  right_vote_btn, tie_btn, model_names_row, leaderboard_chart]
     )
     
     right_vote_btn.click(
         lambda *args: record_vote(*args, "Right is better"),
         inputs=[prompt_input, left_output, right_output, left_model, right_model],
-        outputs=[result, leaderboard, left_vote_btn, 
+        outputs=[result, leaderboard, elo_leaderboard, left_vote_btn, 
                  right_vote_btn, tie_btn, model_names_row, leaderboard_chart]
     )
     
@@ -353,7 +368,10 @@ with gr.Blocks(css="""
     
     # Update leaderboard and chart on launch
     demo.load(get_leaderboard, outputs=leaderboard)
+    demo.load(get_elo_leaderboard, outputs=elo_leaderboard)
     demo.load(get_leaderboard_chart, outputs=leaderboard_chart)
 
 if __name__ == "__main__":
-    demo.launch(show_api=False)
+    # Initialize ELO ratings before launching the app
+    ensure_elo_ratings_initialized()
+    demo.launch(show_api=False)