From 85a3221e942e2210706f51f60b1e7f32a2c9d550 Mon Sep 17 00:00:00 2001
From: luanhailiang <hi@luanhailiang.cn>
Date: Sat, 7 Oct 2023 12:10:22 +0800
Subject: [PATCH] 2102

---
 show.py | 28 +++++++++++++++++++++-------
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/show.py b/show.py
index 760001f..c524f7d 100644
--- a/show.py
+++ b/show.py
@@ -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码 格式
     }