GPT Image 2 VIP

import requests
import base64

# Configuration
API_KEY = "sk-YOUR_API_KEY"  # Replace with your API Key
API_URL = "https://api2.laozhang.ai/v1/images/generations"

# Request headers
headers = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json"
}

# Generate request
payload = {
    "model": "gpt-image-2-vip",
    "prompt": "极简咖啡海报，准确显示标题“COFFEE MADE SIMPLE”，高对比排版，温暖自然光，干净商业版式。",
    "size": "2048x2048",
    "quality": "high"
}

print("Generating image...")
response = requests.post(API_URL, headers=headers, json=payload, timeout=180)

if response.status_code != 200:
    print(f"Error: {response.status_code} - {response.text}")
    exit(1)

result = response.json()
image_base64 = result.get("data", [{}])[0].get("b64_json")
image_url = result.get("data", [{}])[0].get("url")

if not image_base64 and image_url:
    image_response = requests.get(image_url, timeout=60)
    image_response.raise_for_status()
    image_base64 = base64.b64encode(image_response.content).decode("utf-8")

if not image_base64:
    print("No image data in response")
    print(result)
    exit(1)

with open("output.png", "wb") as f:
    f.write(base64.b64decode(image_base64))

print("Image saved: output.png")