import os
import numpy as np
import pandas as pd
from tqdm import tqdm

if __name__ == "__main__":
    out_preference_csv = "/home/christian/data/interesting_clips_20240528_ear.csv"
    df = pd.read_csv(out_preference_csv)
    num_examples = len(df.index)

    indices = [i for i in range(0, num_examples, 2)]

    print(df.columns)
    agree = []

    for index in tqdm(indices):
        a_row = df.iloc[index]
        b_row = df.iloc[index + 1]

        # print(index, a_row["ear_preference"], b_row["ear_preference"])
        # print(index, a_row["preference"], a_row["ear_preference"])

        if a_row["preference"] == a_row["ear_preference"]:
            agree.append(True)
        else:
            agree.append(False)

    print(f"{np.mean(agree)*100:0.2f}% agree")