wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-18 18:30:19 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Update ideogram nodes to v3

Browse Source
This commit is contained in:
doombubbles 2025-05-01 11:44:23 -07:00
parent c9598ec26d
commit 6a06706d00
1 changed files with 15 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
comfy_extras/nodes/nodes_ideogram.py
View File

@ -62,22 +62,20 @@ class IdeogramGenerate(CustomNode):
headers = {"Api-Key": api_key, "Content-Type": "application/json"} headers = {"Api-Key": api_key, "Content-Type": "application/json"}
payload = { payload = {
"image_request": { "prompt": prompt,
"prompt": prompt, "resolution": resolution,
"resolution": resolution, "model": model,
"model": model, "magic_prompt": magic_prompt_option,
"magic_prompt_option": magic_prompt_option, "num_images": num_images,
"num_images": num_images, "style_type": style_type,
"style_type": style_type,
}
} }
if negative_prompt: if negative_prompt:
payload["image_request"]["negative_prompt"] = negative_prompt payload["negative_prompt"] = negative_prompt
if seed: if seed:
payload["image_request"]["seed"] = seed payload["seed"] = seed
response = requests.post("https://api.ideogram.ai/generate", headers=headers, json=payload) response = requests.post("https://api.ideogram.ai/v1/ideogram-v3/generate", headers=headers, json=payload)
response.raise_for_status() response.raise_for_status()
images = [] images = []
@ -133,21 +131,19 @@ class IdeogramEdit(CustomNode):
mask_pil.save(mask_bytes, format="PNG") mask_pil.save(mask_bytes, format="PNG")
files = { files = {
"image_file": ("image.png", image_bytes.getvalue()), "image": ("image.png", image_bytes.getvalue()),
"mask": ("mask.png", mask_bytes.getvalue()), "mask": ("mask.png", mask_bytes.getvalue()),
} }
data = { data = {
"prompt": prompt, "prompt": prompt,
"model": model, "magic_prompt": magic_prompt_option,
"magic_prompt_option": magic_prompt_option,
"num_images": num_images, "num_images": num_images,
"style_type": style_type,
} }
if seed: if seed:
data["seed"] = seed data["seed"] = seed
response = requests.post("https://api.ideogram.ai/edit", headers=headers, files=files, data=data) response = requests.post("https://api.ideogram.ai/v1/ideogram-v3/edit", headers=headers, files=files, data=data)
response.raise_for_status() response.raise_for_status()
for item in response.json()["data"]: for item in response.json()["data"]:
@ -198,15 +194,14 @@ class IdeogramRemix(CustomNode):
image_pil.save(image_bytes, format="PNG") image_pil.save(image_bytes, format="PNG")
files = { files = {
"image_file": ("image.png", image_bytes.getvalue()), "image": ("image.png", image_bytes.getvalue()),
} }
data = { data = {
"prompt": prompt, "prompt": prompt,
"resolution": resolution, "resolution": resolution,
"model": model,
"image_weight": image_weight, "image_weight": image_weight,
"magic_prompt_option": magic_prompt_option, "magic_prompt": magic_prompt_option,
"num_images": num_images, "num_images": num_images,
"style_type": style_type, "style_type": style_type,
} }
@ -218,9 +213,7 @@ class IdeogramRemix(CustomNode):
# data = {"image_request": data} # data = {"image_request": data}
response = requests.post("https://api.ideogram.ai/remix", headers=headers, files=files, data={ response = requests.post("https://api.ideogram.ai/v1/ideogram-v3/remix", headers=headers, files=files, data=data)
"image_request": json.dumps(data)
})
response.raise_for_status() response.raise_for_status()
for item in response.json()["data"]: for item in response.json()["data"]: