chatgpt生成图片

最近chatgpt火热，chatgpt也能根据文本生成图片

1废话不多说，上代码

exports.getimg = async (types, id, propmt) => {
    try {
        propmt = propmt.slice(propmt.indexOf("图"), propmt.length - 1)
        const resonse = await openai.createImage({
            prompt: propmt,
            n: 1,
            size: "512x512",
        });
        image_url = response.data.data[0].url;
        await SendMessage.SendMessage(types, `[CQ:image,file=${image_url}]`, id)
    } catch (e) {
        if (e) {
            await SendMessage.SendMessage(types, `图片出错了${e}`, id)
        }
    }

}

其中n是返回的图片数目，size是尺寸，propmt是输入的文本

2 .代码

1.首先是要下载openai提供的包

npm insatll openai

2.引入包

const {Configuration, OpenAIApi} = require("openai");

3 完整代码

const {Configuration, OpenAIApi} = require("openai");
const configuration = new Configuration({
    apiKey: token.apikey, //自己的apikey,要有openai账号
});
const openai = new OpenAIApi(configuration);
exports.getimg = async (types, id, propmt) => {
    try {
        propmt = propmt.slice(propmt.indexOf("图"), propmt.length - 1)
        const resonse = await openai.createImage({
            prompt: propmt,
            n: 1,
            size: "512x512",
        });
        image_url = response.data.data[0].url;
        // console.log(image_url)
        // fs.writeFileSync("chatgpt.txt", "Human:" + propmt + "\n" + image_url)
        await SendMessage.SendMessage(types, `[CQ:image,file=${image_url}]`, id)

    } catch (e) {
        if (e) {
            await SendMessage.SendMessage(types, `图片出错了${e}`, id)
        }
    }

}

4.在群聊中调用getimg方法

//只有管理才能调用此方法
if (data.user_id === config.manager[0]) {
      if (data.message.includes("画图")) {
            chatgpt.getimg("group", data.group_id, data.message)
        }
} 

3.结果

4.整个QQ机器人项目已经开源git