generate-image.py

#!/usr/bin/env python3
"""
AIGate API — Generate Image (Python)
Docs: https://aigatecloud.com/docs/

Install: pip install requests python-dotenv
Run:     python examples/generate-image.py
"""

import os
import sys
import requests
from dotenv import load_dotenv

load_dotenv()

API_KEY  = os.getenv("AIGATE_KEY", "sk-aigate-YOUR_KEY_HERE")
BASE_URL = os.getenv("AIGATE_BASE_URL", "https://api.aigatecloud.com/v1")


def generate_image(prompt: str, aspect_ratio: str = "9:16", use_case: str = "viral_background", style: str = "luxury") -> dict:
    resp = requests.post(
        f"{BASE_URL}/generate/image",
        json={
            "prompt":       prompt,
            "aspect_ratio": aspect_ratio,
            "use_case":     use_case,
            "style":        style,
        },
        headers={
            "Authorization": f"Bearer {API_KEY}",
            "Content-Type":  "application/json",
        },
        timeout=60,
    )

    data = resp.json()

    if not resp.ok or not data.get("ok"):
        error = data.get("error", f"HTTP {resp.status_code}")
        if resp.status_code == 401:
            print(f"Error: {error}")
            print("→ Register free at https://aigatecloud.com to get a real API key.")
        else:
            print(f"Error: {error}")
        sys.exit(1)

    return data


if __name__ == "__main__":
    result = generate_image(
        prompt="luxury lifestyle, gold tones, aspirational mood, social media ready",
        aspect_ratio="9:16",
        use_case="viral_background",
        style="luxury",
    )

    asset = result["asset"]
    print("✓ Image generated!")
    print(f"  Source:  {result['source']} ({result['provider']})")
    print(f"  URL:     {asset['url']}")
    print(f"  Size:    {asset.get('width', '?')}×{asset.get('height', '?')}")
    print(f"  Credits: −{result['credits_charged']} (balance: {result['credits_balance']})")