音频分离回调

from flask import Flask, request, jsonify
import requests

app = Flask(__name__)

@app.route('/vocal-separation-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}, 消息: {msg}")
    
    if code == 200:
        # 任务成功完成
        print("人声分离完成")
        
        if vocal_info:
            print("分离结果:")
            print(f"  原始音频: {vocal_info.get('origin_url')}")
            
            # 根据不同的分离类型处理不同的文件
            if vocal_info.get('instrumental_url'):
                # separate_vocal 类型
                print(f"  纯伴奏: {vocal_info.get('instrumental_url')}")
                print(f"  纯人声: {vocal_info.get('vocal_url')}")
            else:
                # split_stem 类型
                print(f"  人声: {vocal_info.get('vocal_url')}")
                print(f"  和声: {vocal_info.get('backing_vocals_url')}")
                print(f"  鼓声: {vocal_info.get('drums_url')}")
                print(f"  贝斯: {vocal_info.get('bass_url')}")
                print(f"  吉他: {vocal_info.get('guitar_url')}")
                print(f"  键盘: {vocal_info.get('keyboard_url')}")
                print(f"  打击乐: {vocal_info.get('percussion_url')}")
                print(f"  弦乐: {vocal_info.get('strings_url')}")
                print(f"  合成器: {vocal_info.get('synth_url')}")
                print(f"  音效: {vocal_info.get('fx_url')}")
                print(f"  铜管乐: {vocal_info.get('brass_url')}")
                print(f"  木管乐: {vocal_info.get('woodwinds_url')}")
            
            # 下载分离后的音频文件
            def download_file(url, filename):
                if not url:
                    return
                    
                try:
                    response = requests.get(url)
                    if response.status_code == 200:
                        with open(filename, "wb") as f:
                            f.write(response.content)
                        print(f"已保存: {filename}")
                except Exception as e:
                    print(f"下载失败 {filename}: {e}")
            
            # 下载所有可用的音频文件
            for key, url in vocal_info.items():
                if url and key.endswith('_url'):
                    filename = f"{task_id}_{key.replace('_url', '')}.mp3"
                    download_file(url, filename)
                
    else:
        # 任务失败
        print(f"人声分离失败: {msg}")
        
        # 处理失败情况...
        if code == 400:
            print("参数错误或音频格式不支持")
        elif code == 451:
            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'] ?? null;

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

if ($code === 200) {
    // 任务成功完成
    error_log("人声分离完成");
    
    if ($vocalInfo) {
        error_log("分离结果:");
        error_log("  原始音频: " . ($vocalInfo['origin_url'] ?? ''));
        
        // 根据不同的分离类型处理不同的文件
        if (isset($vocalInfo['instrumental_url'])) {
            // separate_vocal 类型
            error_log("  纯伴奏: " . ($vocalInfo['instrumental_url'] ?? ''));
            error_log("  纯人声: " . ($vocalInfo['vocal_url'] ?? ''));
        } else {
            // split_stem 类型
            error_log("  人声: " . ($vocalInfo['vocal_url'] ?? ''));
            error_log("  和声: " . ($vocalInfo['backing_vocals_url'] ?? ''));
            error_log("  鼓声: " . ($vocalInfo['drums_url'] ?? ''));
            error_log("  贝斯: " . ($vocalInfo['bass_url'] ?? ''));
            error_log("  吉他: " . ($vocalInfo['guitar_url'] ?? ''));
            error_log("  键盘: " . ($vocalInfo['keyboard_url'] ?? ''));
            error_log("  打击乐: " . ($vocalInfo['percussion_url'] ?? ''));
            error_log("  弦乐: " . ($vocalInfo['strings_url'] ?? ''));
            error_log("  合成器: " . ($vocalInfo['synth_url'] ?? ''));
            error_log("  音效: " . ($vocalInfo['fx_url'] ?? ''));
            error_log("  铜管乐: " . ($vocalInfo['brass_url'] ?? ''));
            error_log("  木管乐: " . ($vocalInfo['woodwinds_url'] ?? ''));
        }
        
        // 下载分离后的音频文件
        function downloadFile($url, $filename) {
            if (!$url) return;
            
            try {
                $content = file_get_contents($url);
                if ($content !== false) {
                    file_put_contents($filename, $content);
                    error_log("已保存: $filename");
                }
            } catch (Exception $e) {
                error_log("下载失败 $filename: " . $e->getMessage());
            }
        }
        
        // 下载所有可用的音频文件
        foreach ($vocalInfo as $key => $url) {
            if ($url && strpos($key, '_url') !== false) {
                $filename = $taskId . '_' . str_replace('_url', '', $key) . '.mp3';
                downloadFile($url, $filename);
            }
        }
    }
    
} else {
    // 任务失败
    error_log("人声分离失败: $msg");
    
    // 处理失败情况...
    if ($code === 400) {
        error_log("参数错误或音频格式不支持");
    } elseif ($code === 451) {
        error_log("源音频文件下载失败");
    } elseif ($code === 500) {
        error_log("服务器内部错误");
    }
}

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