import { FireworksAI } from '@simplesagar/fireworksai';

const fireworksAI = new FireworksAI({
    apiKey: process.env["FIREWORKS_API_KEY"] ?? "",
});

async function run() {
    const result = await fireworksAI.images.generateFromPrompt({
      accountId: "fireworks",
      modelId: "stable-diffusion-xl-1024-v1-0",
      bodyText2imageGen: {
        prompt: "A futuristic cityscape",
        negativePrompt: "cloudy day",
        height: 1024,
        width: 1024,
        cfgScale: 7,
        sampler: "K_EULER",
        samples: 1,
        seed: 0,
        steps: 30,
        safetyCheck: false,
      },
    });
  
    if (result instanceof ReadableStream) {
      const reader = result.getReader();
      const chunks = [];
  
      while (true) {
        const { done, value } = await reader.read();
        if (done) break;
        chunks.push(value);
      }
  
      const buffer = Buffer.concat(chunks);
  
      // Convert buffer to Base64
      const base64String = buffer.toString('base64');
      const dataUrl = `data:image/png;base64,${base64String}`;
  
      // Log the Data URL to the console
      const logStatement = `console.log("%c ", \`font-size: 1px; padding: 100px; background: url(${dataUrl}) no-repeat; background-size: contain;\`);`;
      console.log(logStatement);
    }
  }
  
  
  run();