from Util import ConfigUtil
from Util.Logger import *
from Util.MjUtil import *

if __name__ == '__main__':
    # 配置文件
    config = ConfigUtil.getConfig()
    # 第三方接入TOKEN
    ApiToken = config['api']['ApiToken']
    # 第三方接入服务器的Server地址
    ApiServer = config['api']['ApiServer']

    # 绘鸭的Stable Diffusion 日志文件
    LOG_SD = Logger('../Log/HuiYa_SD.log', level='debug')
    log_sd = LOG_SD.logger
    # 绘鸭的Midjourney日志文件
    LOG_MJ = Logger('../Log/HuiYa_MJ.log', level='debug')
    log_mj = LOG_MJ.logger

    # 提示词
    prompt = 'A beautiful Chinese girl with long curly hair --ar 16:9'
    # input_img = None  # 不使用垫图,在绘鸭中一般不会有这种情况发生
    input_img = "Image/mote.png"  # 使用垫图

    # 生图
    res1 = imagine(api_server=ApiServer, api_token=ApiToken, prompt=prompt, type_code=MS_RELAX, input_img=input_img)
    if res1['code'] == 1:
        taskId = res1['result']
        log_mj.info("生图任务创建成功:task_id = " + taskId)
    else:
        log_mj.error("生图任务创建生成失败," + str(res1))