diff --git a/BaiHu/Image/48/1.png b/BaiHu/Image/48/1.png new file mode 100644 index 00000000..cae85d2b Binary files /dev/null and b/BaiHu/Image/48/1.png differ diff --git a/BaiHu/Image/48/2.png b/BaiHu/Image/48/2.png new file mode 100644 index 00000000..8be31ac8 Binary files /dev/null and b/BaiHu/Image/48/2.png differ diff --git a/BaiHu/Image/48/3.png b/BaiHu/Image/48/3.png new file mode 100644 index 00000000..6d4c6412 Binary files /dev/null and b/BaiHu/Image/48/3.png differ diff --git a/BaiHu/Image/48/4.png b/BaiHu/Image/48/4.png new file mode 100644 index 00000000..3adc4a82 Binary files /dev/null and b/BaiHu/Image/48/4.png differ diff --git a/BaiHu/Image/48/5.png b/BaiHu/Image/48/5.png new file mode 100644 index 00000000..ab3c892c Binary files /dev/null and b/BaiHu/Image/48/5.png differ diff --git a/BaiHu/Tools/addModelWithInterface.py b/BaiHu/Tools/addModelWithInterface.py new file mode 100644 index 00000000..e6a754a2 --- /dev/null +++ b/BaiHu/Tools/addModelWithInterface.py @@ -0,0 +1,59 @@ +# pip install psycopg2 +import requests + +from Util.AesUtil import * +from Util.CommonUtil import set_http_proxy +from Util.PgUtil import * + +# 模型编号 +model_id = 48 +# 模型名称 +model_name = '海的女儿' +# 模板类型 +# 1:SD TXT2IMG 2:SD IMG2IMG 3:COMFY_UI +model_type_id = 2 +''' +1 卡通风格 +2 儿童摄影 +3 风格写真 +4 双人系列 +''' +style_id = 2 +# 是不是封面 +is_cover = 1 + +# 提示信息 +memo = '1、请上传正面半身照。; 2、确保脸部无遮挡;3、图片要求不小于1.5MB(正常手机拍照即可)' +# 星级 +star = 5 +# 上传照片数量 +img_count = 1 + +if __name__ == '__main__': + # 禁止代理 + set_http_proxy("") + # 配置文件 + config = ConfigUtil.getConfig() + # 处理机编号 + machine_id = config['system']['machine_id'] + # 获取签名 + en_data = getenData(machine_id) + # WEB服务器地址 + web_url = config['webServer']['web_url'] + # 请求的地址 + url = web_url + '/QingLong/HuiYa/addModel' + + # 要发送的数据,可以是字典形式 + data = { + "enData": en_data, + "model_name": model_name, + "memo": memo, + "star": star, + "img_count": img_count, + "model_type_id": model_type_id, + "style_id": style_id, + "is_cover": is_cover + } + # 发送POST请求 + response = requests.post(url, data=data) + print(response.text) diff --git a/BaiHu/Util/CommonUtil.py b/BaiHu/Util/CommonUtil.py index db2b7096..959e64f0 100644 --- a/BaiHu/Util/CommonUtil.py +++ b/BaiHu/Util/CommonUtil.py @@ -6,6 +6,21 @@ from datetime import datetime import requests from PIL import Image import io +import urllib.parse +import urllib.request + + +# 是否启用代理 +def set_http_proxy(proxy): + if proxy == None: # Use system default setting + proxy_support = urllib.request.ProxyHandler() + elif proxy == '': # Don't use any proxy + proxy_support = urllib.request.ProxyHandler({}) + else: # Use proxy + proxy_support = urllib.request.ProxyHandler({'http': '%s' % proxy, 'https': '%s' % proxy}) + opener = urllib.request.build_opener(proxy_support) + urllib.request.install_opener(opener) + def submit_post(url: str, data: dict): """ @@ -103,4 +118,3 @@ def split_4_image(image_path, out_path): bottom_right.save(fi) return res - diff --git a/BaiHu/Util/SDUtil.py b/BaiHu/Util/SDUtil.py index 8c0149d9..2215cf38 100644 --- a/BaiHu/Util/SDUtil.py +++ b/BaiHu/Util/SDUtil.py @@ -3,26 +3,12 @@ import time import urllib.parse import urllib.request -import urllib.parse -import urllib.request import websocket from Util.CommonUtil import * -# 是否启用代理 -def set_http_proxy(proxy): - if proxy == None: # Use system default setting - proxy_support = urllib.request.ProxyHandler() - elif proxy == '': # Don't use any proxy - proxy_support = urllib.request.ProxyHandler({}) - else: # Use proxy - proxy_support = urllib.request.ProxyHandler({'http': '%s' % proxy, 'https': '%s' % proxy}) - opener = urllib.request.build_opener(proxy_support) - urllib.request.install_opener(opener) - - # 定义一个函数向服务器队列发送提示信息 def queue_prompt(server_address, client_id, prompt): p = {"prompt": prompt, "client_id": client_id} diff --git a/BaiHu/Util/__pycache__/CommonUtil.cpython-310.pyc b/BaiHu/Util/__pycache__/CommonUtil.cpython-310.pyc index 16481382..16759582 100644 Binary files a/BaiHu/Util/__pycache__/CommonUtil.cpython-310.pyc and b/BaiHu/Util/__pycache__/CommonUtil.cpython-310.pyc differ diff --git a/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java b/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java index 2eedff0a..b544f33b 100644 --- a/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java +++ b/QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java @@ -525,7 +525,7 @@ public class HuiYaController extends Controller { */ @Before({POST.class}) @EmptyInterface({"enData", "model_name", "memo"}) - @IsNumericInterface({"model_id", "star", "img_count", "model_type_id", "style_id", "is_cover"}) + @IsNumericInterface({"star", "img_count", "model_type_id", "style_id", "is_cover"}) public void addModel(String enData, String model_name, String memo, int star, int img_count, int model_type_id, int style_id, int is_cover) { int success = AesUtil.CheckSignature(enData); if (success <= 0) {