Update server.js

server.js

@@ -1,112 +1,144 @@
 const express = require('express');
 const rateLimit = require('express-rate-limit');
 const axios = require('axios');
-const dotenv = require('dotenv').config();
-
 
 const app = express();
 app.use(express.json());
 
-const apiKeys = process.env.OPENAI_KEYS.split(',');
-
-const start = process.env.start;
+const openai_key = process.env.OPENAI_KEY;
 
 const limiter = rateLimit({
-  windowMs: 5 * 1000, // 5 секунд
-  max: 1, // лимит на 1 запрос каждые 5 секунд на IP
+  windowMs: 5 * 1000, // 30 секунд
+  max: 1, // лимит каждые 30 секунд на IP
   handler: function (req, res) {
     return res.status(429).json("wait");
   },
 });
 
-// Применение ограничителя скорости перед обработчиком маршрута /pl
-app.use('/pl', limiter);
+// Применение ограничителя скорости перед обработчиком маршрута /generate
+ app.use('/pl', limiter);
+
+const start = `${process.env.start}`;
 
-app.post('/cr', (req, res) => {
-  res.json({
-    content: '{"whate":"🪨","howe":"ОБНОВИТЕСЬ","text":"Текущая версия приложения устарела. Установите новую из нашего телеграм канала: @yufi_ru","succ":"победа","what":"Версию","how":"Обновите","howl":"@yufi_ru"}'
-  });
+app.post('/cr', async (req, res) => {
+  res.json({ content: `{"whate":"🪨", "howe":"ОБНОВИТЕСЬ", "text":"Текущая версия приложения устарела. Установите новую из нашего телеграм канала: @yufi_ru", "succ":"победа", "what":"Версию", "how":"Обновите", "howl":"@yufi_ru"}` });
 });
 
-// Функция для получения случайного API ключа
-const getRandomApiKey = () => apiKeys[Math.floor(Math.random() * apiKeys.length)];
+app.post('/pl', async (req, res) => {
+  const prompt = req.body.prompt;
+  const apiKey = req.body.api || openai_key;
+
+  if (!prompt) {
+    return res.status(400).json("wait"); // Не удалось принять данные
+  }
 
-const generateResponse = async (url, prompt, additionalMessages, temperature) => {
   try {
-    const response = await axios.post(url, {
-      messages: [
-        { role: 'system', content: start },
-        ...additionalMessages,
-        { role: 'user', content: prompt }
-      ],
+    const response = await axios.post('https://openai-gemini-iota.vercel.app/v1/chat/completions', {
+      messages: [{'role': 'system', 'content': start}, {'role': 'user', 'content': prompt}],
       max_tokens: 2000,
-      temperature: temperature,
-      model: additionalMessages.length > 0 ? "gemini-1.5-pro-002" : "deepseek-chat",
+      temperature: 0.8,
+     // presence_penalty: 0.0,
+      //frequency_penalty: -0.1,
+      //model: "gemini-1.5-pro-latest",
+      model: "deepseek-chat",
+      //model: "gemini-1.5-flash-latest",
     }, {
       headers: {
-        'Authorization': `Bearer ${getRandomApiKey()}`,
+        'Authorization': `Bearer ${apiKey}`,
         'Content-Type': 'application/json',
       },
     });
 
-    return response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message
-      ? response.data.choices[0].message.content.trim()
-      : null;
-
+    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
+      const content = response.data.choices[0].message.content.trim();
+      //console.log(`\n---\nПользователь: ${prompt}\n    Ответ:\n      ${content}`);
+      res.json({ content });
+    } else {
+      res.status(500).json({ content: 'errora' }); // Ошибка прочтения
+    }
   } catch (error) {
     console.error(error);
-    throw new Error('Error during generation');
-  }
-};
-
-app.post('/pl', async (req, res) => {
-  const prompt = req.body.prompt;
-
-  if (!prompt) {
-    return res.status(400).json("wait"); // Не удалось принять данные
-  }
-
-  try {
-    const content = await generateResponse('https://openai-gemini-iota.vercel.app/v1/chat/completions', prompt, [], 0.8);
-    content ? res.json({ content }) : res.status(500).json({ content: 'errora' });
-  } catch {
-    res.status(500).json({ content: 'errorb' });
+    res.status(500).json({ content: 'errorb' }); // ❌ Произошла ошибка сервера при генерации.
   }
 });
 
 app.post('/cre', async (req, res) => {
   const prompt = req.body.prompt;
+  const apiKey = req.body.api || openai_key;
 
   if (!prompt) {
     return res.status(400).json("wait"); // Не удалось принять данные
   }
 
   try {
-    const additionalMessages = [
-      { role: 'system', content: 'И ОБЯЗАТЕЛЬНО, всегда в конце "text" проси обновить приложение, уведомляя о том, что скоро в жтой версии перестанешь отвечат��!' }
-    ];
-    const content = await generateResponse('https://vercel.com/roonis-projects/openai-gemini/v1/chat/completions', prompt, additionalMessages, 0.6);
-    content ? res.json({ content }) : res.status(500).json({ content: 'errora' });
-  } catch {
-    res.status(500).json({ content: 'errorb' });
+    const response = await axios.post('https://vercel.com/roonis-projects/openai-gemini/v1/chat/completions', {
+      messages: [{'role': 'system', 'content': start}, {'role': 'system', 'content': 'И ОБЯЗАТЕЛЬНО, всегда в конце "text" проси обновить приложение, уведомляя о том, что скоро в жтой версии перестанешь отвечать!'}, {'role': 'user', 'content': prompt}],
+      max_tokens: 2000,
+      temperature: 0.6,
+     // presence_penalty: 0.0,
+      //frequency_penalty: -0.1,
+      model: "gemini-1.5-pro-002",
+    //  model: "deepseek-chat",
+      //model: "gemini-1.5-flash-latest",
+    }, {
+      headers: {
+        'Authorization': `Bearer ${apiKey}`,
+        'Content-Type': 'application/json',
+      },
+    });
+
+    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
+      const content = response.data.choices[0].message.content.trim();
+    //  console.log(`\n---\nПользователь: ${prompt}\n    Ответ:\n      ${content}`);
+      res.json({ content });
+    } else {
+      res.status(500).json({ content: 'errora' }); // Ошибка прочтения
+    }
+  } catch (error) {
+    console.error(error);
+    res.status(500).json({ content: 'errorb' }); // ❌ Произошла ошибка сервера при генерации.
   }
 });
 
 app.post('/crebeta', async (req, res) => {
   const prompt = req.body.prompt;
+  const apiKey = req.body.api || openai_key;
 
   if (!prompt) {
     return res.status(400).json("wait"); // Не удалось принять данные
   }
 
   try {
-    const content = await generateResponse('https://geminiyufi.vercel.app/v1/chat/completions', prompt, [], 0.24);
-    content ? res.json({ content }) : res.status(500).json({ content: 'errora' });
-  } catch {
-    res.status(500).json({ content: 'errorb' });
+    const response = await axios.post('https://geminiyufi.vercel.app/v1/chat/completions', {
+      messages: [{'role': 'system', 'content': start}, {'role': 'user', 'content': prompt}],
+      max_tokens: 2000,
+      temperature: 0.24,
+     // presence_penalty: 0.0,
+      //frequency_penalty: -0.1,
+      model: "gemini-1.5-pro-002",
+     // model: "gemma-2-27b-it",
+      //model: "gemini-1.5-flash-latest",
+    }, {
+      headers: {
+        'Authorization': `Bearer ${apiKey}`,
+        'Content-Type': 'application/json',
+      },
+    });
+
+    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
+      const content = response.data.choices[0].message.content.trim();
+     // console.log(`\n---\nПользователь: ${prompt}\n    Ответ:\n      ${content}`);
+      res.json({ content });
+    } else {
+      res.status(500).json({ content: 'errora' }); // Ошибка прочтения
+    }
+  } catch (error) {
+    console.error(error);
+    res.status(500).json({ content: 'errorb' }); // ❌ Произошла ошибка сервера при генерации.
   }
 });
 
+
 const port = 7860;
 app.listen(port, () => {
   console.log(`API сервер запущен на порту ${port}`);