Procházet zdrojové kódy

绘图模型添加model=2

关习习 před 2 týdny
rodič
revize
3b4d025314
1 změnil soubory, kde provedl 14 přidání a 14 odebrání
  1. 14 14
      main.py

+ 14 - 14
main.py

@@ -98,16 +98,16 @@ def on_command(group_id, user_id,command):
         stone_stat = ch_stone(command)
         send_group_message_word(group_id, stone_stat)
         save_group_message(group_id, my_name, stone_stat)
-    elif str(command).startswith("/猫猫画画"):
-        pic_word = str(command)[len("/猫猫画画"):]
+    elif str(command).startswith("/猫猫画画2"):
+        pic_word = str(command)[len("/猫猫画画2"):]
         current_time = time.time()
         # 判断距离上次请求图片的时间间隔是否小于60秒(1分钟)
         if current_time - last_request_time >= 0:
             send_group_message_word(group_id, "图片正在生成中...")
             last_request_time = current_time
-            send_group_message_pic(group_id, user_id, story_start(AI_lora_getpic_prompt(pic_word)))
-    elif str(command).startswith("/整点瑟瑟"):
-        pic_word = str(command)[len("/整点瑟瑟"):]
+            send_group_message_pic(group_id, user_id, story_start2(AI_lora_getpic_prompt(pic_word)))
+    elif str(command).startswith("/整点瑟瑟2"):
+        pic_word = str(command)[len("/整点瑟瑟2"):]
         current_time = time.time()
         # 判断距离上次请求图片的时间间隔是否小于60秒(1分钟)
         if group_id == 637612718:
@@ -116,19 +116,21 @@ def on_command(group_id, user_id,command):
             if current_time - last_request_time >= 0:
                 send_group_message_word(group_id, "图片正在生成中...")
                 last_request_time = current_time
-                message_id = send_group_message_pic(group_id, user_id, story_start("nsfw,"+AI_lora_getpic_prompt(pic_word)))['data']['message_id']
+                message_id = \
+                send_group_message_pic(group_id, user_id, story_start2("nsfw,best quality,masterpiece," + AI_lora_getpic_prompt(pic_word)))[
+                    'data']['message_id']
                 time.sleep(12)
                 del_group_message(message_id)
-    elif str(command).startswith("/猫猫画画2"):
-        pic_word = str(command)[len("/猫猫画画2"):]
+    elif str(command).startswith("/猫猫画画"):
+        pic_word = str(command)[len("/猫猫画画"):]
         current_time = time.time()
         # 判断距离上次请求图片的时间间隔是否小于60秒(1分钟)
         if current_time - last_request_time >= 0:
             send_group_message_word(group_id, "图片正在生成中...")
             last_request_time = current_time
-            send_group_message_pic(group_id, user_id, story_start2(AI_lora_getpic_prompt(pic_word)))
-    elif str(command).startswith("/整点瑟瑟2"):
-        pic_word = str(command)[len("/整点瑟瑟2"):]
+            send_group_message_pic(group_id, user_id, story_start(AI_lora_getpic_prompt(pic_word)))
+    elif str(command).startswith("/整点瑟瑟"):
+        pic_word = str(command)[len("/整点瑟瑟"):]
         current_time = time.time()
         # 判断距离上次请求图片的时间间隔是否小于60秒(1分钟)
         if group_id == 637612718:
@@ -137,9 +139,7 @@ def on_command(group_id, user_id,command):
             if current_time - last_request_time >= 0:
                 send_group_message_word(group_id, "图片正在生成中...")
                 last_request_time = current_time
-                message_id = \
-                send_group_message_pic(group_id, user_id, story_start2("nsfw,best quality,masterpiece," + AI_lora_getpic_prompt(pic_word)))[
-                    'data']['message_id']
+                message_id = send_group_message_pic(group_id, user_id, story_start("nsfw,"+AI_lora_getpic_prompt(pic_word)))['data']['message_id']
                 time.sleep(12)
                 del_group_message(message_id)
     elif str(command).startswith("/语音合成"):