音频分离回调

const express = require('express');
const https = require('https');
const fs = require('fs');
const path = require('path');
const app = express();

app.use(express.json());

app.post('/vocal-removal-callback', (req, res) => {
  const { code, msg, data } = req.body;
  
  console.log('收到人声分离回调:', {
    taskId: data.task_id,
    status: code,
    message: msg
  });
  
  if (code === 200) {
    // 分离成功
    console.log('人声分离完成');
    const vocalInfo = data.vocal_removal_info;
    
    // 根据回调内容判断分离类型并下载相应文件
    let audioTypes = [];
    
    if (vocalInfo.instrumental_url) {
      // separate_vocal 类型
      audioTypes = [
        { name: 'original', url: vocalInfo.origin_url },
        { name: 'vocal', url: vocalInfo.vocal_url },
        { name: 'instrumental', url: vocalInfo.instrumental_url }
      ];
    } else {
      // split_stem 类型
      audioTypes = [
        { name: 'original', url: vocalInfo.origin_url },
        { name: 'vocal', url: vocalInfo.vocal_url },
        { name: 'backing_vocals', url: vocalInfo.backing_vocals_url },
        { name: 'drums', url: vocalInfo.drums_url },
        { name: 'bass', url: vocalInfo.bass_url },
        { name: 'guitar', url: vocalInfo.guitar_url },
        { name: 'keyboard', url: vocalInfo.keyboard_url },
        { name: 'percussion', url: vocalInfo.percussion_url },
        { name: 'strings', url: vocalInfo.strings_url },
        { name: 'synth', url: vocalInfo.synth_url },
        { name: 'fx', url: vocalInfo.fx_url },
        { name: 'brass', url: vocalInfo.brass_url },
        { name: 'woodwinds', url: vocalInfo.woodwinds_url }
      ];
    }
    
    // 下载所有分离后的音频文件
    audioTypes.forEach(audio => {
      if (audio.url) {
        const filename = `${data.task_id}_${audio.name}.mp3`;
        const file = fs.createWriteStream(filename);
        
        https.get(audio.url, (response) => {
          response.pipe(file);
          
          file.on('finish', () => {
            file.close();
            console.log(`${audio.name} 音频已下载为 ${filename}`);
          });
        }).on('error', (err) => {
          console.error(`${audio.name} 音频下载失败:`, err.message);
        });
      }
    });
    
  } else {
    // 分离失败
    console.log('人声分离失败:', msg);
    
    // 处理失败情况...
    if (code === 400) {
      console.log('源音频无效或参数错误');
    } else if (code === 429) {
      console.log('积分不足');
    } else if (code === 500) {
      console.log('服务器内部错误');
    }
  }
  
  // 返回200状态码确认收到回调
  res.status(200).json({ status: 'received' });
});

app.listen(3000, () => {
  console.log('回调服务器运行在端口 3000');
});