Image Gen Council

import asyncio, aiohttp, base64

IMAGE_COUNCIL = {
    "byteplus-seedream": {
        "url": "https://ark.cn-beijing.volces.com/api/v3/images/generations",
        "model": "Dola-Seedream-5.0-lite",
        "key": "BYTEPLUS_API_KEY"
    },
    "venice-flux": {
        "url": "https://api.venice.ai/api/v1/images/generations",
        "model": "flux-1",
        "key": "VENICE_API_KEY"
    }
}

async def generate_image(prompt: str):
    """Generate images with multiple Image Gen Council models."""
    async def query_one(name, config):
        headers = {"Authorization": f"Bearer {config['key']}", "Content-Type": "application/json"}
        payload = {"model": config["model"], "prompt": prompt, "size": "1024x1024", "n": 1}
        async with aiohttp.ClientSession() as session:
            async with session.post(config["url"], headers=headers, json=payload) as resp:
                return name, await resp.json()

    tasks = [query_one(n, c) for n, c in IMAGE_COUNCIL.items()]
    return dict(await asyncio.gather(*tasks, return_exceptions=True))

result = asyncio.run(generate_image(
    "A futuristic cityscape at sunset, cyberpunk style, highly detailed"
))
for name, resp in result.items():
    if "data" in resp and resp["data"]:
        print(f"\n=== {name} ===")
        print(f"Image URL: {resp['data'][0].get('url', 'N/A')[:100]}")