mirror of
https://github.com/Expand-sys/expandschatbot
synced 2026-03-22 12:37:10 +11:00
115 lines
4 KiB
JavaScript
115 lines
4 KiB
JavaScript
const { Configuration, OpenAIApi } = require("openai");
|
|
const { grpc } = require("@improbable-eng/grpc-web")
|
|
const {NodeHttpTransport } = require("@improbable-eng/grpc-web-node-http-transport")
|
|
grpc.setDefaultTransport(NodeHttpTransport());
|
|
|
|
const GenerationService = require("../../generation/generation_pb_service")
|
|
const Generation = require("../../generation/generation_pb")
|
|
|
|
|
|
|
|
const configuration = new Configuration({
|
|
apiKey: process.env.OPENAI_API_KEY,
|
|
});
|
|
const openai = new OpenAIApi(configuration);
|
|
|
|
|
|
|
|
module.exports = {
|
|
name: 'image2',
|
|
description: 'AI image generation',
|
|
async execute(message, args){
|
|
const imageParams = new Generation.ImageParameters();
|
|
imageParams.setWidth(512);
|
|
imageParams.setHeight(512);
|
|
imageParams.addSeed(11223344);
|
|
imageParams.setSamples(1);
|
|
imageParams.setSteps(75);
|
|
|
|
// Use the `k-dpmpp-2` sampler
|
|
const transformType = new Generation.TransformType();
|
|
transformType.setDiffusion(Generation.DiffusionSampler.SAMPLER_K_DPMPP_2M);
|
|
imageParams.setTransform(transformType);
|
|
|
|
// Use Stable Diffusion 2.0
|
|
const request = new Generation.Request();
|
|
request.setEngineId("stable-diffusion-512-v2-1");
|
|
request.setRequestedType(Generation.ArtifactType.ARTIFACT_IMAGE);
|
|
request.setClassifier(new Generation.ClassifierParameters());
|
|
|
|
// Use a CFG scale of `13`
|
|
const samplerParams = new Generation.SamplerParameters();
|
|
samplerParams.setCfgScale(13);
|
|
|
|
const stepParams = new Generation.StepParameter();
|
|
const scheduleParameters = new Generation.ScheduleParameters();
|
|
|
|
// Set the schedule to `0`, this changes when doing an initial image generation
|
|
stepParams.setScaledStep(0);
|
|
stepParams.setSampler(samplerParams);
|
|
stepParams.setSchedule(scheduleParameters);
|
|
|
|
imageParams.addParameters(stepParams);
|
|
request.setImage(imageParams);
|
|
|
|
// Set our text prompt
|
|
const promptText = new Generation.Prompt();
|
|
promptText.setText(
|
|
`${message.content.slice(7, message.content.length)}`
|
|
);
|
|
|
|
request.addPrompt(promptText);
|
|
|
|
// Authenticate using your API key, don't commit your key to a public repository!
|
|
const metadata = new grpc.Metadata();
|
|
metadata.set("Authorization", "Bearer " + process.env.API_KEY);
|
|
|
|
// Create a generation client
|
|
const generationClient = new GenerationService.GenerationServiceClient(
|
|
"https://grpc.stability.ai",
|
|
{}
|
|
);
|
|
|
|
// Send the request using the `metadata` with our key from earlier
|
|
const generation = generationClient.generate(request, metadata);
|
|
|
|
// Set up a callback to handle data being returned
|
|
generation.on("data", (data) => {
|
|
data.getArtifactsList().forEach((artifact) => {
|
|
// Oh no! We were filtered by the NSFW classifier!
|
|
if (
|
|
artifact.getType() === Generation.ArtifactType.ARTIFACT_TEXT &&
|
|
artifact.getFinishReason() === Generation.FinishReason.FILTER
|
|
) {
|
|
return console.error("Your image was filtered by the NSFW classifier.");
|
|
}
|
|
|
|
// Make sure we have an image
|
|
if (artifact.getType() !== Generation.ArtifactType.ARTIFACT_IMAGE) return;
|
|
|
|
// You can convert the raw binary into a base64 string
|
|
const base64Image = btoa(
|
|
new Uint8Array(artifact.getBinary()).reduce(
|
|
(data, byte) => data + String.fromCodePoint(byte),
|
|
""
|
|
)
|
|
);
|
|
|
|
// Here's how you get the seed back if you set it to `0` (random)
|
|
const seed = artifact.getSeed();
|
|
|
|
// We're done!
|
|
sfbuffer = new Buffer.from(base64Image, "base64");
|
|
message.reply({files: [{ attachment: sfbuffer, name: 'newName.png' }]})
|
|
});
|
|
});
|
|
|
|
// Anything other than `status.code === 0` is an error
|
|
generation.on("status", (status) => {
|
|
if (status.code === 0) return;
|
|
console.error(
|
|
"Your image could not be generated. You might not have enough credits."
|
|
);
|
|
});
|
|
},
|
|
};
|