Generator Image - Gen Ai

/**
 * Generator Image - Gen Ai
 * Author  : gienetic
 * Base    : https://play.google.com/store/apps/details?id=com.sparksolutionz.genai
 */

const axios = require("axios");

/**
 * Generate image from Pollinations AI
 * @param {string} prompt - teks prompt
 * @param {string} model - model AI (flux, sdxl, midjourney, anime, realistic, turbo)
 * @param {object} opts - { width, height, seed, nologo, enhance, hidewatermark }
 * @returns {Promise<Buffer>} - image buffer
 */
async function generatePollinations(prompt, model = "flux", opts = {}) {
  const {
    width = 960,
    height = 1280,
    seed = Math.floor(Math.random() * 999999),
    nologo = true,
    enhance = true,
    hidewatermark = true,
  } = opts;

  try {
    const query = new URLSearchParams({
      model,
      width,
      height,
      seed,
    });

    if (nologo) query.set("nologo", "true");
    if (enhance) query.set("enhance", "true");
    if (hidewatermark) query.set("hidewatermark", "true");

    const url = `https://image.pollinations.ai/prompt/${encodeURIComponent(
      prompt
    )}?${query.toString()}`;

    const res = await axios.get(url, {
      responseType: "arraybuffer",
    });

    return Buffer.from(res.data, "binary");
  } catch (err) {
    console.error("❌ Gagal generate image:", err.message);
    throw new Error("Gagal generate image dari Pollinations");
  }
}

module.exports = { generatePollinations };

//Contoh Pemanggilan
//await generatePollinations(text, "flux");