import requests
import time

API_BASE = "https://api.apipod.ai/v1"
headers = {"Authorization": "Bearer $API_KEY"}

# 1. Submit generation task
resp = requests.post(
    "https://api.apipod.ai/v1/images/generations",
    headers=headers,
    json={
        "model": "gpt-image-2",
        "prompt": "A professional menu design...",
        "quality": "1K",
        "aspect_ratio": "1:1"
    }
)
task_id = resp.json()["data"]["task_id"]

# 2. Poll status until completed
while True:
    res = requests.get(
        f"https://api.apipod.ai/v1/images/status/{task_id}",
        headers=headers
    ).json()

    if res["data"]["status"] == "completed":
        print(res["data"]["result"][0])
        break

    time.sleep(2)