import requests

response = requests.post(
    "https://connect.sivi.ai/api/prod/v2/general/designs-from-prompt",
    headers={"sivi-api-key": "YOUR_API_KEY", "Content-Type": "application/json"},
    json={
        "type": "displayAds",
        "subtype": "displayAds-half-page-ad",
        "dimension": {"width": 300, "height": 600},
        "prompt": "generate a design for my t-shirt shop for summer campaign with 20% off",
        "assets": {
            "images": [],
            "logos": []
        },
        "language": "english",
        "numOfVariants": 4,
        "outputFormat": ["jpg", "png"],
        "settings": {
            "mode": "custom",
            "colors": ["#005C99", "#00BFA6", "#333333", "#F0F0F0"]
        }
    }
)

result = response.json()
design_id = result["body"]["designId"]

# Step 3: Retrieve the generated design variants
variants_response = requests.post(
    "https://connect.sivi.ai/api/prod/v2/general/get-design-variants",
    headers={"sivi-api-key": "YOUR_API_KEY", "Content-Type": "application/json"},
    json={
        "designId": design_id,
        "type": "generated",
        "limit": 10
    }
)

variants = variants_response.json()
for v in variants["body"]["variations"]:
    print(f"Variant: {v['variantId']}")
    print(f"Image: {v['variantImageUrl']}")
    print(f"Edit: {v['variantEditLink']}")