llm-pdf/image.js

Command Line tool to automate LLM and image models to generate a pdf by converting the generated text into markdown format and save it as a pdf document.

import fs from "fs/promises";
import Together from "together-ai";
// import dotenv from "dotenv";

// dotenv.config();

async function generateImage(prompt) {
  try {
    const together = new Together({ apiKey: "26d7651e279a9c77d6b0a1e3fdd5167788978e10ca35a02a02a0047fa6c714cc" });
    const imageFile = `./images/image${Date.now()}.png`;

    const response = await together.images.create({
      model: "black-forest-labs/FLUX.1-schnell-Free",
      prompt: prompt,
      width: 1440,
      height: 880,
      steps: 4,
      n: 1,
      response_format: "b64_json"
    });

    const base64Image = response.data[0].b64_json;
    const imageBuffer = Buffer.from(base64Image, "base64");

    await fs.writeFile(imageFile, imageBuffer);
    return `![Image](${imageFile})`;
  } catch (error) {
    console.error("An error occurred:", error);
  }
}

export default generateImage;