diff --git a/commands/message/genai/imagen.js b/commands/message/genai/imagen.js index 29080fa..9470c39 100644 --- a/commands/message/genai/imagen.js +++ b/commands/message/genai/imagen.js @@ -13,8 +13,8 @@ module.exports = { label: 'text', aliases: ['aiimg'], metadata: { - description: `${iconPill("generative_ai", "LIMITED TESTING")}\n\nGenerate images with Imagen 2`, - description_short: 'Create Images with Imagen', + description: `${iconPill("generative_ai", "LIMITED TESTING")}\n\nGenerate images with Imagen 3`, + description_short: 'Create Images with Imagen 3', examples: ['imagen a painting of northern lights'], category: 'limited', usage: 'imagen ' @@ -26,9 +26,11 @@ module.exports = { if(!args.text) return editOrReply(context, createEmbed("warning", context, `Missing Parameter (prompt).`)) + const IMAGE_COUNT = 4; + try{ let loadingEmbeds = []; - for (let i = 0; i < 4; i++) { + for (let i = 0; i < IMAGE_COUNT; i++) { loadingEmbeds.push(createEmbed("defaultNoFooter", context, { url: "https://bignutty.gitlab.io", author: { @@ -43,7 +45,7 @@ module.exports = { await editOrReply(context, {embeds: loadingEmbeds}); - let res = await googleGenaiImagen(context, args.text); + let res = await googleGenaiImagen(context, args.text, IMAGE_COUNT); // Construct Embeds let files = []; diff --git a/labscore/api/index.js b/labscore/api/index.js index 13fadff..aabf0c6 100644 --- a/labscore/api/index.js +++ b/labscore/api/index.js @@ -42,9 +42,10 @@ async function request(path, type, headers, args, host) { throw new Error("unsupported, must either use GET or POST"); } -module.exports.googleGenaiImagen = async function(context, prompt){ +module.exports.googleGenaiImagen = async function(context, prompt, imageCount = 2){ return await request(Api.GOOGLE_GENERATIVEAI_IMAGEN, "GET", {}, { - prompt: prompt + prompt: prompt, + image_count: imageCount }) }