import json, copy, ipdb

cnt = 1052836  #1230k
for i in range(1, 19):
    f = open('./egobody' + str(i) + '.json', 'r')
    data = json.load(f)
    all = []
    for item in data:
        cur = copy.deepcopy(item)
        cur['id'] = str(cnt)
        cur['image'] = "/data7tb/gzy/PoseLLM/Data/" + cur['image']
        cur['conversations'][0]['value'] = "Please provide the SMPL parameters of the single person in the image."
        root_pose = cur['conversations'][1]['root_pose'][0]
        body_pose = cur['conversations'][1]['body_pose'][0]
        shape = cur['conversations'][1]['shape'][0]
        answer = "Sure. Here are the SMPL parameters of this person:\n"
        answer += "root_pose:" + str(root_pose) + '\n'
        answer += "body_pose:" + str(body_pose) + '\n'
        answer += "shape:" + str(shape)
        cur['conversations'][1]['value'] = answer
        all.append(cur)
        cnt += 1
    g = open('./text/EgoBody-' + str(i) + '.json', 'w+')
    json.dump(all, g, indent=4)
    print("finish", i)
print(cnt)