# %%
import pandas as pd
import pickle
import datasets
# ファイルパスを指定
file_path = 'elo_results.pkl'

# 'rb'は「バイナリ読み込みモード」を意味します
with open(file_path, 'rb') as file:
    # pickleを使用してファイルの内容を読み込む
    data = pickle.load(file)

# 読み込んだデータを表示
print(data)

# %%

data["full"]["elo_rating_final"]

# %%

df2 = data["full"]["leaderboard_table_df"]
df2 = df2.sort_values("rating", ascending=False)

# index名がdrop_listに含まれる行を削除
drop_list = [
    "weblab-GENIAC/Tanuki-8x8B-dpo-v1.0",
    "Llama-3.1-Swallow-70B-v0.1.Q8_0.gguf",
]

df2 = df2.reset_index()
df2 = df2[~df2["index"].isin(drop_list)]
ds2 = datasets.Dataset.from_pandas(df2)
ds2.push_to_hub("kanhatakeyama/chatbot-arena-ja-elo-rating")


# %%