import { GoogleGenAI, GenerateVideosOperation } from "@google/genai";
import { NextRequest, NextResponse } from "next/server";

const ai = new GoogleGenAI({ 
  apiKey: process.env.GEMINI_API_KEY,
  httpOptions: {
    headers: { 'User-Agent': 'aistudio-build' }
  }
});

export async function POST(req: NextRequest) {
  try {
    const { operationName } = await req.json();
    if (!operationName) return NextResponse.json({ error: "Missing operationName" }, { status: 400 });

    const op = new GenerateVideosOperation();
    op.name = operationName;
    const updated = await ai.operations.getVideosOperation({ operation: op });
    
    let uri = null;
    if (updated.done) {
       uri = updated.response?.generatedVideos?.[0]?.video?.uri;
    }

    return NextResponse.json({ done: updated.done, uri });
  } catch (error: any) {
    console.error("Video Poll Error:", error);
    return NextResponse.json({ error: error.message }, { status: 500 });
  }
}