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

app = Flask(__name__)

@app.route('/runway-extend-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')
    
    print(f"Received Runway video extension callback:")
    print(f"Task ID: {task_id}")
    print(f"Status: {code}, Message: {msg}")
    
    if code == 200:
        # Video extension successful
        video_id = callback_data.get('video_id')
        video_url = callback_data.get('video_url')
        image_url = callback_data.get('image_url')
        
        print("Video extension successful!")
        print(f"Video ID: {video_id}")
        print(f"Video URL: {video_url}")
        print(f"Cover URL: {image_url}")
        
        # Download video file
        if video_url:
            try:
                video_filename = f"runway_extend_{task_id}.mp4"
                download_file(video_url, video_filename)
                print("Video download successful")
            except Exception as e:
                print(f"Video download failed: {e}")
        
        # Download cover image
        if image_url:
            try:
                image_filename = f"runway_extend_cover_{task_id}.png"
                download_file(image_url, image_filename)
                print("Cover download successful")
            except Exception as e:
                print(f"Cover download failed: {e}")
                
    else:
        # Video extension failed
        print(f"Runway video extension failed: {msg}")
        
        # Handle specific error types
        if code == 400:
            print("Client error - Check input parameters and content")
            if 'Inappropriate content' in msg:
                print("Content moderation failed - Please replace image or video")
            elif 'Incorrect image format' in msg:
                print("Format error - Please check image format")
            elif 'concurrent generations' in msg:
                print("Concurrency limit - Please wait or upgrade membership")
            elif 'NSFW content' in msg:
                print("Content violation - Please modify prompt")
        elif code == 500:
            print("Server error - Please try again later")
    
    # 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 locally"""
    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['task_id'] ?? '';

error_log("Received Runway video extension callback:");
error_log("Task ID: $taskId");
error_log("Status: $code, Message: $msg");

if ($code === 200) {
    // Video extension successful
    $videoId = $callbackData['video_id'] ?? '';
    $videoUrl = $callbackData['video_url'] ?? '';
    $imageUrl = $callbackData['image_url'] ?? '';
    
    error_log("Video extension successful!");
    error_log("Video ID: $videoId");
    error_log("Video URL: $videoUrl");
    error_log("Cover URL: $imageUrl");
    
    // Download video file
    if (!empty($videoUrl)) {
        try {
            $videoFilename = "runway_extend_{$taskId}.mp4";
            downloadFile($videoUrl, $videoFilename);
            error_log("Video download successful");
        } catch (Exception $e) {
            error_log("Video download failed: " . $e->getMessage());
        }
    }
    
    // Download cover image
    if (!empty($imageUrl)) {
        try {
            $imageFilename = "runway_extend_cover_{$taskId}.png";
            downloadFile($imageUrl, $imageFilename);
            error_log("Cover download successful");
        } catch (Exception $e) {
            error_log("Cover download failed: " . $e->getMessage());
        }
    }
    
} else {
    // Video extension failed
    error_log("Runway video extension failed: $msg");
    
    // Handle specific error types
    if ($code === 400) {
        error_log("Client error - Check input parameters and content");
        if (strpos($msg, 'Inappropriate content') !== false) {
            error_log("Content moderation failed - Please replace image or video");
        } elseif (strpos($msg, 'Incorrect image format') !== false) {
            error_log("Format error - Please check image format");
        } elseif (strpos($msg, 'concurrent generations') !== false) {
            error_log("Concurrency limit - Please wait or upgrade membership");
        } elseif (strpos($msg, 'NSFW content') !== false) {
            error_log("Content violation - Please modify prompt");
        }
    } elseif ($code === 500) {
        error_log("Server error - Please try again later");
    }
}

// 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");
    }
}
?>