Music Video Generation Callbacks

from flask import Flask, request, jsonify
import requests

app = Flask(__name__)

@app.route('/video-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')
    video_url = callback_data.get('video_url')
    
    print(f"Received music video callback: {task_id}, Status: {code}")
    
    if code == 200 and video_url:
        # Video generation successful
        print(f"Music video ready: {video_url}")
        
        # Download video file
        download_video_file(video_url, task_id)
        
    else:
        # Task failed
        print(f"Music video generation 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

def download_video_file(url, task_id):
    """Download video file from the provided URL"""
    try:
        response = requests.get(url, stream=True)
        if response.status_code == 200:
            filename = f"video_{task_id}.mp4"
            with open(filename, "wb") as f:
                for chunk in response.iter_content(chunk_size=8192):
                    f.write(chunk)
            print(f"Video file downloaded: {filename}")
        else:
            print(f"Failed to download video file: HTTP {response.status_code}")
    except Exception as e:
        print(f"Video download failed: {e}")

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'] ?? '';
$videoUrl = $callbackData['video_url'] ?? '';

error_log("Received music video callback: $taskId, Status: $code");

if ($code === 200 && $videoUrl) {
    // Video generation successful
    error_log("Music video ready: $videoUrl");
    
    // Download video file
    downloadVideoFile($videoUrl, $taskId);
    
} else {
    // Task failed
    error_log("Music video generation 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");
    }
}

function downloadVideoFile($url, $taskId) {
    try {
        $context = stream_context_create([
            'http' => [
                'timeout' => 300 // 5 minutes timeout for large video files
            ]
        ]);
        
        $videoContent = file_get_contents($url, false, $context);
        if ($videoContent !== false) {
            $filename = "video_{$taskId}.mp4";
            file_put_contents($filename, $videoContent);
            error_log("Video file downloaded: $filename");
        } else {
            error_log("Failed to download video file from URL");
        }
    } catch (Exception $e) {
        error_log("Video download failed: " . $e->getMessage());
    }
}

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