from PIL import Image
from io import BytesIO
import requests
import base64
import os

============ 配置 ============

API_BASE = "https://yibuapi.com"
API_KEY = "sk-xxx"
MODEL = "doubao-seedream-5-0-260128"

input_image_path = "C:\Users\52416\Desktop\pexels-anand-mylsamy-339541556-14023462.jpg"
edit_prompt = "让ta戴上墨镜和帽子"
size = "2048x2048"

==============================

def image_to_data_url(path: str) -> str:
with open(path, "rb") as f:
b64 = base64.b64encode(f.read()).decode("utf-8")
ext = os.path.splitext(path)[1].lower().lstrip(".")
mime_map = {"png": "image/png", "jpg": "image/jpeg", "jpeg": "image/jpeg", "webp": "image/webp"}
mime_type = mime_map.get(ext, "image/png")
return f"data:{mime_type};base64,{b64}"
print(f"正在加载输入图片: {input_image_path}")
input_image = Image.open(input_image_path)
print(f"图片尺寸: {input_image.size}, 模式: {input_image.mode}")

response = requests.post(
f"{API_BASE}/v1/images/generations",
headers={
"Authorization": f"Bearer {API_KEY}",
"Content-Type": "application/json",
},
json={
"model": MODEL,
"prompt": edit_prompt,
"image": image_to_data_url(input_image_path), # data URL 格式
"size": size,
"n": 1,
"response_format": "b64_json",
},
timeout=120,
)

print(f"响应状态码: {response.status_code}")
if response.status_code != 200:
print(f"请求失败: {response.text}")
exit(1)

result = response.json()
for i, img_data in enumerate(result.get("data", [])):
if img_data.get("b64_json"):
image_bytes = base64.b64decode(img_data["b64_json"])
image = Image.open(BytesIO(image_bytes))
save_path = os.path.join(os.getcwd(), f"edited_{i}.png")
image.save(save_path)
print(f"图片已保存: {save_path}, 尺寸: {image.size}, 大小: {len(image_bytes):,} 字节")
elif img_data.get("url"):
print(f"图片URL: {img_data['url']}")

print("完成!")

doubao-seedream-5-0-260128图改图示例代码

============ 配置 ============#

==============================#

============ 配置 ============

==============================