生成图像 (Generate-2) 回调

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

app = Flask(__name__)

@app.route('/nanobanana-generate-2-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')
    success_flag = callback_data.get('successFlag')
    error_code = callback_data.get('errorCode')
    error_message = callback_data.get('errorMessage')
    response_data = callback_data.get('response') or {}
    result_image_url = response_data.get('resultImageUrl')
    
    print("收到 NanoBanana Generate-2 回调:")
    print(f"任务 ID: {task_id}")
    print(f"HTTP code: {code}, successFlag: {success_flag}, errorCode: {error_code}")
    print(f"msg: {msg}, errorMessage: {error_message}")
    
    if success_flag == 1:
        # 任务成功完成
        print("NanoBanana Generate-2 任务成功完成")
        print(f"生成图像 URL: {result_image_url}")
        
        # 下载生成的图像
        if result_image_url:
            try:
                result_filename = f"nanobanana_generate2_result_{task_id}.jpg"
                download_file(result_image_url, result_filename)
                print(f"生成图像已下载为 {result_filename}")
            except Exception as e:
                print(f"生成图像下载失败: {e}")
    else:
        # 任务失败
        print(f"NanoBanana Generate-2 任务失败: {error_message or msg}")
        
        if error_code == 422:
            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'] ?? '';
$successFlag = $callbackData['successFlag'] ?? null;
$errorCode = $callbackData['errorCode'] ?? null;
$errorMessage = $callbackData['errorMessage'] ?? '';
$response = $callbackData['response'] ?? [];
$resultImageUrl = $response['resultImageUrl'] ?? '';

error_log("收到 NanoBanana Generate-2 回调:");
error_log("任务 ID: $taskId");
error_log("HTTP code: $code, successFlag: $successFlag, errorCode: $errorCode");
error_log("msg: $msg, errorMessage: $errorMessage");

if ($successFlag === 1) {
    // 任务成功完成
    error_log("NanoBanana Generate-2 任务成功完成");
    
    error_log("生成图像 URL: $resultImageUrl");
    
    // 下载生成的图像
    if (!empty($resultImageUrl)) {
        try {
            $resultFilename = "nanobanana_generate2_result_{$taskId}.jpg";
            downloadFile($resultImageUrl, $resultFilename);
            error_log("生成图像已下载为 $resultFilename");
        } catch (Exception $e) {
            error_log("生成图像下载失败: " . $e->getMessage());
        }
    }
} else {
    // 任务失败
    error_log("NanoBanana Generate-2 任务失败: " . ($errorMessage ?: $msg));
    
    if ($errorCode === 422) {
        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("保存文件到本地失败");
    }
}
?>