import os
import random
from Util import ConfigUtil
from Util . SDUtil import *
from Util . CommonUtil import *
# 获取Model对象
def getModel ( model_id ) :
file_name = r ' ../JSON/ ' + str ( model_id ) + ' .json '
# 打开JSON文件
if not os . path . exists ( file_name ) :
printf ( " 文件： " + file_name + " 不存在，跳过！ " )
return
with open ( file_name , ' r ' , encoding = ' utf-8 ' ) as file :
data = json . load ( file )
return data
# 整体的生成任务
def runWebUI ( model_id , task_type_code , input_image , total_tasks , prompt ) :
global data
data = getModel ( model_id ) # data是全局变量
# 生成多少张
data [ ' batch_size ' ] = total_tasks
# 图片
args = data [ ' alwayson_scripts ' ] [ ' ControlNet ' ] [ ' args ' ]
pose_image = data [ ' pose_image ' ]
args [ 0 ] [ ' input_image ' ] = encode_image ( input_image )
# 如果有特殊的姿态要求,那就用模板指定的姿态图
if len ( pose_image ) > 0 :
if len ( args ) > 1 :
args [ 1 ] [ ' input_image ' ] = encode_image ( pose_image )
else : # 如果没有特殊的姿态要求，那么就统一用一张图片
if len ( args ) > 1 :
args [ 1 ] [ ' input_image ' ] = encode_image ( input_image [ 0 ] )
# 提示词
data [ ' prompt ' ] = prompt
# 开始生成图片
printf ( " 开始生成图片... " )
# 设置目标文件夹
target_folder = " Out "
# 创建以当前日期命名的目录
folder = os . path . join ( target_folder , ' Images/ ' + task_type_code + " / " + str ( model_id ) + ' / ' )
if not os . path . exists ( folder ) :
os . makedirs ( folder )
# 调用生图服务
response = submit_post ( txt2img_url , data )
# 保存为图片
for i in range ( 0 , total_tasks ) :
# 创建文件名
png_file = str ( uuid . uuid4 ( ) ) + ' .png '
# 创建文件的完整路径
file_path = os . path . join ( folder , png_file )
save_encoded_image ( response . json ( ) [ ' images ' ] [ i ] , file_path )
printf ( " 本轮图片生成成功, 共完成 " + str ( total_tasks ) + " 张图片 " )
if __name__ == ' __main__ ' :
# 打开配置文件
config = ConfigUtil . getConfig ( )
# 配置文件
config = ConfigUtil . getConfig ( )
# 处理机编号
machine_id = config [ ' system ' ] [ ' machine_id ' ]
# 生图服务地址
txt2img_url = config [ ' webui ' ] [ ' txt2img_url ' ]
# WEB服务器地址
web_url = config [ ' webui ' ] [ ' web_url ' ]
# 不使用代理
set_http_proxy ( " " )
prompt = ' (masterpiece, best quality:1.2),1boy,blue eyes,bodysuit,breasts,building,city,Smooth metal surface,Close fitting mecha,Part of the complex mecha structure,Streamlined mecha,lips,looking at viewer,mecha,medium breasts,realistic,science fiction,short hair,solo,<lora:cncnj95e878c73b9ogcg:0.8>,<lora:ip-adapter-faceid-plusv2_sdxl_lora:0.5>, '
# 第20个少惠林学员
for k in range ( 49 , 50 ) :
total_count = 2
runWebUI ( 15 , ' User ' , ' ../ShlImages/shl_xs_ ' + str ( k ) + ' .jpg ' , total_count ,
prompt ) # 原来的input_image是单张图片，
