4o Image API 快速开始

class FourOImageAPI {
  constructor(apiKey) {
    this.apiKey = apiKey;
    this.baseUrl = 'https://api.kie.ai/api/v1/gpt4o-image';
  }
  
  async generateImage(options) {
    const response = await fetch(`${this.baseUrl}/generate`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${this.apiKey}`,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(options)
    });
    
    const result = await response.json();
    if (result.code !== 200) {
      throw new Error(`生成失败: ${result.msg}`);
    }
    
    return result.data.taskId;
  }
  
  async waitForCompletion(taskId, maxWaitTime = 300000) { // 最长等待5分钟
    const startTime = Date.now();
    
    while (Date.now() - startTime < maxWaitTime) {
      const status = await this.getTaskStatus(taskId);
      
      if (status.status === 'SUCCESS') {
        return status.response;
      } else if (status.status === 'CREATE_TASK_FAILED' || status.status === 'GENERATE_FAILED') {
        throw new Error(`生成失败: ${status.status}`);
      }
      
      // 等待10秒后再次检查
      await new Promise(resolve => setTimeout(resolve, 10000));
    }
    
    throw new Error('生成超时');
  }
  
  async getTaskStatus(taskId) {
    const response = await fetch(`${this.baseUrl}/record-info?taskId=${taskId}`, {
      headers: {
        'Authorization': `Bearer ${this.apiKey}`
      }
    });
    
    const result = await response.json();
    return result.data;
  }
  
  async getDownloadUrl(imageUrl) {
    const response = await fetch(`${this.baseUrl}/download-url`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${this.apiKey}`,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({ imageUrl })
    });
    
    const result = await response.json();
    return result.data.downloadUrl;
  }
}

// 使用示例
async function main() {
  const api = new FourOImageAPI('YOUR_API_KEY');
  
  try {
    // 文本转图像生成
    console.log('开始生成图像...');
    const taskId = await api.generateImage({
      prompt: '一个未来主义的城市景观，有飞行汽车和霓虹灯，赛博朋克风格',
      size: '16:9',
      nVariants: 2,
      isEnhance: true,
      enableFallback: true
    });
    
    // 等待完成
    console.log(`任务ID: ${taskId}。等待完成...`);
    const result = await api.waitForCompletion(taskId);
    
    console.log('图像生成成功！');
    result.result_urls.forEach((url, index) => {
      console.log(`图像 ${index + 1}: ${url}`);
    });
    
    // 获取下载URL
    const downloadUrl = await api.getDownloadUrl(result.result_urls[0]);
    console.log('下载URL:', downloadUrl);
    
    // 图像编辑示例
    console.log('\n开始图像编辑...');
    const editTaskId = await api.generateImage({
      filesUrl: [result.result_urls[0]],
      prompt: '在天空中添加美丽的彩虹',
      size: '3:2'
    });
    
    const editResult = await api.waitForCompletion(editTaskId);
    console.log('图像编辑成功！');
    console.log('编辑后图像:', editResult.result_urls[0]);
    
  } catch (error) {
    console.error('错误:', error.message);
  }
}

main();