Load pollinations models from an endpoint

8948354bf0a353bda2468edabc41b20627a48a4e

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/scripts/extensions/stable-diffusion/index.js+11 -22 Viewed

+}
 async function loadPollinationsModels() {
 -    return [
 +    const result = await fetch('/api/sd/pollinations/models', {
 -        {
 +        method: 'POST',
 -            value: 'flux',
 +        headers: getRequestHeaders(),
 -            text: 'FLUX.1 [schnell]',
 +    });
 -        },
++
 -        {
 +    if (result.ok) {
 -            value: 'flux-realism',
 +        const data = await result.json();
 -            text: 'FLUX Realism',
 +        return data;
 -        },
 +    }
 -        {
++
 -            value: 'flux-anime',
 +    return [];
 -            text: 'FLUX Anime',
 -        },
 -        {
 -            value: 'flux-3d',
 -            text: 'FLUX 3D',
 -        },
 -        {
 -            value: 'turbo',
 -            text: 'SDXL Turbo',
 -        },
 -    ];
+}
 async function loadTogetherAIModels() {

src/endpoints/stable-diffusion.js+25 -0 Viewed

 const pollinations = express.Router();
 +pollinations.post('/models', jsonParser, async (_request, response) => {
 +    try {
 +        const modelsUrl = new URL('https://image.pollinations.ai/models');
 +        const result = await fetch(modelsUrl);
++
 +        if (!result.ok) {
 +            console.log('Pollinations returned an error.', result.status, result.statusText);
 +            throw new Error('Pollinations request failed.');
 +        }
++
 +        const data = await result.json();
++
 +        if (!Array.isArray(data)) {
 +            console.log('Pollinations returned invalid data.');
 +            throw new Error('Pollinations request failed.');
 +        }
++
 +        const models = data.map(x => ({ value: x, text: x }));
 +        return response.send(models);
 +    } catch (error) {
 +        console.log(error);
 +        return response.sendStatus(500);
 +    }
 +});
++
 pollinations.post('/generate', jsonParser, async (request, response) => {
     try {
         const promptUrl = new URL(`https://image.pollinations.ai/prompt/${encodeURIComponent(request.body.prompt)}`);