SunsetWarmLUT

import { NextResponse } from "next/server";
import OpenAI from "openai";

// Define the structure of the incoming request for image generation
interface ImageGenerationRequest {
  prompt: string; // Description for the image generation
  n?: number; // Number of images to generate
  size?: string; // Size of the generated images
}

// Create an instance of OpenAI with the API key from environment variables
const openai = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY,
});

// Function to generate an image using the OpenAI API
async function generateImage(data: ImageGenerationRequest): Promise<string> {
  console.log("Generating image with data:", JSON.stringify(data));
  
  try {
    const response = await openai.images.generate({
      model: "dall-e-3", // Specify DALL-E 3 model
      prompt: data.prompt,
      n: data.n || 1, // Default to generating 1 image if not specified
      size: data.size || "1024x1024", // Default size if not specified
    });

    console.log("OpenAI API response:", JSON.stringify(response));

    // Check if we received a valid image URL in the response
    if (response.data && response.data[0] && response.data[0].url) {
      console.log("Generated image URL:", response.data[0].url);
      return response.data[0].url; // Return the image URL
    }

    throw new Error("No image URL in the response"); // Throw error if no URL found
  } catch (error) {
    console.error("Error in generateImage:", error);
    throw error; // Propagate the error for further handling
  }
}

// Main POST function to handle incoming image generation requests
export async function POST(req: Request) {
  console.log("Received POST request");

  try {
    // Parse the incoming JSON request
    const data: ImageGenerationRequest = await req.json();
    console.log("Parsed request data:", JSON.stringify(data));

    // Call OpenAI API to generate the image using parsed data
    const imageUrl = await generateImage(data);

    console.log("Image generated successfully");

    // Return the response with the generated image URL
    return NextResponse.json({ imageUrl }, { status: 200 });
  } catch (error) {
    console.error("Error in POST handler:", error);
    // Handle any unexpected errors and return appropriate response
    return NextResponse.json(
      {
        error: "Failed to generate image.",
        details: error instanceof Error ? error.message : String(error),
      },
      { status: 500 } // Internal server error status code
    );
  }
}