"""Export leaderboard data to CSV."""

import os

import pandas as pd

from src.envs import EVAL_REQUESTS_PATH, EVAL_RESULTS_PATH
from src.leaderboard.read_evals import get_raw_eval_results
from src.populate import get_leaderboard_df


def export_leaderboard_to_csv(output_path="leaderboard_export.csv"):
    """Export the current leaderboard to a CSV file."""
    raw_data = get_raw_eval_results(EVAL_RESULTS_PATH, EVAL_REQUESTS_PATH)
    from src.display.utils import COLS, BENCHMARK_COLS
    df = get_leaderboard_df(EVAL_RESULTS_PATH, EVAL_REQUESTS_PATH, COLS, BENCHMARK_COLS, raw_data)
    df.to_csv(output_path, index=False)
    print(f"Leaderboard exported to {output_path}")


if __name__ == "__main__":
    export_leaderboard_to_csv()