import requests
import time

# 1. Submit Task
headers = {"Authorization": "Bearer YOUR_KEY"}
resp = requests.post(
    "https://api.apipod.ai/v1/images/generations",
    headers=headers,
    json={
        "model": "gemini-3-pro-image-preview",
        "prompt": "A futuristic banana..."
    }
)
task_id = resp.json()["data"]["task_id"]

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

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

    time.sleep(1)