import { NextRequest, NextResponse } from 'next/server';
import { GrokClient } from '@/lib/providers/grok-client';

export async function POST(req: NextRequest) {
    try {
        const { prompt, apiKey, cookies, imageCount = 1 } = await req.json();

        if (!prompt) {
            return NextResponse.json({ error: "Prompt is required" }, { status: 400 });
        }

        if (!apiKey && !cookies) {
            return NextResponse.json(
                { error: "Grok API key or cookies required. Configure in Settings." },
                { status: 401 }
            );
        }

        console.log(`[Grok API Route] Generating ${imageCount} image(s) for: "${prompt.substring(0, 30)}..."`);

        const client = new GrokClient({ apiKey, cookies });
        const results = await client.generate(prompt, imageCount);

        // Download images as base64 for storage
        const images = await Promise.all(
            results.map(async (img) => {
                let base64 = img.data;
                if (!base64 && img.url && !img.url.startsWith('data:')) {
                    try {
                        base64 = await client.downloadAsBase64(img.url);
                    } catch (e) {
                        console.warn("[Grok API Route] Failed to download image:", e);
                    }
                }
                return {
                    data: base64 || '',
                    url: img.url,
                    prompt: img.prompt,
                    model: img.model,
                    aspectRatio: '1:1' // Grok default
                };
            })
        );

        const validImages = images.filter(img => img.data || img.url);

        if (validImages.length === 0) {
            throw new Error("No valid images generated");
        }

        return NextResponse.json({ images: validImages });

    } catch (error: any) {
        console.error("[Grok API Route] Error:", error);
        return NextResponse.json(
            { error: error.message || "Grok generation failed" },
            { status: 500 }
        );
    }
}