import json


# file_array = ["./Image/C17/1.jpg"]
# 根据输入的图片集合，填充到prompt_data中
def fill_input(prompt_data, file_array):
    # 是不是不存在配置节
    if "source_images" not in prompt_data:
        print("配置文件不包含source_images配置节，程序无法继续！")
        exit(0)
    # 是不是个数不匹配
    if len(prompt_data["source_images"]) != len(file_array):
        print("输入的图片数量与预替换的图片数量不一致，程序无法继续！")
        exit(0)
    # 替换
    for i in range(len(prompt_data["source_images"])):
        x = prompt_data["source_images"][i]
        # 替换配置节内容
        array = x.split(",")
        prompt_data[array[0]][array[1]][array[2]] = file_array[i]
    return prompt_data


if __name__ == '__main__':
    # JSON文件
    json_file = r'../JSON/17_test.json'

    # 读取JSON文件
    with open(json_file, 'r', encoding='utf-8') as file:
        prompt_data = json.load(file)

    if "source_images" not in prompt_data:
        print("未正确配置source_images配置节，程序无法继续！")
        exit(0)
    else:
        print("已正确配置了source_images配置节！")

    # 输入的图片地址
    file_path = ["./Image/C17/1.jpg"]
    if len(prompt_data["source_images"]) != len(file_path):
        print("输入的图片数量与预替换的图片数量不一致，程序无法继续！")
        exit(0)

    for i in range(len(prompt_data["source_images"])):
        x = prompt_data["source_images"][i]
        # 替换配置节内容
        array = x.split(",")
        prompt_data[array[0]][array[1]][array[2]] = file_path[i]
        # print(prompt_data["331"]["inputs"]["image"])

    # print(json.dumps(prompt_data, sort_keys=True, indent=4)) # 美化格式化输出
    print(json.dumps(prompt_data, sort_keys=True, ensure_ascii=False))  # 生成可以正确显示中文的json格式输出

    # 一、姿势图  风格图 都是需要封装到json文件内部的是，是固定的，是死的，不需要代码去处理
    # 二、源图  source_images，这个需要代码去处理，因为
    # (1)  每个输入放在哪个配置节下，是不一样的
    # (2)  可能不止一张