|
|
# pip install psycopg2
|
|
|
|
|
|
from Util.PgUtil import *
|
|
|
|
|
|
# 修改操作（插入、更新或删除）
|
|
|
sql = "INSERT INTO t_hy_model(model_id,model_name,memo,star,update_time,img_count,b_use,sort_id,model_type_id) VALUES (%s,%s, %s,%s,now(),%s,1,1,%s);"
|
|
|
|
|
|
# 模型编号
|
|
|
model_id = 31
|
|
|
# 模型名称
|
|
|
model_name = '图片皮克斯风格化'
|
|
|
# 提示信息
|
|
|
memo = '1、请上传正面半身照。; 2、确保脸部无遮挡；3、图片要求不小于1.5MB（正常手机拍照即可）'
|
|
|
# 星级
|
|
|
star = 5
|
|
|
# 上传照片数量
|
|
|
img_count = 1
|
|
|
# 模板类型
|
|
|
# 1：WEBUI 2:COMFY_UI
|
|
|
model_type_id = 1
|
|
|
|
|
|
params = (model_id, model_name, memo, star, img_count, model_type_id)
|
|
|
execute_modify(sql, params)
|
|
|
|
|
|
# 模板的提示词初始化
|
|
|
prompts = [{"prompt_id": 1, "prompt": "1111", "img_url": "https://oss上的根据模型和指定提示词生成的样张.jpg"},
|
|
|
{"prompt_id": 2, "prompt": "2222", "img_url": "2222.jpg"},
|
|
|
{"prompt_id": 3, "prompt": "3333", "img_url": "3333.jpg"},
|
|
|
{"prompt_id": 4, "prompt": "4444", "img_url": "4444.jpg"},
|
|
|
{"prompt_id": 5, "prompt": "5555", "img_url": "5555.jpg"}, ]
|
|
|
|
|
|
sql = "insert into t_hy_model_prompt(model_id,prompt_id,prompt,img_url) values(%s,%s,%s,%s)"
|
|
|
for x in prompts:
|
|
|
prompt_id = x['prompt_id']
|
|
|
prompt = x['prompt']
|
|
|
img_url = x['img_url']
|
|
|
params = (model_id, prompt_id, prompt, img_url)
|
|
|
execute_modify(sql, params)
|
