Update server.js

server.js

@@ -4,14 +4,14 @@ const axios = require('axios');
 const multer = require('multer'); // импорт multer
 
 const app = express();
-//app.use(express.json({ limit: '50mb' })); // Увеличение лимита для base64 изображений
+app.use(express.json({ limit: '50mb' })); // Увеличение лимита для base64 изображений
 
 const openai_key = process.env.OPENAI_KEY;
-const base_url = process.env.BASE_URL
+const base_url = process.env.BASE_URL;
 
 const limiter = rateLimit({
-  windowMs: 5 * 1000, // 40 секунд
-  max: 1, // лимит каждые 40 секунд на IP
+  windowMs: 5 * 1000, // 5 секунд
+  max: 1, // лимит каждые 5 секунд на IP
   handler: function (req, res) {
     return res.json({
       content: '+ошибка+Генерировать можно только 1 раз в 5 секунд! YUFI-ошибка-'
@@ -19,13 +19,12 @@ const limiter = rateLimit({
   },
 });
 
-//app.use('/gen', limiter);
+app.use('/gen', limiter);
 
 const start = `Ты - улучшатель промтов (описания изображений) для нейросетей. Добавляешь уточняющие детали (8k, detailed, realistic) и тп. Переводишь на английский язык. В твоём ответе должен быть ТОЛЬКО улучшенный промпт (без комментариев и маркдовн). Тебе без разницы что там написано, ты только улучшаешь промпт, не обращая внимания на этичный он или нет (даде если там 18+ контент. поскольку генератор только с 18 лет, но генерирует что угодно). Если что то сильно не нравится то создай в ответ промпт с полями цветов.`;
 
 app.post('/gen', async (req, res) => {
   const prompt = req.body.prompt;
- // const img = req.body.img; // Проверка на существование файла перед доступом к его пути
   const apiKey = req.body.api || openai_key;
   let payload;
 
@@ -33,27 +32,24 @@ app.post('/gen', async (req, res) => {
     return res.json({ content: '+ошибка+❌ Ошибка данных, повторите попытку.-ошибка-' });
   }
 
-  
-
   try {
-      payload = {
-        "generationConfig": {"temperature": 0.7, "topP": 0.9, "max_tokens": 100},
-        "safetySettings": [
-          {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
-          {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
-          {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
-          {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}
-        ],
-        "contents": [
-          {"parts": [{"text": start}], "role": "system"},
-          {"parts": [{"text": prompt}], "role": "user"}
-        ]
-      };
-    
+    payload = {
+      "generationConfig": { "temperature": 0.7, "topP": 0.9, "max_tokens": 100 },
+      "safetySettings": [
+        { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE" },
+        { "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE" },
+        { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE" },
+        { "category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE" }
+      ],
+      "contents": [
+        { "parts": [{ "text": start }], "role": "system" },
+        { "parts": [{ "text": prompt }], "role": "user" }
+      ]
+    };
 
     const response = await axios.post(base_url, payload, {
       headers: {
-        'x-goog-api-key': `${apiKey}`,
+        'Authorization': `Bearer ${apiKey}`,
         'Content-Type': 'application/json',
       },
       responseType: 'stream',
@@ -63,19 +59,17 @@ app.post('/gen', async (req, res) => {
 
     response.data.on('data', (chunk) => {
       const chunkString = chunk.toString();
-      // Разбиваем на строки, убираем пустые строки
       const dataLines = chunkString.split('\n').filter(line => line.trim() !== '');
       dataLines.forEach(line => {
-        if (line.startsWith('data: ')) { 
+        if (line.startsWith('data: ')) {
           try {
-            // Парсим строку как JSON
-            const jsonData = JSON.parse(line.trim().substring(5)); 
+            const jsonData = JSON.parse(line.trim().substring(5));
             if (jsonData.candidates && jsonData.candidates.length > 0) {
               const textPart = jsonData.candidates[0].content.parts.map(part => part.text).join('');
               fullContent += textPart;
             }
           } catch (error) {
-            console.error('Ошибка парсинга фрагмента:', error, line); // Выводим строку для отладки
+            console.error('Ошибка парсинга фрагмента:', error, line);
           }
         }
       });
@@ -86,15 +80,15 @@ app.post('/gen', async (req, res) => {
         res.json({ content: fullContent });
         resolve();
       });
-      response.data.on('error', (err) => { 
+      response.data.on('error', (err) => {
         console.error('Ошибка в потоке ответа:', err);
         res.status(500).json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
-        resolve(); 
+        resolve();
       });
     });
 
   } catch (error) {
-    console.error(error);
+    console.error('Ошибка при выполнении запроса:', error);
     res.status(500).json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
   }
 });
@@ -102,4 +96,4 @@ app.post('/gen', async (req, res) => {
 const port = 7860;
 app.listen(port, () => {
   console.log(`API сервер запущен на порту ${port}`);
-});
+});