34 lines
1.1 KiB
JavaScript
34 lines
1.1 KiB
JavaScript
const fs = require("fs");
|
|
const yaml = require("js-yaml");
|
|
const axios = require("axios");
|
|
|
|
const supportbotai = yaml.load(fs.readFileSync("./Configs/supportbot-ai.yml", "utf8"));
|
|
|
|
async function generateImage(prompt, negativePrompt, cfgScale) {
|
|
try {
|
|
const response = await axios.post('https://api.pepperai.xyz/generate', {
|
|
prompt: prompt,
|
|
negativePrompt: negativePrompt,
|
|
cfgScale: cfgScale
|
|
}, {
|
|
headers: {
|
|
'Authorization': `Bearer ${supportbotai.General.PepperAI_Key}`,
|
|
'Content-Type': 'application/json'
|
|
}
|
|
});
|
|
|
|
console.log('PepperAI Response:', response.data); // Log the entire response for debugging
|
|
|
|
if (response.data && response.data.imageUrl) {
|
|
return response.data.imageUrl;
|
|
} else {
|
|
throw new Error(`Invalid response structure: ${JSON.stringify(response.data)}`);
|
|
}
|
|
} catch (error) {
|
|
console.error('PepperAI Image Generation Error:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
module.exports = generateImage;
|