Add Vocals Callbacks

from flask import Flask, request, jsonify
import requests

app = Flask(__name__)

@app.route('/vocals-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')
    callback_type = callback_data.get('callbackType')
    vocals_data = callback_data.get('data', [])
    
    print(f"Received vocal generation callback: {task_id}, Type: {callback_type}, Status: {code}")
    
    if code == 200:
        # Task successful
        if callback_type == 'text':
            print("Text generation completed")
            
        elif callback_type == 'first':
            print("First vocal track generated")
            if vocals_data:
                track = vocals_data[0]
                print(f"Track title: {track.get('title')}")
                print(f"Vocal audio link: {track.get('audio_url')}")
                print(f"Duration: {track.get('duration')} seconds")
                print(f"Vocals prompt: {track.get('prompt')}")
                
        elif callback_type == 'complete':
            print("All vocal tracks generated")
            print(f"Total {len(vocals_data)} vocal tracks generated:")
            
            for i, track in enumerate(vocals_data):
                print(f"Track {i + 1}: {track.get('title')} - {track.get('audio_url')}")
                print(f"Vocals: {track.get('prompt')}")
                
                # Download vocal audio example
                try:
                    audio_url = track.get('audio_url')
                    if audio_url:
                        response = requests.get(audio_url)
                        if response.status_code == 200:
                            filename = f"vocals_{task_id}_{i + 1}.mp3"
                            with open(filename, "wb") as f:
                                f.write(response.content)
                            print(f"Vocal audio saved as {filename}")
                except Exception as e:
                    print(f"Vocal audio download failed: {e}")
                    
    else:
        # Task failed
        print(f"Vocal generation failed: {msg}")
        
        # Handle failure cases...
        if code == 400:
            print("Parameter error or vocal content violation")
        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'] ?? '';
$callbackType = $callbackData['callbackType'] ?? '';
$vocalsData = $callbackData['data'] ?? [];

error_log("Received vocal generation callback: $taskId, Type: $callbackType, Status: $code");

if ($code === 200) {
    // Task successful
    switch ($callbackType) {
        case 'text':
            error_log("Text generation completed");
            break;
            
        case 'first':
            error_log("First vocal track generated");
            if (!empty($vocalsData)) {
                $track = $vocalsData[0];
                error_log("Track title: " . ($track['title'] ?? ''));
                error_log("Vocal audio link: " . ($track['audio_url'] ?? ''));
                error_log("Duration: " . ($track['duration'] ?? 0) . " seconds");
                error_log("Vocals prompt: " . ($track['prompt'] ?? ''));
            }
            break;
            
        case 'complete':
            error_log("All vocal tracks generated");
            error_log("Total " . count($vocalsData) . " vocal tracks generated:");
            
            foreach ($vocalsData as $index => $track) {
                $title = $track['title'] ?? '';
                $audioUrl = $track['audio_url'] ?? '';
                $prompt = $track['prompt'] ?? '';
                error_log("Track " . ($index + 1) . ": $title - $audioUrl");
                error_log("Vocals: $prompt");
                
                // Download vocal audio example
                try {
                    if ($audioUrl) {
                        $audioContent = file_get_contents($audioUrl);
                        if ($audioContent !== false) {
                            $filename = "vocals_{$taskId}_" . ($index + 1) . ".mp3";
                            file_put_contents($filename, $audioContent);
                            error_log("Vocal audio saved as $filename");
                        }
                    }
                } catch (Exception $e) {
                    error_log("Vocal audio download failed: " . $e->getMessage());
                }
            }
            break;
    }
    
} else {
    // Task failed
    error_log("Vocal generation failed: $msg");
    
    // Handle failure cases...
    if ($code === 400) {
        error_log("Parameter error or vocal content violation");
    } 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']);
?>