from Util import ConfigUtil
from Util.CommonUtil import *


# 图生图
def img2img(url, img_path, v_data, save_path):
    image = encode_image(img_path)
    v_data['init_images'] = [image]
    v_data['alwayson_scripts']['controlnet']['args'][0]['image'] = image
    response = submit_post(url, v_data)
    save_encoded_image(response.json()['images'][0], save_path)


if __name__ == '__main__':
    # 配置文件
    config = ConfigUtil.getConfig()
    # 图生图服务地址
    img2img_url = config['webui']['img2img_url']

    # 图片路径
    image_path = "../Image/mote.png"
    # 生成的JSON文件
    data = \
        {
            "init_images": [],
            "sampler_name": "Euler",
            "alwayson_scripts": {
                "controlnet": {
                    "args": [
                        {
                            "image": "",
                            "module": "none", "model": "control_openpose-fp16 [9ca67cc5]",
                        }
                    ]
                }
            }
        }
    img2img(img2img_url, image_path, data, "../Out/result.png")
    # print(json.dumps(data, sort_keys=True, ensure_ascii=False))  # 生成可以正确显示中文的json格式输出