import { openAsBlob } from "node:fs";

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

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

async function run() {
	const result = await fireworksAI.images.generateFromControlNet({
		accountId: "fireworks",
		modelId: "stable-diffusion-xl-1024-v1-0",
		bodyControlNetGen: {
			controlImage: await openAsBlob(
				"/Users/plaso/Documents/API/inputs/profile_img.jpg",
			),
			prompt: "make the image look like a painting",
			cfgScale: 7,
			conditioningScale: 0.5,
			controlNetName: "canny",
			height: 1024,
			negativePrompt: "cloudy day",
			safetyCheck: false,
			sampler: "K_EULER",
			samples: 1,
			seed: 0,
			stepScheduleEnd: 1,
			stepScheduleStart: 0,
			steps: 30,
			width: 1024,
		},
	});

	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);

		const base64String = buffer.toString("base64");
		const dataUrl = `data:image/png;base64,${base64String}`;

		console.log(
			`console.log("%c ", "font-size: 1px; padding: 100px; background: url(${dataUrl}) no-repeat; background-size: contain;");`,
		);
	}
}

run();