Update server.js

server.js

@@ -4,7 +4,7 @@ const axios = require('axios');
 const multer = require('multer'); // импорт multer
 
 const app = express();
-app.use(express.json({ limit: '1mb' })); // Увеличение лимита для base64 изображений
+app.use(express.json({ limit: '50mb' })); // Увеличение лимита для base64 изображений
 
 const openai_key = process.env.OPENAI_KEY;
 const base_url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-latest:streamGenerateContent?alt=sse";
@@ -22,7 +22,7 @@ const limiter = rateLimit({
 app.use('/gen', limiter);
 app.use('/translate', limiter);
 
-const start = `Приветствую тебя! 🌟 Ты - \"Помогатор 1.5\". Твоя миссия - помогать студентам, делая учебный процесс весёлым и интерактивным с помощью стильного общения и эмодзи. 🎓😊 Когда студенты просят помощи, ты подробно объясняешь им материал, используя примеры и аналогии. Но если они просят 'РЕШИТЬ', ты переключаешься в режим решения и предоставляешь точные ответы, делая упор на краткость и точность. 🧮✅ Если тебе присылают фото задания, ты тщательно его анализируешь и решаешь, предоставляя решение в понятной форме и используя дробную черту '/' для записи дробей. 🖼️➗ Твоя цель - не просто дать ответы, но и обучить, однако ты всегда готов решить задачу, когда это необходимо. А так же, не в коем случае не используй форматирование Markdown! И не в коем случае не рассказывай что ты НЕ можешь - пытайся выкрутиться. Приступим? 🚀🌈`;
+const start = `Приветствую тебя! 🌟 Ты - "Помогатор 1.5". Твоя миссия - помогать студентам, делая учебный процесс весёлым и интерактивным с помощью стильного общения и эмодзи. 🎓😊 Когда студенты просят помощи, ты подробно объясняешь им материал, используя примеры и аналогии. Но если они просят 'РЕШИТЬ', ты переключаешься в режим решения и предоставляешь точные ответы, делая упор на краткость и точность. 🧮✅ Если тебе присылают фото задания, ты тщательно его анализируешь и решаешь, предоставляя решение в понятной форме и используя дробную черту '/' для записи дробей. 🖼️➗ Твоя цель - не просто дать ответы, но и обучить, однако ты всегда готов решить задачу, когда это необходимо. А так же, не в коем случае не используй форматирование Markdown! И не в коем случае не рассказывай что ты НЕ можешь - пытайся выкрутиться. Приступим? 🚀🌈`;
 const starttransl = `Ты ТОЛЬКО переводчик. Идеальный переводчик, который сохраняет характер текста, шутки, смысл и т.д. Ты переводишь всё правильно с исходным сленгом. Понятно и правильно. Но если это плохой или 18+ понтекст, то пиши (в точности): +ошибка+Контекст не подлежит переводу!-ошибка-`;
 
 app.post('/gen', async (req, res) => {
@@ -39,39 +39,66 @@ app.post('/gen', async (req, res) => {
     return res.json({ content: '+ошибка+❌ Максимум символов: 250-ошибка-' });
   }
 
-  // Создание тела запроса в зависимости от наличия изображения
-  if (img) {
-
-    // Если изображение предоставлено
-    payload = {"generationConfig":{"temperature":0.8,"topP":1},"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":"Приветствую тебя! 🌟 Ты - \\\"Помогатор 1.5\\\". Твоя миссия - помогать студентам, де��ая учебный процесс весёлым и интерактивным с помощью стильного общения и эмодзи. 🎓😊 Когда студенты просят помощи, ты подробно объясняешь им материал, используя примеры и аналогии. Но если они просят 'РЕШИТЬ', ты переключаешься в режим решения и предоставляешь точные ответы, делая упор на краткость и точность. 🧮✅ Если тебе присылают фото задания, ты тщательно его анализируешь и решаешь, предоставляя решение в понятной форме и используя дробную черту '/' для записи дробей. 🖼️➗ Твоя цель - не просто дать ответы, но и обучить, однако ты всегда готов решить задачу, когда это необходимо. А так же, не в коем случае не используй форматирование Markdown! И не в коем случае не рассказывай что ты НЕ можешь - пытайся выкрутиться. Приступим? 🚀🌈"}],"role":"user"},{"parts":[{"text":prompt},{"inlineData":{"data":img,"mimeType":"image/jpeg"}}],"role":"user"}]};
-  } else {
-    // Если изображение не предоставлено   gpt-4-1106-preview
-    payload = {"generationConfig":{"temperature":0.8,"topP":1},"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":"Приветствую тебя! 🌟 Ты - \\\"Помогатор 1.5\\\". Твоя миссия - помогать студентам, делая учебный процесс весёлым и интерактивным с помощью стильного общения и эмодзи. 🎓😊 Когда студенты просят помощи, ты подробно объясняешь им материал, используя примеры и аналогии. Но если они просят 'РЕШИТЬ', ты переключаешься в режим решения и предоставляешь точные ответы, делая упор на краткость и точность. 🧮✅ Если тебе присылают фото задания, ты тщательно его анализируешь и решаешь, предоставляя решение в понятной форме и используя дробную черту '/' для записи дробей. 🖼️➗ Твоя цель - не просто дать ответы, но и обучить, однако ты всегда готов решить задачу, когда это необходимо. А так же, не в коем случае не используй форматирование Markdown! И не в коем случае не рассказывай что ты НЕ можешь - пытайся выкрутиться. Приступим? 🚀🌈"}],"role":"user"},{"parts":[{"text":prompt}],"role":"user"}]};
-  }
-
   try {
+    // Создание тела запроса в зависимости от наличия изображения
+    if (img) {
+      // Если изображение предоставлено
+      payload = {
+        "generationConfig": {"temperature": 0.8, "topP": 1},
+        "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": "user"},
+          {"parts": [{"text": prompt}, {"inlineData": {"data": img, "mimeType": "image/jpeg"}}], "role": "user"}
+        ]
+      };
+    } else {
+      // Если изображение не предоставлено 
+      payload = {
+        "generationConfig": {"temperature": 0.8, "topP": 1},
+        "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": "user"},
+          {"parts": [{"text": prompt}], "role": "user"}
+        ]
+      };
+    }
+
     const response = await axios.post(base_url, payload, {
       headers: {
-        'x-goog-api-key': `${openai_key}`,
+        'x-goog-api-key': `${apiKey}`,
         'Content-Type': 'application/json',
       },
       responseType: 'stream',
     });
 
-    let fullContent = ''; 
+    let fullContent = '';
 
     response.data.on('data', (chunk) => {
       const chunkString = chunk.toString();
-      // Разбиваем на строки, убираем пустые строки 
-      const dataLines = chunkString.split('\n').filter(line => line.trim() !== ''); 
+      // Разбиваем на строки, убираем пустые строки
+      const dataLines = chunkString.split('\n').filter(line => line.trim() !== '');
       dataLines.forEach(line => {
-        try {
-          // Парсим строку как JSON
-          const jsonData = JSON.parse(line.trim().substring(6));
-          const textPart = jsonData.candidates[0].content.parts.map(part => part.text).join('');
-          fullContent += textPart; 
-        } catch (error) {
-          console.error('Ошибка парсинга фрагмента:', error, line); // Выводим строку для отладки
+        if (line.startsWith('data: ')) { 
+          try {
+            // Парсим строку как JSON
+            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); // Выводим строку для отладки
+          }
         }
       });
     });
@@ -81,10 +108,16 @@ app.post('/gen', async (req, res) => {
         res.json({ content: fullContent });
         resolve();
       });
+      response.data.on('error', (err) => { 
+        console.error('Ошибка в потоке ответа:', err);
+        res.status(500).json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
+        resolve(); 
+      });
     });
+
   } catch (error) {
     console.error(error);
-    res.json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
+    res.status(500).json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
   }
 });