# -*- coding: utf-8 -*-
"""
Created on Mon Jan  6 18:36:08 2025

@author: Alex
"""

import pandas as pd
import json
import numpy as np

trialNum = 260 #TODO: could be gained from data
test = pd.read_csv("./Data/emotFeedbackbinary.csv",header=None)
test.columns=["id","workerID","experimentData","reward"]
data = test["experimentData"]

for part in range(0,len(data)):
    partData = json.loads(data[part])
    partData["id"] = [part]*trialNum
    partData["reward"] = [test["reward"][part]]*trialNum
    partData["completionTime"] = [(partData['endTime'][0]-partData['startTime'][0])/60000]*trialNum
    for k in partData.keys():
        if type(partData[k]) is not list:
            elongate = [partData[k]]*trialNum
            partData[k] = elongate
        elif len(partData[k])==1:
            partData[k] = partData[k]*trialNum
    if part==0:
        df = pd.DataFrame.from_dict(partData)
    else:
        df = pd.concat([df,pd.DataFrame.from_dict(partData)])
        
df = df.reset_index()

df.to_csv("./data/fullPilot2.csv",index=False)

paymentData = pd.concat([test.workerID,test.reward],axis=1)
paymentData.to_csv('./data/bonus.txt', sep=',', index=False)