Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,829 Bytes
8a142a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import os
import pandas as pd
import random
from .models import model_names
def load_leaderboard_data():
"""
Loads the leaderboard data from the leaderboard CSV file.
Returns the data in a format compatible with the application.
"""
# Initialize the results structure
results = {"wins": {}, "losses": {}, "ties": {}, "votes": 0}
try:
# Define the path to the CSV file for leaderboard
csv_path = os.path.join('utils', 'arena_df_leaderboard.csv')
# Check if the file exists and load it
if os.path.exists(csv_path):
df = pd.read_csv(csv_path)
# Process the data into our structure
for _, row in df.iterrows():
model = row['model']
results["wins"][model] = row['wins']
results["losses"][model] = row['losses']
results["ties"][model] = row['ties']
# Calculate total votes
for model in results["wins"].keys():
results["votes"] += results["wins"][model] + results["losses"][model] + results["ties"][model] // 2
else:
# If file doesn't exist, pre-populate with some data
for model in model_names:
results["wins"][model] = random.randint(0, 10)
results["losses"][model] = random.randint(0, 10)
results["ties"][model] = random.randint(0, 5)
# Calculate total votes
for model in model_names:
results["votes"] += results["wins"][model] + results["losses"][model] + results["ties"][model] // 2
return results
except Exception as e:
print(f"Error loading leaderboard data: {e}")
# Return the initialized structure if file can't be loaded
return results
def save_leaderboard_data(results):
"""
Saves the current leaderboard results back to the CSV file.
Parameters:
- results: The results dictionary containing wins, losses, ties, and votes
"""
try:
# Define the path to the CSV file
csv_path = os.path.join('utils', 'arena_df_leaderboard.csv')
# Convert the results dictionary to a DataFrame
data = []
for model in results["wins"].keys():
data.append({
'model': model,
'wins': results["wins"].get(model, 0),
'losses': results["losses"].get(model, 0),
'ties': results["ties"].get(model, 0)
})
df = pd.DataFrame(data)
# Save to CSV
df.to_csv(csv_path, index=False)
print(f"Leaderboard data saved successfully to {csv_path}")
except Exception as e:
print(f"Error saving leaderboard data: {e}")
|