import requests
import base64
from pathlib import Path
 
API_BASE = "https://camel.kr777.top/v1"
API_KEY = "sk-your-api-key"  # 替换为你的 API Key
 
def generate_image(prompt: str, output_file: str = "output.png"):
    """生成图片并保存"""
    response = requests.post(
        f"{API_BASE}/images/generations",
        headers={
            "Authorization": f"Bearer {API_KEY}",
            "Content-Type": "application/json"
        },
        json={
            "model": "gpt-image-2",
            "prompt": prompt,
            "n": 1,
            "size": "1024x1024"
        },
        timeout=120
    )
 
    response.raise_for_status()
    result = response.json()
 
    image_b64 = result['data'][0]['b64_json']
    image_data = base64.b64decode(image_b64)
    Path(output_file).write_bytes(image_data)
 
    print(f"✅ 图片已保存到: {output_file}")
    return result
 
generate_image(
    prompt="A cute cat sitting on a laptop, digital art style",
    output_file="cat_on_laptop.png"
)

const axios = require('axios');
const fs = require('fs');
 
const API_BASE = 'https://camel.kr777.top/v1';
const API_KEY = 'sk-your-api-key';  // 替换为你的 API Key
 
async function generateImage(prompt, outputFile = 'output.png') {
  try {
    const response = await axios.post(
      `${API_BASE}/images/generations`,
      {
        model: 'gpt-image-2',
        prompt: prompt,
        n: 1,
        size: '1024x1024'
      },
      {
        headers: {
          'Authorization': `Bearer ${API_KEY}`,
          'Content-Type': 'application/json'
        },
        timeout: 120000
      }
    );
 
    const imageB64 = response.data.data[0].b64_json;
    const imageBuffer = Buffer.from(imageB64, 'base64');
    fs.writeFileSync(outputFile, imageBuffer);
 
    console.log(`✅ 图片已保存到: ${outputFile}`);
    return response.data;
  } catch (error) {
    console.error('生成失败:', error.response?.data || error.message);
    throw error;
  }
}
 
generateImage(
  'A cute cat sitting on a laptop, digital art style',
  'cat_on_laptop.png'
);

curl -X POST https://camel.kr777.top/v1/images/generations \
  -H "Authorization: Bearer sk-your-api-key" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-image-2",
    "prompt": "A cute cat sitting on a laptop",
    "n": 1,
    "size": "1024x1024"
  }' \
  | jq -r '.data[0].b64_json' \
  | base64 -d > output.png