Update server.js

server.js

@@ -16,51 +16,49 @@ app.all('/', (req, res, next) => {
     if (req.method !== 'POST') {
         res.status(405).send('Только POST');
     } else {
-        next();
-    }
-});
-
-app.post('/generate-image', async (req, res) => {
-    const { prompt } = req.body;
-
-    if (!prompt) {
-        return res.status(400).send('Missing "prompt" in request body');
-    }
+        const { prompt } = req.body;
 
-    try {
-        // Динамический импорт node-fetch
-        const fetch = (await import('node-fetch')).default;
+        if (!prompt) {
+            return res.status(400).send('Missing "prompt" in request body');
+        }
+    
+        try {
+            // Динамический импорт node-fetch
+            const fetch = (await import('node-fetch')).default;
 
-        console.log('Sending request to Hugging Face API...');
+            console.log('Sending request to Hugging Face API...');
 
-        const response = await fetch('https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3-medium', {
-            method: 'POST',
-            headers: {
-                'Authorization': `Bearer ${apiKey}`,
-                'Content-Type': 'application/json'
-            },
-            body: JSON.stringify({
-                inputs: prompt
-            }),
-            // Добавляем тайм-аут в 60 секунд
-            timeout: 60000
-        });
+            const response = await fetch('https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3-medium', {
+                method: 'POST',
+                headers: {
+                    'Authorization': `Bearer ${apiKey}`,
+                    'Content-Type': 'application/json'
+                },
+                body: JSON.stringify({
+                    inputs: prompt
+                }),
+                // Добавляем тайм-аут в 60 секунд
+                timeout: 60000
+            });
 
-        if (!response.ok) {
-            throw new Error(`Error from Hugging Face API: ${response.statusText}`);
-        }
+            if (!response.ok) {
+                throw new Error(`Error from Hugging Face API: ${response.statusText}`);
+            }
 
-        console.log('Received response from Hugging Face API, processing image...');
+            console.log('Received response from Hugging Face API, processing image...');
 
-        const imageBuffer = await response.buffer();
-        const base64Image = imageBuffer.toString('base64');
+            const imageBuffer = await response.buffer();
+            const base64Image = imageBuffer.toString('base64');
 
-        res.json({ image: base64Image });
-    } catch (error) {
-        console.error('Error generating image:', error.message);
-        res.status(500).send('Error generating image');
+            res.json({ image: base64Image });
+        } catch (error) {
+            console.error('Error generating image:', error.message);
+            res.status(500).send('Error generating image');
+        }
     }
 });
+    
+});
 
 app.listen(port, () => {
     console.log(`Server is running on http://localhost:${port}`);