Update app.py

app.py

@@ -162,12 +162,22 @@ def user_ask(state, chatbot1, chatbot2, textbox):
 import pandas as pd
 
 # Function to generate leaderboard data
+
 def generate_leaderboard():
-    elo_ratings = get_user_elo_ratings()  # Assuming this function returns a dict of {bot_name: elo_score}
-    leaderboard_data = pd.DataFrame(list(elo_ratings.iter(batch_size=100)), columns=['Chatbot', 'ELO Score'])
+    elo_ratings_dataset = get_user_elo_ratings()  # Returns a Hugging Face dataset
+
+    # Convert the Hugging Face dataset to a pandas DataFrame
+    leaderboard_data = pd.DataFrame(elo_ratings_dataset)
+
+    # Assuming the dataset has columns 'bot_name' and 'elo_rating'
+    leaderboard_data.columns = ['Chatbot', 'ELO Score']
+
+    # Sort the DataFrame based on the ELO Score in descending order
     leaderboard_data = leaderboard_data.sort_values('ELO Score', ascending=False)
+
     return leaderboard_data
 
+
 # Gradio interface setup
 with gr.Blocks() as demo:
     state = gr.State({})