Python 图像生成

from openai import OpenAI

client = OpenAI(api_key="sk-xxx", base_url="https://api.crazyrouter.com/v1")

response = client.images.generate(
    model="dall-e-3",
    prompt="一只穿着宇航服的猫在月球上散步，数字艺术风格",
    size="1024x1024",
    quality="hd",
    n=1
)

image_url = response.data[0].url
print(f"图片地址: {image_url}")

response = client.images.generate(
    model="gpt-image-2",
    prompt="一个现代简约风格的 Logo，主题是人工智能",
    size="1024x1024",
    quality="low",
    output_format="png",
    n=1
)

print(response.data[0].url)

# 图像编辑（需要原图）
response = client.images.edit(
    model="gpt-image-2",
    image=open("original.png", "rb"),
    prompt="把背景改成海滩日落",
    size="1024x1024",
    quality="low"
)

print(response.data[0].url)

import requests

response = client.images.generate(
    model="dall-e-3",
    prompt="赛博朋克风格的城市夜景",
    size="1792x1024"
)

# 下载图片
image_url = response.data[0].url
img_data = requests.get(image_url).content
with open("cyberpunk_city.png", "wb") as f:
    f.write(img_data)
print("图片已保存")