# obtain interactable dataset from semantics dataset
import json
import os
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--input', type=str, required=True)
parser.add_argument('--output', type=str, required=True)
args = parser.parse_args()

with open(args.input, 'r') as fr:
    data = json.load(fr)

data['categories'] = [{'id': 1, 'name': 'interactable'}]
for annotation in data['annotations']:
    annotation['category_id'] = 1

with open(args.output, 'w') as fw:
    json.dump(data, fw, indent=4)