import time
import requests
base_url = "https://smew.ai/v1"
api_key = "YOUR_API_KEY"
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json",
}
submit = requests.post(
f"{base_url}/images/generations",
headers=headers,
json={
"model": "gpt-image-2",
"prompt": "一只可爱的猫咪在阳光下打盹",
"n": 1,
"size": "1:1",
"resolution": "2k",
},
)
submit.raise_for_status()
task_id = submit.json()["data"][0]["task_id"]
print(f"任务已提交: {task_id}")
while True:
result = requests.get(
f"{base_url}/tasks/{task_id}",
headers={"Authorization": f"Bearer {api_key}"},
)
result.raise_for_status()
data = result.json()["data"]
status = data["status"]
if status == "completed":
image_url = data["result"]["images"][0]["url"][0]
print(f"图片生成完成: {image_url}")
break
if status == "failed":
raise RuntimeError(data.get("error", {}).get("message", "任务失败"))
print(f"任务处理中... 进度: {data.get('progress', 0)}%")
time.sleep(5)