from Liblib.LibLibUtil import LibLibUtil

if __name__ == '__main__':
    """

"""
    generate_params = {
        "templateUuid": "6f7c4652458d4802969f8d089cf5b91f",  # 参数模板ID F.1文生图
        "generateParams": {
            "prompt": "filmfotos, Asian portrait,A young woman wearing a green baseball cap,covering one eye with her hand",
            "steps": 20,  # 采样步数
            "width": 768,  # 宽
            "height": 1024,  # 高
            "imgCount": 1,  # 图片数量
            "seed": -1,  # 随机种子值，-1 表示随机
            "restoreFaces": 0,  # 面部修复，0 关闭，1 开启
            # Lora添加，最多5个，不添加lora时请删除此结构体
            "additionalNetwork": [
                {
                    "modelId": "169505112cee468b95d5e4a5db0e5669",  # LoRA的模型版本uuid
                    "weight": 1.0  # LoRA权重
                }
            ]
        }
    }

    liblib = LibLibUtil()
    # 调用生成接口
    response = liblib.post_request(
        "/api/generate/webui/text2img",
        generate_params
    )
    # {'generateUuid': 'f16463d55e144214ae12d5177f8256fc'}
    print(response)