音乐扩展回调

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

app = Flask(__name__)

@app.route('/suno-extend-callback', methods=['POST'])
def handle_callback():
    data = request.json
    
    code = data.get('code')
    msg = data.get('msg')
    callback_data = data.get('data', {})
    callback_type = callback_data.get('callbackType')
    task_id = callback_data.get('task_id')
    tracks = callback_data.get('data', [])
    
    print(f"收到 Suno 音乐扩展回调:")
    print(f"任务 ID: {task_id}, 类型: {callback_type}")
    print(f"状态: {code}, 消息: {msg}")
    
    if code == 200:
        # 任务进展或成功完成
        print(f"回调类型: {callback_type}")
        print(f"音乐数量: {len(tracks)}")
        
        if callback_type == 'text':
            print("文本生成完成，等待音乐扩展...")
        elif callback_type == 'first':
            print("第一首扩展完成，正在处理剩余音乐...")
            download_tracks(tracks, task_id)
        elif callback_type == 'complete':
            print("全部音乐扩展完成！")
            download_tracks(tracks, task_id)
            
    else:
        # 任务失败
        print(f"Suno 音乐扩展失败: {msg}")
        
        # 处理特定错误类型
        if code == 400:
            print("验证错误 - 检查版权内容")
        elif code == 408:
            print("速率限制 - 请等待后重试")
        elif code == 413:
            print("内容冲突 - 音频与现有作品匹配")
        elif code == 501:
            print("扩展失败 - 可能需要调整参数")
        elif code == 531:
            print("服务器错误已退还积分 - 可安全重试")
    
    # 返回 200 状态码确认收到回调
    return jsonify({'status': 'received'}), 200

def download_tracks(tracks, task_id):
    """下载音乐和封面图片"""
    for i, track in enumerate(tracks):
        track_id = track.get('id')
        audio_url = track.get('audio_url')
        image_url = track.get('image_url')
        title = track.get('title')
        duration = track.get('duration')
        
        print(f"音乐 {i + 1}: {title} ({duration}秒)")
        
        # 下载音频文件
        if audio_url:
            try:
                audio_filename = f"suno_extend_{task_id}_{track_id}.mp3"
                download_file(audio_url, audio_filename)
                print(f"音频下载成功: {track_id}")
            except Exception as e:
                print(f"音频下载失败 {track_id}: {e}")
        
        # 下载封面图片
        if image_url:
            try:
                image_filename = f"suno_extend_cover_{task_id}_{track_id}.jpeg"
                download_file(image_url, image_filename)
                print(f"封面下载成功: {track_id}")
            except Exception as e:
                print(f"封面下载失败 {track_id}: {e}")

def download_file(url, filename):
    """从 URL 下载文件并保存到本地"""
    response = requests.get(url, stream=True)
    response.raise_for_status()
    
    os.makedirs('downloads', exist_ok=True)
    filepath = os.path.join('downloads', filename)
    
    with open(filepath, 'wb') as f:
        for chunk in response.iter_content(chunk_size=8192):
            f.write(chunk)

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'] ?? [];
$callbackType = $callbackData['callbackType'] ?? '';
$taskId = $callbackData['task_id'] ?? '';
$tracks = $callbackData['data'] ?? [];

error_log("收到 Suno 音乐扩展回调:");
error_log("任务 ID: $taskId, 类型: $callbackType");
error_log("状态: $code, 消息: $msg");

if ($code === 200) {
    // 任务进展或成功完成
    error_log("回调类型: $callbackType");
    error_log("音乐数量: " . count($tracks));
    
    switch ($callbackType) {
        case 'text':
            error_log("文本生成完成，等待音乐扩展...");
            break;
            
        case 'first':
            error_log("第一首扩展完成，正在处理剩余音乐...");
            downloadTracks($tracks, $taskId);
            break;
            
        case 'complete':
            error_log("全部音乐扩展完成！");
            downloadTracks($tracks, $taskId);
            break;
    }
    
} else {
    // 任务失败
    error_log("Suno 音乐扩展失败: $msg");
    
    // 处理特定错误类型
    if ($code === 400) {
        error_log("验证错误 - 检查版权内容");
    } elseif ($code === 408) {
        error_log("速率限制 - 请等待后重试");
    } elseif ($code === 413) {
        error_log("内容冲突 - 音频与现有作品匹配");
    } elseif ($code === 501) {
        error_log("扩展失败 - 可能需要调整参数");
    } elseif ($code === 531) {
        error_log("服务器错误已退还积分 - 可安全重试");
    }
}

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

function downloadTracks($tracks, $taskId) {
    foreach ($tracks as $i => $track) {
        $trackId = $track['id'] ?? '';
        $audioUrl = $track['audio_url'] ?? '';
        $imageUrl = $track['image_url'] ?? '';
        $title = $track['title'] ?? '';
        $duration = $track['duration'] ?? 0;
        
        error_log("音乐 " . ($i + 1) . ": $title ({$duration}秒)");
        
        // 下载音频文件
        if (!empty($audioUrl)) {
            try {
                $audioFilename = "suno_extend_{$taskId}_{$trackId}.mp3";
                downloadFile($audioUrl, $audioFilename);
                error_log("音频下载成功: $trackId");
            } catch (Exception $e) {
                error_log("音频下载失败 $trackId: " . $e->getMessage());
            }
        }
        
        // 下载封面图片
        if (!empty($imageUrl)) {
            try {
                $imageFilename = "suno_extend_cover_{$taskId}_{$trackId}.jpeg";
                downloadFile($imageUrl, $imageFilename);
                error_log("封面下载成功: $trackId");
            } catch (Exception $e) {
                error_log("封面下载失败 $trackId: " . $e->getMessage());
            }
        }
    }
}

function downloadFile($url, $filename) {
    $downloadDir = 'downloads';
    if (!is_dir($downloadDir)) {
        mkdir($downloadDir, 0755, true);
    }
    
    $filepath = $downloadDir . '/' . $filename;
    
    $fileContent = file_get_contents($url);
    if ($fileContent === false) {
        throw new Exception("从 URL 下载文件失败");
    }
    
    $result = file_put_contents($filepath, $fileContent);
    if ($result === false) {
        throw new Exception("保存文件到本地失败");
    }
}
?>