diff --git a/BaiHu/Temp/3142b1aa-b277-81c0-cad6-d18947b8c78e.jpg b/BaiHu/Temp/3142b1aa-b277-81c0-cad6-d18947b8c78e.jpg
new file mode 100644
index 00000000..5daf69f5
Binary files /dev/null and b/BaiHu/Temp/3142b1aa-b277-81c0-cad6-d18947b8c78e.jpg differ
diff --git a/BaiHu/Temp/42b811f1-948e-4763-0977-21457b592197.jpg b/BaiHu/Temp/42b811f1-948e-4763-0977-21457b592197.jpg
new file mode 100644
index 00000000..1311225d
Binary files /dev/null and b/BaiHu/Temp/42b811f1-948e-4763-0977-21457b592197.jpg differ
diff --git a/BaiHu/Temp/75fc78e8-67c6-bad2-1262-b33be1a5d132.jpg b/BaiHu/Temp/75fc78e8-67c6-bad2-1262-b33be1a5d132.jpg
new file mode 100644
index 00000000..7160b7dc
Binary files /dev/null and b/BaiHu/Temp/75fc78e8-67c6-bad2-1262-b33be1a5d132.jpg differ
diff --git a/BaiHu/Temp/8894c9d4-8d75-f36e-02e5-4ec778ba4b49.jpg b/BaiHu/Temp/8894c9d4-8d75-f36e-02e5-4ec778ba4b49.jpg
new file mode 100644
index 00000000..d01f3745
Binary files /dev/null and b/BaiHu/Temp/8894c9d4-8d75-f36e-02e5-4ec778ba4b49.jpg differ
diff --git a/BaiHu/Temp/8979df20-8ab4-5c79-30bf-244dbc5b0d74.jpg b/BaiHu/Temp/8979df20-8ab4-5c79-30bf-244dbc5b0d74.jpg
new file mode 100644
index 00000000..791944c4
Binary files /dev/null and b/BaiHu/Temp/8979df20-8ab4-5c79-30bf-244dbc5b0d74.jpg differ
diff --git a/BaiHu/Temp/93815b89-f37c-80f9-c1a7-364a1618ee9f.jpg b/BaiHu/Temp/93815b89-f37c-80f9-c1a7-364a1618ee9f.jpg
new file mode 100644
index 00000000..283a448c
Binary files /dev/null and b/BaiHu/Temp/93815b89-f37c-80f9-c1a7-364a1618ee9f.jpg differ
diff --git a/BaiHu/Temp/a96fb6fb-e73b-ba80-094c-1d99ae6f84ca.jpg b/BaiHu/Temp/a96fb6fb-e73b-ba80-094c-1d99ae6f84ca.jpg
new file mode 100644
index 00000000..0779773c
Binary files /dev/null and b/BaiHu/Temp/a96fb6fb-e73b-ba80-094c-1d99ae6f84ca.jpg differ
diff --git a/BaiHu/Temp/ac8a5d26-2840-997e-6e11-84be5ddf2898.jpg b/BaiHu/Temp/ac8a5d26-2840-997e-6e11-84be5ddf2898.jpg
new file mode 100644
index 00000000..426c453d
Binary files /dev/null and b/BaiHu/Temp/ac8a5d26-2840-997e-6e11-84be5ddf2898.jpg differ
diff --git a/BaiHu/Temp/b3ed1920-cd78-f39f-9529-966d4ee422a8.jpg b/BaiHu/Temp/b3ed1920-cd78-f39f-9529-966d4ee422a8.jpg
new file mode 100644
index 00000000..804acd79
Binary files /dev/null and b/BaiHu/Temp/b3ed1920-cd78-f39f-9529-966d4ee422a8.jpg differ
diff --git a/BaiHu/Temp/bb2e39ce-b797-d522-ad26-9cc778b82efd.jpg b/BaiHu/Temp/bb2e39ce-b797-d522-ad26-9cc778b82efd.jpg
new file mode 100644
index 00000000..40ca5251
Binary files /dev/null and b/BaiHu/Temp/bb2e39ce-b797-d522-ad26-9cc778b82efd.jpg differ
diff --git a/BaiHu/Temp/bff2e92a-98bf-5b6c-b58a-ce56d9bae436.jpg b/BaiHu/Temp/bff2e92a-98bf-5b6c-b58a-ce56d9bae436.jpg
new file mode 100644
index 00000000..eebca285
Binary files /dev/null and b/BaiHu/Temp/bff2e92a-98bf-5b6c-b58a-ce56d9bae436.jpg differ
diff --git a/BaiHu/Temp/c3471e7b-282f-5e77-2270-b66a3188f9c2.jpg b/BaiHu/Temp/c3471e7b-282f-5e77-2270-b66a3188f9c2.jpg
new file mode 100644
index 00000000..d1ae9449
Binary files /dev/null and b/BaiHu/Temp/c3471e7b-282f-5e77-2270-b66a3188f9c2.jpg differ
diff --git a/BaiHu/Temp/d593662f-abea-4e4f-57c2-16c5cff84022.jpg b/BaiHu/Temp/d593662f-abea-4e4f-57c2-16c5cff84022.jpg
new file mode 100644
index 00000000..9be45dc2
Binary files /dev/null and b/BaiHu/Temp/d593662f-abea-4e4f-57c2-16c5cff84022.jpg differ
diff --git a/BaiHu/Temp/ea202c69-5fa1-977d-3265-99ffb9bb2301.jpg b/BaiHu/Temp/ea202c69-5fa1-977d-3265-99ffb9bb2301.jpg
new file mode 100644
index 00000000..bacfdf73
Binary files /dev/null and b/BaiHu/Temp/ea202c69-5fa1-977d-3265-99ffb9bb2301.jpg differ
diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index 46962b4a..b4e7e533 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -129,7 +129,7 @@ def runComfyUI(model_id, json_data, input_image, target_folder):
 
 
 # 4、风格转绘
-def webui_convert(model_id, json_data, input_image, target_folder):
+def webui_convert(model_id, prompt_id, json_data, input_image, target_folder):
     # 1、需要反推提示词
     prompt = get_wd_14(wd_url, input_image[0])
 
@@ -140,18 +140,21 @@ def webui_convert(model_id, json_data, input_image, target_folder):
     img.close()
 
     # 填充反推的提示词，以原图片的宽高为参数
-    json_data['prompt'] = prompt+json_data['prompt']
+    json_data['prompt'] = prompt + json_data['prompt']
     json_data['width'] = width
     json_data['height'] = height
 
     # 图片
-    args = json_data['alwayson_scripts']['ControlNet']['args']
-    args[0]['input_image'] = encode_image(input_image[0])
+    json_data['init_images'] = [encode_image(input_image[0])]
 
     # 开始生成图片
     printf("开始生成图片...")
     # 调用图生图服务
     response = submit_post(img2img_url, json_data)
+
+    if response.status_code == 500:
+        print(response.text)
+        exit(0)
     # 创建文件的完整路径
     file_path = os.path.join(target_folder, str(uuid.uuid4()) + '.png')
     save_encoded_image(response.json()['images'][0], file_path)