<?php
ini_set('display_errors', 1);
ini_set('display_startup_errors', 1);
error_reporting(E_ALL);
// =================== КОНФИГУРАЦИЯ ===================
define('REPLICATE_API_KEY', 'r8_39RbLdKHSJeOcNjIrdpx1p7oGTmolK41vWUZf'); // Замените на ваш API-ключ от replicate.com
define('REPLICATE_API_URL', 'https://api.replicate.com/v1/predictions');

// Модели (owner/model:version)
define('MODEL_IMAGE', 'black-forest-labs/flux-schnell');
define('MODEL_VIDEO', 'genmo/mochi-1-preview');
define('MODEL_AUDIO', 'meta/musicgen');

// =================== ОБРАБОТЧИКИ AJAX ===================
if ($_SERVER['REQUEST_METHOD'] === 'POST') {
    header('Content-Type: application/json');
    
    $action = $_POST['action'] ?? '';
    $prompt = trim($_POST['prompt'] ?? '');
    
    if (empty($prompt)) {
        echo json_encode(['error' => 'Промпт не может быть пустым']);
        exit;
    }
    
    try {
        switch ($action) {
            case 'generate_image':
                $result = generateImage($prompt);
                break;
            case 'generate_video':
                $result = generateVideo($prompt);
                break;
            case 'generate_music':
                $result = generateMusic($prompt);
                break;
            default:
                throw new Exception('Неизвестное действие');
        }
        echo json_encode($result);
    } catch (Exception $e) {
        echo json_encode(['error' => $e->getMessage()]);
    }
    exit;
}

// =================== ФУНКЦИИ ===================
function replicateRequest($modelVersion, $input, $timeout = 120) {
    // 1. Создаём предсказание
    $ch = curl_init(REPLICATE_API_URL);
    curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
    curl_setopt($ch, CURLOPT_HTTPHEADER, [
        'Authorization: Token ' . REPLICATE_API_KEY,
        'Content-Type: application/json'
    ]);
    curl_setopt($ch, CURLOPT_POST, true);
    curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode([
        'version' => $modelVersion,
        'input' => $input
    ]));
    
    $response = curl_exec($ch);
    $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
    curl_close($ch);
    
    if ($httpCode !== 201) {
        $error = json_decode($response, true)['detail'] ?? 'Ошибка создания предсказания';
        throw new Exception($error);
    }
    
    $prediction = json_decode($response, true);
    $getUrl = $prediction['urls']['get'];
    $predictionId = $prediction['id'];
    
    // 2. Ожидаем завершения (polling)
    $start = time();
    while (time() - $start < $timeout) {
        $ch = curl_init($getUrl);
        curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
        curl_setopt($ch, CURLOPT_HTTPHEADER, [
            'Authorization: Token ' . REPLICATE_API_KEY
        ]);
        $response = curl_exec($ch);
        $status = json_decode($response, true);
        curl_close($ch);
        
        if ($status['status'] === 'succeeded') {
            // Извлекаем URL результата (модели могут возвращать массив или строку)
            $output = $status['output'];
            if (is_array($output)) {
                $output = reset($output);
            }
            return ['url' => $output];
        } elseif ($status['status'] === 'failed') {
            throw new Exception('Ошибка генерации: ' . ($status['error'] ?? 'неизвестная ошибка'));
        }
        sleep(2);
    }
    throw new Exception('Превышено время ожидания (возможно, очередь Replicate)');
}

function generateImage($prompt) {
    // Модель flux-schnell: принимает prompt, output_format (webp/jpeg/png)
    $input = [
        'prompt' => $prompt,
        'go_fast' => true,
        'guidance' => 3,
        'output_format' => 'webp'
    ];
    return replicateRequest('black-forest-labs/flux-schnell', $input);
}

function generateVideo($prompt) {
    // Mochi-1 preview: принимает prompt, num_frames=  (опционально)
    $input = [
        'prompt' => $prompt,
        'num_inference_steps' => 50,
        'guidance_scale' => 4.5,
    ];
    // Версия модели genmo/mochi-1-preview
    return replicateRequest('genmo/mochi-1-preview', $input, 180); // 3 минуты на видео
}

function generateMusic($prompt) {
    // Meta MusicGen: принимает prompt, duration (секунды), temperature и др.
    $input = [
        'prompt' => $prompt,
        'duration' => 8,
        'temperature' => 0.8,
        'top_k' => 250,
        'top_p' => 0.95
    ];
    // Используем популярную версию Meta MusicGen (large)
    return replicateRequest('meta/musicgen', $input);
}
?>
<!DOCTYPE html>
<html lang="ru">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>AI Генератор (Replicate) | Картинки, Видео, Музыка</title>
    <style>
        * {
            margin: 0;
            padding: 0;
            box-sizing: border-box;
        }
        body {
            background: linear-gradient(145deg, #0f172a 0%, #1e293b 100%);
            font-family: 'Segoe UI', system-ui, -apple-system, 'Inter', sans-serif;
            min-height: 100vh;
            padding: 2rem 1rem;
            color: #f1f5f9;
        }
        .container {
            max-width: 1300px;
            margin: 0 auto;
        }
        h1 {
            text-align: center;
            font-size: 2.5rem;
            font-weight: 700;
            background: linear-gradient(135deg, #c084fc, #60a5fa);
            -webkit-background-clip: text;
            background-clip: text;
            color: transparent;
            margin-bottom: 0.5rem;
        }
        .sub {
            text-align: center;
            color: #94a3b8;
            margin-bottom: 3rem;
        }
        .cards {
            display: grid;
            grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
            gap: 2rem;
        }
        .card {
            background: rgba(30, 41, 59, 0.6);
            backdrop-filter: blur(10px);
            border-radius: 2rem;
            padding: 1.8rem;
            border: 1px solid rgba(148, 163, 184, 0.2);
            transition: transform 0.2s, box-shadow 0.2s;
            box-shadow: 0 20px 35px -12px rgba(0,0,0,0.3);
        }
        .card:hover {
            transform: translateY(-5px);
            box-shadow: 0 25px 40px -15px rgba(0,0,0,0.5);
            border-color: rgba(192, 132, 252, 0.4);
        }
        .card h2 {
            display: flex;
            align-items: center;
            gap: 0.75rem;
            font-size: 1.8rem;
            margin-bottom: 1rem;
            border-left: 4px solid #c084fc;
            padding-left: 1rem;
        }
        .icon {
            font-size: 2rem;
        }
        textarea, input {
            width: 100%;
            background: #0f172a;
            border: 1px solid #334155;
            border-radius: 1.2rem;
            padding: 0.9rem 1rem;
            color: #f1f5f9;
            font-size: 1rem;
            font-family: inherit;
            resize: vertical;
            transition: all 0.2s;
        }
        textarea:focus, input:focus {
            outline: none;
            border-color: #c084fc;
            box-shadow: 0 0 0 2px rgba(192,132,252,0.3);
        }
        button {
            background: linear-gradient(95deg, #8b5cf6, #6366f1);
            border: none;
            border-radius: 2rem;
            padding: 0.8rem 1.5rem;
            font-weight: 600;
            font-size: 1rem;
            color: white;
            cursor: pointer;
            margin-top: 1.2rem;
            width: 100%;
            transition: all 0.2s;
            display: flex;
            justify-content: center;
            align-items: center;
            gap: 8px;
        }
        button:hover {
            background: linear-gradient(95deg, #7c3aed, #4f46e5);
            transform: scale(0.98);
            box-shadow: 0 10px 20px -5px #4f46e5;
        }
        .result-area {
            margin-top: 1.5rem;
            background: #0f172a;
            border-radius: 1.2rem;
            padding: 1rem;
            text-align: center;
            min-height: 100px;
            display: flex;
            flex-direction: column;
            align-items: center;
            justify-content: center;
        }
        .result-area img, .result-area video, .result-area audio {
            max-width: 100%;
            border-radius: 1rem;
            margin-top: 0.5rem;
            box-shadow: 0 8px 20px black;
        }
        .loading {
            display: inline-block;
            width: 30px;
            height: 30px;
            border: 3px solid #334155;
            border-radius: 50%;
            border-top-color: #c084fc;
            animation: spin 1s ease infinite;
        }
        @keyframes spin { to { transform: rotate(360deg); } }
        .error {
            color: #f87171;
            background: rgba(248,113,113,0.1);
            padding: 0.6rem;
            border-radius: 1rem;
        }
        footer {
            text-align: center;
            margin-top: 3rem;
            color: #64748b;
            font-size: 0.85rem;
        }
        a {
            color: #a78bfa;
            text-decoration: none;
        }
        .inline-code {
            background: #0f172a;
            padding: 0.2rem 0.4rem;
            border-radius: 8px;
            font-family: monospace;
        }
        @media (max-width: 700px) {
            .cards { grid-template-columns: 1fr; }
            h1 { font-size: 1.8rem; }
        }
    </style>
</head>
<body>
<div class="container">
    <h1>✨ Neural Forge AI</h1>
    <div class="sub">Генерация картинок, видео и музыки через Replicate API</div>

    <div class="cards">
        <!-- КАРТИНКИ -->
        <div class="card">
            <h2><span class="icon">🖼️</span> Изображение</h2>
            <textarea id="imgPrompt" rows="3" placeholder="Опишите что нарисовать... Например: cyberpunk cat, neon city, 4k"></textarea>
            <button id="genImageBtn">🎨 Сгенерировать картинку</button>
            <div id="imgResult" class="result-area">✨ Ваше изображение появится здесь</div>
        </div>

        <!-- ВИДЕО -->
        <div class="card">
            <h2><span class="icon">🎬</span> Видео (Genmo Mochi-1)</h2>
            <textarea id="videoPrompt" rows="3" placeholder="Опишите сцену для видео... Например: astronaut riding a horse on Mars, cinematic lighting"></textarea>
            <button id="genVideoBtn">📹 Сгенерировать видео (~1-2 минуты)</button>
            <div id="videoResult" class="result-area">🎥 Видео появится после обработки</div>
        </div>

        <!-- МУЗЫКА -->
        <div class="card">
            <h2><span class="icon">🎵</span> Музыка (MusicGen)</h2>
            <textarea id="musicPrompt" rows="3" placeholder="Опишите трек: epic cinematic orchestra, lo-fi chillhop, synthwave 80s"></textarea>
            <button id="genMusicBtn">🎶 Сгенерировать аудио (8 секунд)</button>
            <div id="musicResult" class="result-area">🎧 Аудиофрагмент будет здесь</div>
        </div>
    </div>
    <footer>
        ⚡ Модели: Flux Schnell (изображения), Mochi-1 (видео), MusicGen (аудио). <br>
        Требуется API ключ Replicate. Время генерации зависит от очереди сервиса.
    </footer>
</div>

<script>
    const API_KEY_PLACEHOLDER = "<?php echo REPLICATE_API_KEY !== 'r8_...' ? 'установлен' : 'не задан'; ?>";
    if (API_KEY_PLACEHOLDER === 'не задан') {
        console.warn('⚠️ Замените API-ключ в коде index.php на ваш реальный ключ replicate.com');
    }

    async function generate(action, prompt, resultDiv, type) {
        if (!prompt.trim()) {
            resultDiv.innerHTML = '<div class="error">❌ Введите описание</div>';
            return;
        }
        resultDiv.innerHTML = '<div class="loading"></div><span style="margin-top: 10px;">Генерация... (может занять до 2 минут)</span>';

        try {
            const formData = new FormData();
            formData.append('action', action);
            formData.append('prompt', prompt);
            
            const response = await fetch('', { // текущий файл
                method: 'POST',
                body: formData
            });
            const data = await response.json();
            
            if (data.error) {
                resultDiv.innerHTML = `<div class="error">⚠️ Ошибка: ${data.error}</div>`;
                return;
            }
            
            if (data.url) {
                if (type === 'image') {
                    resultDiv.innerHTML = `<img src="${data.url}" alt="generated" style="max-width:100%; border-radius:1rem;">`;
                } else if (type === 'video') {
                    resultDiv.innerHTML = `<video controls autoplay loop muted style="max-width:100%; border-radius:1rem;"><source src="${data.url}" type="video/mp4">Ваш браузер не поддерживает видео</video>`;
                } else if (type === 'audio') {
                    resultDiv.innerHTML = `<audio controls src="${data.url}" style="width:100%;"></audio><br><small><a href="${data.url}" target="_blank">🔗 Открыть в новой вкладке</a></small>`;
                } else {
                    resultDiv.innerHTML = `<a href="${data.url}" target="_blank">✅ Результат готов (открыть)</a>`;
                }
            } else {
                resultDiv.innerHTML = '<div class="error">Неожиданный ответ от сервера</div>';
            }
        } catch (err) {
            resultDiv.innerHTML = `<div class="error">❌ Ошибка запроса: ${err.message}</div>`;
        }
    }

    document.getElementById('genImageBtn').onclick = () => {
        const prompt = document.getElementById('imgPrompt').value;
        generate('generate_image', prompt, document.getElementById('imgResult'), 'image');
    };
    document.getElementById('genVideoBtn').onclick = () => {
        const prompt = document.getElementById('videoPrompt').value;
        generate('generate_video', prompt, document.getElementById('videoResult'), 'video');
    };
    document.getElementById('genMusicBtn').onclick = () => {
        const prompt = document.getElementById('musicPrompt').value;
        generate('generate_music', prompt, document.getElementById('musicResult'), 'audio');
    };
</script>
</body>
</html>