main
luanhailiang 2023-10-07 12:10:22 +08:00
parent ff85571b47
commit 85a3221e94
1 changed files with 21 additions and 7 deletions

28
show.py
View File

@ -1,4 +1,5 @@
import cv2
import os
import json
import time
import logging
@ -31,7 +32,10 @@ def mj():
print(request.json)
if request.json["progress"] == "100%":
print(request.json["imageUrl"])
sendmsg(request.json["imageUrl"])
if request.json['state'] == str(479169837):
sendmsg(request.json["imageUrl"],479169837)
return 'ok'
sendmsg(request.json["imageUrl"],5464741)
download_image(request.json["imageUrl"], "MJ", request.json["state"])
return 'ok'
@ -42,7 +46,7 @@ def cq():
return 'ok'
if request.json["message_type"] != "group":
return 'ok'
if request.json["group_id"] != 116277572:
if request.json["group_id"] != 116277572 and request.json["group_id"] != 479169837:
return 'ok'
# if request.json["user_id"] != 2132960:
# return 'ok'
@ -98,16 +102,19 @@ def handle(data):
'state': name
}
if data["group_id"] == 479169837:
params['state'] = str(479169837)
response = imagine(params)
print(response.json())
time.sleep(3)
params['prompt'] = prompt + " --iw 2 --niji 5"
params['state'] = name+"_niji"
# params['state'] = name+"_niji"
response = imagine(params)
print(response.json())
time.sleep(3)
params['prompt'] = prompt + " --iw 2 --style raw"
params['state'] = name+"_raw"
# params['state'] = name+"_raw"
response = imagine(params)
print(response.json())
@ -121,7 +128,14 @@ def download_image(url, t, name):
:param local_filename: 要保存图片的本地文件名
"""
local_filename = f"y:\美术\参考收集\{t}\{datetime.now().strftime('%Y%m%d%H%M%S')}_{name}"
local_filename = f"y:\美术\参考收集\{t}\{name}"
if not os.path.exists(local_filename):
# 如果目录不存在,则创建它
os.makedirs(local_filename)
print(f'目录{local_filename}已创建')
local_filename += "\\"+datetime.now().strftime('%Y%m%d_%H%M%S')
response = requests.get(url)
if response.status_code == 200: # HTTP状态码200表示请求成功
# 获取Content-Type响应头
@ -177,13 +191,13 @@ def imagine(params):
return response
def sendmsg(url):
def sendmsg(url,id):
# 设置 go-cqhttp 服务的 URL 和端口
api_url = 'http://localhost:5700/send_msg'
# 设置要发送的消息参数
data = {
'group_id': 5464741, # 接收消息的用户的 QQ 号
'group_id': id, # 接收消息的用户的 QQ 号
'message_type': 'group', # 消息类型
'message': f'[CQ:image,file={url}]' # 消息内容,使用 CQ码 格式
}