Audio Separation Callbacks

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

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}")
    
    if code == 200 and vocal_info:
        # Vocal separation successful
        print("Vocal separation completed successfully")
        
        # Determine separation type based on callback content
        if vocal_info.get('instrumental_url'):
            # separate_vocal type
            audio_types = [
                ('original', vocal_info.get('origin_url')),
                ('vocal', vocal_info.get('vocal_url')),
                ('instrumental', vocal_info.get('instrumental_url'))
            ]
        else:
            # split_stem type
            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'))
            ]
        
        # Download all separated audio files
        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)
                        
                        # Check file size
                        file_size = os.path.getsize(filename)
                        print(f"{audio_name} audio downloaded as {filename}, size: {file_size} bytes")
                        
                except Exception as e:
                    print(f"{audio_name} audio download failed: {e}")
        
    else:
        # Task failed
        print(f"Vocal separation failed: {msg}")
        
        # Handle failure cases...
        if code == 400:
            print("Invalid source audio or parameter error")
        elif code == 429:
            print("Insufficient credits")
        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");

if ($code === 200 && $vocalInfo) {
    // Vocal separation successful
    error_log("Vocal separation completed successfully");
    
    // Determine separation type and download corresponding files
    if (isset($vocalInfo['instrumental_url'])) {
        // separate_vocal type
        $audioTypes = [
            'original' => $vocalInfo['origin_url'] ?? '',
            'vocal' => $vocalInfo['vocal_url'] ?? '',
            'instrumental' => $vocalInfo['instrumental_url'] ?? ''
        ];
    } else {
        // split_stem type
        $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'] ?? ''
        ];
    }
    
    // Download all separated audio files
    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);
                    
                    // Check file size
                    $fileSize = filesize($filename);
                    error_log("$audioName audio downloaded as $filename, size: $fileSize bytes");
                }
            } catch (Exception $e) {
                error_log("$audioName audio download failed: " . $e->getMessage());
            }
        }
    }
    
} else {
    // Task failed
    error_log("Vocal separation failed: $msg");
    
    // Handle failure cases...
    if ($code === 400) {
        error_log("Invalid source audio or parameter error");
    } elseif ($code === 429) {
        error_log("Insufficient credits");
    } 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']);
?>