You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
1.1 KiB
32 lines
1.1 KiB
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))
|