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