Luma API 快速开始

class LumaAPI {
  constructor(apiKey) {
    this.apiKey = apiKey;
    this.baseUrl = 'https://api.kie.ai/api/v1/modify';
  }
  
  async modifyVideo(prompt, videoUrl, options = {}) {
    const response = await fetch(`${this.baseUrl}/generate`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${this.apiKey}`,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({
        prompt,
        videoUrl,
        callBackUrl: options.callBackUrl,
        watermark: options.watermark,
        ...options
      })
    });
    
    const result = await response.json();
    if (result.code !== 200) {
      throw new Error(`生成失败: ${result.msg}`);
    }
    
    return result.data.taskId;
  }
  
  async waitForCompletion(taskId, maxWaitTime = 900000) { // 15 分钟最大
    const startTime = Date.now();
    
    while (Date.now() - startTime < maxWaitTime) {
      const status = await this.getTaskStatus(taskId);
      
      if (status.successFlag === 1) {
        return status.response;
      } else if (status.successFlag === 2 || status.successFlag === 3) {
        throw new Error(`生成失败: ${status.errorMessage || '未知错误'}`);
      }
      
      // 下次检查前等待 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 function main() {
  const api = new LumaAPI('YOUR_API_KEY');
  
  try {
    // 视频修改
    console.log('开始视频修改...');
    const taskId = await api.modifyVideo(
      'A futuristic cityscape at night with towering glass spires reaching into a starry sky. Neon lights in blue and purple illuminate the buildings while flying vehicles glide silently between the structures. Holographic advertisements flicker and change on building facades.',
      'https://example.com/input-video.mp4',
      { 
        callBackUrl: 'https://your-callback-url.com/luma-callback',
        watermark: 'your-watermark-id'
      }
    );
    
    // 等待完成
    console.log(`任务 ID: ${taskId}。等待完成...`);
    const result = await api.waitForCompletion(taskId);
    
    console.log('视频修改成功！');
    console.log('结果视频 URL:', result.resultUrls);
    console.log('原始视频 URL:', result.originUrls);
    
  } catch (error) {
    console.error('错误:', error.message);
  }
}

main();