<?php

namespace App\Helpers\FirstParty\AI;

use Illuminate\Support\Facades\Http;

class RunwareAI
{
    public static function generateSchnellImage($uuid, $prompt, $width = 1024, $height = 1024)
    {
        $api_key = config('services.runware.api_key');

        try {
            $response = Http::timeout(60)
                ->withHeaders([
                    'Content-Type' => 'application/json',
                    'Authorization' => 'Bearer '.$api_key,
                ])
                ->post('https://api.runware.ai/v1', [
                    [
                        'taskUUID' => $uuid,
                        'taskType' => 'imageInference',
                        'width' => $width,
                        'height' => $height,
                        'numberResults' => 1,
                        'outputFormat' => 'WEBP',
                        'outputType' => ['URL'],
                        'includeCost' => true,
                        'inputImages' => [],
                        'positivePrompt' => $prompt,
                        'model' => 'runware:100@1',
                    ],
                ]);

            // Check if the request was successful
            if ($response->successful()) {
                $data = $response->json();

                // Extract the image URL from the response
                if (isset($data['data']) && count($data['data']) > 0) {
                    $imageData = collect($data['data'])
                        ->where('taskType', 'imageInference')
                        ->first();

                    if ($imageData && isset($imageData['imageURL'])) {
                        return $imageData['imageURL'];
                    }
                }

                throw new \Exception('Image URL not found in response');
            }

            throw new \Exception('API request failed: '.$response->status().' - '.$response->body());
        } catch (\Exception $e) {
            // Log the error or handle as needed
            \Log::error('RunwareAI API Error: '.$e->getMessage());
            throw $e;
        }
    }
}