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

app = Flask(__name__)

@app.route('/veo3-1-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')
    
    print(f"Received Veo3.1 video generation callback:")
    print(f"Task ID: {task_id}")
    print(f"Status: {code}, Message: {msg}")
    
    if code == 200:
    # Video generation successful
    info = callback_data.get('info', {})
    result_urls = info.get('resultUrls')
    origin_urls = info.get('originUrls')
    resolution = info.get('resolution')
    fallback_flag = callback_data.get('fallbackFlag', False)
    
    print("Video generation successful!")
    print(f"Generated video URLs: {result_urls}")
    print(f"Video resolution: {resolution}")
    print(f"Using fallback model: {'Yes' if fallback_flag else 'No'}")
    if origin_urls:
        print(f"Original video URLs: {origin_urls}")
    
    # Download generated video files
    if result_urls:
        for i, url in enumerate(result_urls):
            if url:
                try:
                    video_filename = f"veo3.1_generated_{task_id}_{i}.mp4"
                    download_file(url, video_filename)
                    print(f"Video {i + 1} downloaded successfully")
                except Exception as e:
                    print(f"Video {i + 1} download failed: {e}")
    
    # Download original video files (if exists)
    if origin_urls:
        for i, url in enumerate(origin_urls):
            if url:
                try:
                    original_filename = f"veo3.1_original_{task_id}_{i}.mp4"
                    download_file(url, original_filename)
                    print(f"Original video {i + 1} downloaded successfully")
                except Exception as e:
                    print(f"Original video {i + 1} download failed: {e}")

else:
    # Video generation failed
    print(f"Veo3.1 video generation failed: {msg}")
    
    # Handle specific error types
    if code == 400:
        print("Client error - Check prompts and content policies")
        if 'content policies' in msg:
            print("Content review failed - Please modify prompts")
        elif 'English prompts' in msg:
            print("Language error - Only English prompts are supported")
        elif 'unsafe image' in msg:
            print("Image safety check failed - Please change image")
    elif code == 422:
        print("Fallback failed - Consider enabling fallback functionality (enableFallback: true)")
    elif code == 500:
        print("Server internal error - Please try again later")
    elif code == 501:
        print("Task failed - Video generation failed")
    
    # Return 200 status code to confirm callback received
    return jsonify({'code': 200, 'msg': 'success'}), 200

def download_file(url, filename):
    """Download file from URL and save to local"""
    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');

// 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['taskId'] ?? '';

error_log("Received Veo3.1 video generation callback:");
error_log("Task ID: $taskId");
error_log("Status: $code, Message: $msg");

if ($code === 200) {
    // Video generation successful
    $info = $callbackData['info'] ?? [];
    $resultUrls = $info['resultUrls'] ?? '';
    $originUrls = $info['originUrls'] ?? '';
    $resolution = $info['resolution'] ?? '';
    $fallbackFlag = $callbackData['fallbackFlag'] ?? false;
    
    error_log("Video generation successful!");
    error_log("Generated video URLs: $resultUrls");
    error_log("Video resolution: $resolution");
    error_log("Using fallback model: " . ($fallbackFlag ? 'Yes' : 'No'));
    if (!empty($originUrls)) {
        error_log("Original video URLs: $originUrls");
    }
    
    // Download generated video files
    if (!empty($resultUrls) && is_array($resultUrls)) {
        foreach ($resultUrls as $index => $url) {
            if (!empty($url)) {
                try {
                    $videoFilename = "veo3.1_generated_{$taskId}_{$index}.mp4";
                    downloadFile($url, $videoFilename);
                    error_log("Video " . ($index + 1) . " downloaded successfully");
                } catch (Exception $e) {
                    error_log("Video " . ($index + 1) . " download failed: " . $e->getMessage());
                }
            }
        }
    }
    
    // Download original video files (if exists)
    if (!empty($originUrls) && is_array($originUrls)) {
        foreach ($originUrls as $index => $url) {
            if (!empty($url)) {
                try {
                    $originalFilename = "veo3.1_original_{$taskId}_{$index}.mp4";
                    downloadFile($url, $originalFilename);
                    error_log("Original video " . ($index + 1) . " downloaded successfully");
                } catch (Exception $e) {
                    error_log("Original video " . ($index + 1) . " download failed: " . $e->getMessage());
                }
            }
        }
    }
    
} else {
    // Video generation failed
    error_log("Veo3.1 video generation failed: $msg");
    
    // Handle specific error types
    if ($code === 400) {
        error_log("Client error - Check prompts and content policies");
        if (strpos($msg, 'content policies') !== false) {
            error_log("Content review failed - Please modify prompts");
        } elseif (strpos($msg, 'English prompts') !== false) {
            error_log("Language error - Only English prompts are supported");
        } elseif (strpos($msg, 'unsafe image') !== false) {
            error_log("Image safety check failed - Please change image");
        }
    } elseif ($code === 422) {
        error_log("Fallback failed - Consider enabling fallback functionality (enableFallback: true)");
    } elseif ($code === 500) {
        error_log("Server internal error - Please try again later");
    } elseif ($code === 501) {
        error_log("Task failed - Video generation failed");
    }
}

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

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("Failed to download file from URL");
    }
    
    $result = file_put_contents($filepath, $fileContent);
    if ($result === false) {
        throw new Exception("Failed to save file locally");
    }
}
?>