音频分离回调

from flask import Flask, request, jsonify
import requests
import os

app = Flask(__name__)

@app.route('/vocal-removal-callback', methods=['POST'])
def handle_callback():
    data = request.json
    
    code = data.get('code')
    msg = data.get('msg')
    callback_data = data.get('data', {})
    task_id = callback_data.get('task_id')
    vocal_info = callback_data.get('vocal_removal_info', {})
    
    print(f"收到人声分离回调: {task_id}, 状态: {code}")
    
    if code == 200 and vocal_info:
        # 分离成功
        print("人声分离完成")
        
        # 根据回调内容判断分离类型
        if vocal_info.get('instrumental_url'):
            # separate_vocal 类型
            audio_types = [
                ('original', vocal_info.get('origin_url')),
                ('vocal', vocal_info.get('vocal_url')),
                ('instrumental', vocal_info.get('instrumental_url'))
            ]
        else:
            # split_stem 类型
            audio_types = [
                ('original', vocal_info.get('origin_url')),
                ('vocal', vocal_info.get('vocal_url')),
                ('backing_vocals', vocal_info.get('backing_vocals_url')),
                ('drums', vocal_info.get('drums_url')),
                ('bass', vocal_info.get('bass_url')),
                ('guitar', vocal_info.get('guitar_url')),
                ('keyboard', vocal_info.get('keyboard_url')),
                ('percussion', vocal_info.get('percussion_url')),
                ('strings', vocal_info.get('strings_url')),
                ('synth', vocal_info.get('synth_url')),
                ('fx', vocal_info.get('fx_url')),
                ('brass', vocal_info.get('brass_url')),
                ('woodwinds', vocal_info.get('woodwinds_url'))
            ]
        
        # 下载所有分离后的音频文件
        for audio_name, audio_url in audio_types:
            if audio_url:
                try:
                    response = requests.get(audio_url)
                    if response.status_code == 200:
                        filename = f"{task_id}_{audio_name}.mp3"
                        with open(filename, "wb") as f:
                            f.write(response.content)
                        
                        # 检查文件大小
                        file_size = os.path.getsize(filename)
                        print(f"{audio_name} 音频已下载为 {filename}, 大小: {file_size} bytes")
                        
                except Exception as e:
                    print(f"{audio_name} 音频下载失败: {e}")
                    
    else:
        # 分离失败
        print(f"人声分离失败: {msg}")
        
        # 处理失败情况...
        if code == 400:
            print("源音频无效或参数错误")
        elif code == 429:
            print("积分不足")
        elif code == 500:
            print("服务器内部错误")
    
    # 返回200状态码确认收到回调
    return jsonify({'status': 'received'}), 200

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=3000)

<?php
header('Content-Type: application/json');

// 获取POST数据
$input = file_get_contents('php://input');
$data = json_decode($input, true);

$code = $data['code'] ?? null;
$msg = $data['msg'] ?? '';
$callbackData = $data['data'] ?? [];
$taskId = $callbackData['task_id'] ?? '';
$vocalInfo = $callbackData['vocal_removal_info'] ?? [];

error_log("收到人声分离回调: $taskId, 状态: $code");

if ($code === 200 && !empty($vocalInfo)) {
    // 分离成功
    error_log("人声分离完成");
    
    // 根据回调内容判断分离类型
    if (isset($vocalInfo['instrumental_url'])) {
        // separate_vocal 类型
        $audioTypes = [
            'original' => $vocalInfo['origin_url'] ?? '',
            'vocal' => $vocalInfo['vocal_url'] ?? '',
            'instrumental' => $vocalInfo['instrumental_url'] ?? ''
        ];
    } else {
        // split_stem 类型
        $audioTypes = [
            'original' => $vocalInfo['origin_url'] ?? '',
            'vocal' => $vocalInfo['vocal_url'] ?? '',
            'backing_vocals' => $vocalInfo['backing_vocals_url'] ?? '',
            'drums' => $vocalInfo['drums_url'] ?? '',
            'bass' => $vocalInfo['bass_url'] ?? '',
            'guitar' => $vocalInfo['guitar_url'] ?? '',
            'keyboard' => $vocalInfo['keyboard_url'] ?? '',
            'percussion' => $vocalInfo['percussion_url'] ?? '',
            'strings' => $vocalInfo['strings_url'] ?? '',
            'synth' => $vocalInfo['synth_url'] ?? '',
            'fx' => $vocalInfo['fx_url'] ?? '',
            'brass' => $vocalInfo['brass_url'] ?? '',
            'woodwinds' => $vocalInfo['woodwinds_url'] ?? ''
        ];
    }
    
    // 下载所有分离后的音频文件
    foreach ($audioTypes as $audioName => $audioUrl) {
        if ($audioUrl) {
            try {
                $audioContent = file_get_contents($audioUrl);
                if ($audioContent !== false) {
                    $filename = "{$taskId}_{$audioName}.mp3";
                    file_put_contents($filename, $audioContent);
                    
                    // 检查文件大小
                    $fileSize = filesize($filename);
                    error_log("$audioName 音频已下载为 $filename, 大小: $fileSize bytes");
                }
            } catch (Exception $e) {
                error_log("$audioName 音频下载失败: " . $e->getMessage());
            }
        }
    }
    
} else {
    // 分离失败
    error_log("人声分离失败: $msg");
    
    // 处理失败情况...
    if ($code === 400) {
        error_log("源音频无效或参数错误");
    } elseif ($code === 429) {
        error_log("积分不足");
    } elseif ($code === 500) {
        error_log("服务器内部错误");
    }
}

// 返回200状态码确认收到回调
http_response_code(200);
echo json_encode(['status' => 'received']);
?>