Audio Separation Callbacks

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"Received vocal separation callback: {task_id}, status: {code}, message: {msg}")
    
    if code == 200:
        # Task completed successfully
        print("Vocal separation completed")
        
        if vocal_info:
            print("Separation results:")
            print(f"  Original audio: {vocal_info.get('origin_url')}")
            
            # Handle different separation types
            if vocal_info.get('instrumental_url'):
                # separate_vocal type
                print(f"  Instrumental only: {vocal_info.get('instrumental_url')}")
                print(f"  Vocals only: {vocal_info.get('vocal_url')}")
            else:
                # split_stem type
                print(f"  Vocals: {vocal_info.get('vocal_url')}")
                print(f"  Backing vocals: {vocal_info.get('backing_vocals_url')}")
                print(f"  Drums: {vocal_info.get('drums_url')}")
                print(f"  Bass: {vocal_info.get('bass_url')}")
                print(f"  Guitar: {vocal_info.get('guitar_url')}")
                print(f"  Keyboard: {vocal_info.get('keyboard_url')}")
                print(f"  Percussion: {vocal_info.get('percussion_url')}")
                print(f"  Strings: {vocal_info.get('strings_url')}")
                print(f"  Synthesizer: {vocal_info.get('synth_url')}")
                print(f"  Effects: {vocal_info.get('fx_url')}")
                print(f"  Brass: {vocal_info.get('brass_url')}")
                print(f"  Woodwinds: {vocal_info.get('woodwinds_url')}")
            
            # Download separated audio files
            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"Saved: {filename}")
                except Exception as e:
                    print(f"Download failed {filename}: {e}")
            
            # Download all available audio files
            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:
        # Task failed
        print(f"Vocal separation failed: {msg}")
        
        # Handle failure cases...
        if code == 400:
            print("Parameter error or unsupported audio format")
        elif code == 451:
            print("Source audio file download failed")
        elif code == 500:
            print("Server internal error")
    
    # Return 200 status code to confirm callback received
    return jsonify({'status': 'received'}), 200

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

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

// Get POST data
$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("Received vocal separation callback: $taskId, status: $code, message: $msg");

if ($code === 200) {
    // Task completed successfully
    error_log("Vocal separation completed");
    
    if ($vocalInfo) {
        error_log("Separation results:");
        error_log("  Original audio: " . ($vocalInfo['origin_url'] ?? ''));
        
        // Handle different separation types
        if (isset($vocalInfo['instrumental_url'])) {
            // separate_vocal type
            error_log("  Instrumental only: " . ($vocalInfo['instrumental_url'] ?? ''));
            error_log("  Vocals only: " . ($vocalInfo['vocal_url'] ?? ''));
        } else {
            // split_stem type
            error_log("  Vocals: " . ($vocalInfo['vocal_url'] ?? ''));
            error_log("  Backing vocals: " . ($vocalInfo['backing_vocals_url'] ?? ''));
            error_log("  Drums: " . ($vocalInfo['drums_url'] ?? ''));
            error_log("  Bass: " . ($vocalInfo['bass_url'] ?? ''));
            error_log("  Guitar: " . ($vocalInfo['guitar_url'] ?? ''));
            error_log("  Keyboard: " . ($vocalInfo['keyboard_url'] ?? ''));
            error_log("  Percussion: " . ($vocalInfo['percussion_url'] ?? ''));
            error_log("  Strings: " . ($vocalInfo['strings_url'] ?? ''));
            error_log("  Synthesizer: " . ($vocalInfo['synth_url'] ?? ''));
            error_log("  Effects: " . ($vocalInfo['fx_url'] ?? ''));
            error_log("  Brass: " . ($vocalInfo['brass_url'] ?? ''));
            error_log("  Woodwinds: " . ($vocalInfo['woodwinds_url'] ?? ''));
        }
        
        // Download separated audio files
        function downloadFile($url, $filename) {
            if (!$url) return;
            
            try {
                $content = file_get_contents($url);
                if ($content !== false) {
                    file_put_contents($filename, $content);
                    error_log("Saved: $filename");
                }
            } catch (Exception $e) {
                error_log("Download failed $filename: " . $e->getMessage());
            }
        }
        
        // Download all available audio files
        foreach ($vocalInfo as $key => $url) {
            if ($url && strpos($key, '_url') !== false) {
                $filename = $taskId . '_' . str_replace('_url', '', $key) . '.mp3';
                downloadFile($url, $filename);
            }
        }
    }
    
} else {
    // Task failed
    error_log("Vocal separation failed: $msg");
    
    // Handle failure cases...
    if ($code === 400) {
        error_log("Parameter error or unsupported audio format");
    } elseif ($code === 451) {
        error_log("Source audio file download failed");
    } elseif ($code === 500) {
        error_log("Server internal error");
    }
}

// Return 200 status code to confirm callback received
http_response_code(200);
echo json_encode(['status' => 'received']);
?>