Skip to content

Commit 7d14769

Browse files
committed
feat: enhance image URL handling in base_image_generate_node.py to support data URLs
1 parent 59ff899 commit 7d14769

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,18 @@ def execute(self, model_id, prompt, negative_prompt, dialogue_number, dialogue_t
4242
file_urls = []
4343
for image_url in image_urls:
4444
file_name = 'generated_image.png'
45-
if isinstance(image_url, str) and image_url.startswith('http'):
46-
image_url = requests.get(image_url).content
45+
if isinstance(image_url, str):
46+
if image_url.startswith('http'):
47+
# HTTP URL 情况
48+
image_url = requests.get(image_url).content
49+
elif image_url.startswith('data:image'):
50+
# Data URL 格式 (data:image/png;base64,...)
51+
import base64
52+
header, encoded = image_url.split(',', 1)
53+
image_url = base64.b64decode(encoded)
54+
else:
55+
import base64
56+
image_url = base64.b64decode(image_url)
4757
file = bytes_to_uploaded_file(image_url, file_name)
4858
file_url = self.upload_file(file)
4959
file_urls.append(file_url)

0 commit comments

Comments
 (0)