chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: batch zip image input

Browse Source
This commit is contained in:
bennykok 2024-09-14 21:49:17 -07:00
parent 5554c95f44
commit 2d72cd8175
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy-nodes/external_image_batch.py
View File

@ -56,12 +56,7 @@ class ComfyUIDeployExternalImageBatch:
images_list = json.loads(images) # Assuming images is a JSON array string images_list = json.loads(images) # Assuming images is a JSON array string
print(images_list) print(images_list)
for img_input in images_list: for img_input in images_list:
if img_input.startswith('http'): if img_input.startswith('http') and img_input.endswith('.zip'):
from io import BytesIO
print("Fetching image from url: ", img_input)
response = requests.get(img_input)
image = Image.open(BytesIO(response.content))
elif img_input.startswith('http') and img_input.endswith('.zip'):
print("Fetching zip file from url: ", img_input) print("Fetching zip file from url: ", img_input)
response = requests.get(img_input) response = requests.get(img_input)
zip_file = zipfile.ZipFile(io.BytesIO(response.content)) zip_file = zipfile.ZipFile(io.BytesIO(response.content))
@ -71,6 +66,11 @@ class ComfyUIDeployExternalImageBatch:
image = Image.open(file) image = Image.open(file)
image = self.process_image(image) image = self.process_image(image)
processed_images.append(image) processed_images.append(image)
elif img_input.startswith('http'):
from io import BytesIO
print("Fetching image from url: ", img_input)
response = requests.get(img_input)
image = Image.open(BytesIO(response.content))
elif img_input.startswith('data:image/png;base64,') or img_input.startswith('data:image/jpeg;base64,') or img_input.startswith('data:image/jpg;base64,'): elif img_input.startswith('data:image/png;base64,') or img_input.startswith('data:image/jpeg;base64,') or img_input.startswith('data:image/jpg;base64,'):
import base64 import base64
from io import BytesIO from io import BytesIO