xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-09 18:27:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix API type error and add b64 support for 4o

Browse Source
This commit is contained in:
thot-experiment 2025-04-23 11:35:43 -07:00
parent f65b05e865
commit cf209aa456
2 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_api_nodes/apis/__init__.py
View File

@ -573,7 +573,7 @@ class Quality(str, Enum):
class OpenAIImageEditRequest(BaseModel):
background: Optional[Background] = Field(
background: Optional[str] = Field(
None, description='Background transparency', examples=['opaque']
)
model: str = Field(
@ -596,7 +596,7 @@ class OpenAIImageEditRequest(BaseModel):
description='A text description of the desired edit',
examples=['Give the rocketship rainbow coloring'],
)
quality: Optional[Quality] = Field(
quality: Optional[str] = Field(
None, description='The quality of the edited image', examples=['low']
)
size: Optional[str] = Field(

20
comfy_api_nodes/nodes_api.py
View File

@ -15,6 +15,7 @@ from PIL import Image
import requests
import torch
import math
import base64
def downscale_input(image):
samples = image.movedim(-1,1)
@ -38,13 +39,20 @@ def validate_and_cast_response (response):
# Get base64 image data
image_url = data[0].url
if not image_url:
raise Exception("No image URL was generated in the response")
img_response = requests.get(image_url)
if img_response.status_code != 200:
raise Exception("Failed to download the image")
b64_data = data[0].b64_json
if not image_url and not b64_data:
raise Exception("No image was generated in the response")
if b64_data:
img_data = base64.b64decode(b64_data)
img = Image.open(io.BytesIO(img_data))
elif image_url:
img_response = requests.get(image_url)
if img_response.status_code != 200:
raise Exception("Failed to download the image")
img = Image.open(io.BytesIO(img_response.content))
img = Image.open(io.BytesIO(img_response.content))
img = img.convert("RGB") # Ensure RGB format
# Convert to numpy array, normalize to float32 between 0 and 1