import json
def load_line_json_data(filename):
    data = []
    with open(filename, 'r', encoding='utf-8') as f:
        for line in f.read().strip().split('\n'):
            unit = json.loads(line)
            data.append(unit)
    return data

def save_file(data, path):
    with open(path,'w',encoding='utf-8') as w:
        for unit in data:
            output = json.dumps(unit)
            w.write(output + "\n")
        w.close()

data = load_line_json_data('/home/xj/toolAugEnv/code/toolConstraint/data/query/zk.jsonl')
query_output = open('/home/xj/toolAugEnv/code/toolConstraint/data/api_request/zk_hard_regen.txt','r').read().strip().split('\n')
# assert len(data) == len(query_output)

for unit, output in zip(data[24:], query_output):
    # unit['local_constraint']['transportation'] = None
    unit['query'] = " ".join(x for x in output.split('\t')[1:])
# for unit in data:
#     unit['local_constraint']['transportation'] = None

save_file(data,'/home/xj/toolAugEnv/code/toolConstraint/data/query/zk.jsonl')