Luma 修改视频回调

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

app = Flask(__name__)

@app.route('/luma-modify-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('taskId')
    prompt_json = callback_data.get('promptJson')
    result_urls = callback_data.get('resultUrls', [])
    
    print(f"收到 Luma 视频修改回调:")
    print(f"任务 ID: {task_id}")
    print(f"状态: {code}, 消息: {msg}")
    
    if code == 200:
        # 任务成功完成
        print("Luma 视频修改成功完成")
        
        print(f"任务 ID: {task_id}")
        print(f"原始参数: {prompt_json}")
        print(f"生成的视频 URL: {', '.join(result_urls)}")
        
        # 解析原始参数
        try:
            original_params = json.loads(prompt_json)
            print(f"原始提示: {original_params.get('prompt', 'N/A')}")
            print(f"原始视频 URL: {original_params.get('videoUrl', 'N/A')}")
        except json.JSONDecodeError:
            print("无法解析原始参数")
        
        # 下载生成的视频
        for index, url in enumerate(result_urls):
            try:
                filename = f"luma_result_{task_id}_{index + 1}.mp4"
                download_file(url, filename)
                print(f"视频 {index + 1} 下载为 {filename}")
            except Exception as e:
                print(f"视频 {index + 1} 下载失败: {e}")
                
    else:
        # 任务失败
        print(f"Luma 视频修改失败: {msg}")
        
        # 记录原始参数用于调试
        print(f"原始参数: {prompt_json}")
        
        # 处理特定错误情况
        print("请检查您的输入视频和提示，然后重试")
    
    # 返回 200 状态码确认收到回调
    return jsonify({'status': 'received'}), 200

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'] ?? [];
$taskId = $callbackData['taskId'] ?? '';
$promptJson = $callbackData['promptJson'] ?? '';
$resultUrls = $callbackData['resultUrls'] ?? [];

error_log("收到 Luma 视频修改回调:");
error_log("任务 ID: $taskId");
error_log("状态: $code, 消息: $msg");

if ($code === 200) {
    // 任务成功完成
    error_log("Luma 视频修改成功完成");
    
    error_log("任务 ID: $taskId");
    error_log("原始参数: $promptJson");
    error_log("生成的视频 URL: " . implode(', ', $resultUrls));
    
    // 解析原始参数
    $originalParams = json_decode($promptJson, true);
    if ($originalParams) {
        error_log("原始提示: " . ($originalParams['prompt'] ?? 'N/A'));
        error_log("原始视频 URL: " . ($originalParams['videoUrl'] ?? 'N/A'));
    }
    
    // 下载生成的视频
    foreach ($resultUrls as $index => $url) {
        try {
            $filename = "luma_result_{$taskId}_" . ($index + 1) . ".mp4";
            downloadFile($url, $filename);
            error_log("视频 " . ($index + 1) . " 下载为 $filename");
        } catch (Exception $e) {
            error_log("视频 " . ($index + 1) . " 下载失败: " . $e->getMessage());
        }
    }
    
} else {
    // 任务失败
    error_log("Luma 视频修改失败: $msg");
    
    // 记录原始参数用于调试
    error_log("原始参数: $promptJson");
    
    // 处理特定错误情况
    error_log("请检查您的输入视频和提示，然后重试");
}

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

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("本地保存文件失败");
    }
}
?>