import json
import os
import sys

def generate_output(input_config, time):
    with open(input_config, "r") as f:
        data = json.load(f)
    output_dir_with_taskid = os.path.join(data["output_dir"], str(data["task_id"]))
    # Ensure the output directory exists before writing the output file
    os.makedirs(output_dir_with_taskid, exist_ok=True)
    output_data = {
        "task_id": data["task_id"],
        "convert_time": time,
        "data_dir": data["data_dir"],
        "output_dir": output_dir_with_taskid,
        "episode_num": data["episode_num"],
        "no_language": data["no_language"],
    }
    output_json_path = os.path.join(output_dir_with_taskid, "output.json")
    with open(output_json_path, "w") as f:
        json.dump(output_data, f)

if __name__ == "__main__":
    input_config = sys.argv[1]
    time = int(sys.argv[2])
    generate_output(input_config, time)