Function generateImagesWithDALL_E

src/controllers/generation.controller.ts:373–411

(
  request: FastifyRequestTypebox<typeof GenerateImageResponseSchema>,
  reply: FastifyReplyTypebox<typeof GenerateImageResponseSchema>,
)

};

const generateImagesWithDALL_E = async (
  request: FastifyRequestTypebox<typeof GenerateImageResponseSchema>,
  reply: FastifyReplyTypebox<typeof GenerateImageResponseSchema>,
) => {
  const { numImages, prompt, size, model } = request.body;

  const openai = new OpenAI({
    apiKey: process.env.OPENAI_API_KEY,
    organization: process.env.OPENAI_ORG,
  });

  const promises = Array(numImages)
    .fill(null)
    .map(() => {
      return openai.images
        .generate({
          model: model ?? "dall-e-3",
          prompt: prompt,
          n: 1,
          size: size,
        })
        .catch(() => {
          throw new ApiError(500, "Something went wrong with the AI please try again later");
        });
    });

  try {
    const responses = await Promise.all(promises);
    const imageData = responses.flatMap(response => response["data"]);

    if (!imageData.length) {
      throw new ApiError(406, "Sorry our AI is currently overloaded please try again");
    }
    reply.code(200).send({ response: imageData });
  } catch (e) {
    Sentry.captureException(e);
    throw new ApiError(500, "Something went wrong with the AI please try again later");
  }
};

const generateImagesWithStableDiffusion = async (
  request: FastifyRequestTypebox<typeof GenerateImageResponseSchema>,