Update server.js

server.js

@@ -1,23 +1,25 @@
 const express = require('express');
-const axios = require('axios');
 const rateLimit = require('express-rate-limit');
+const axios = require('axios');
 
 const app = express();
 app.use(express.json());
 
 const openai_key = process.env.OPENAI_KEY;
 
-// Ограничение по IP-адресу: не более одной генерации в 30 секунд на один IP
-const limiter = rateLimit({
+// Определяем лимитер с задержкой 30 секунд на IP
+const generateImageLimiter = rateLimit({
   windowMs: 30 * 1000, // 30 секунд
-  max: 1, // лимит на каждые 30 секунд
-  message: 'Слишком много запросов с этого IP, пожалуйста, попробуйте позже. (Лимит 1 генерация в 30 секунд.',
+  max: 1, // лимит: 1 запрос на IP
+  handler: function (req, res /*next*/) {
+    return res.status(429).json({
+      error: "Слишком много запросов с этого IP, пожалуйста, попробуйте позже. (Лимит 1 генерация в 30 секунд)"
+    });
+  }
 });
 
-app.use(limiter);
-
-// Маршрут для генерации изображения с помощью DALL·E 3
-app.post('/generate-image', async (req, res) => {
+// Применяем лимитер только к определенному маршруту
+app.post('/generate-image', generateImageLimiter, async (req, res) => {
   const prompt = req.body.prompt;
 
   if (!prompt) {
@@ -25,7 +27,6 @@ app.post('/generate-image', async (req, res) => {
   }
 
   try {
-    // Запрос на API OpenAI для генерации изображения
     const response = await axios.post('https://api.openai.com/v1/images/generations', {
       prompt: prompt,
       n: 1,
@@ -39,7 +40,6 @@ app.post('/generate-image', async (req, res) => {
       },
     });
 
-    // Отправляем ответ клиенту
     res.json(response.data);
   } catch (error) {
     console.error(error);