from imgapi_docs_client_20261029 import ImgApiClient, ImgApiError

try:
    with ImgApiClient() as client:  # 从环境变量读取 Key
        result = client.generate_image(
            model="gpt-image-2",
            prompt="白底运动鞋商品主图，柔和棚拍光线",
            resolution="1K",
            aspect_ratio="1:1",
            on_task_created=lambda task_id: print("任务 ID：", task_id),
        )
        for image_url in result.image_urls:
            print("图片地址：", image_url)
except ImgApiError as error:
    print(error)
    if error.task_id:
        print("请保存任务 ID，稍后查询：", error.task_id)
    raise SystemExit(1)  # 不自动重新提交