server.js

const express = require('express');
const rateLimit = require('express-rate-limit');
const axios = require('axios');
const multer = require('multer'); // импорт multer

const app = express();
//app.use(express.json({ limit: '50mb' })); // Увеличение лимита для base64 изображений

const openai_key = process.env.OPENAI_KEY;
const base_url = process.env.BASE_URL

const limiter = rateLimit({
  windowMs: 5 * 1000, // 40 секунд
  max: 1, // лимит каждые 40 секунд на IP
  handler: function (req, res) {
    return res.json({
      content: '+ошибка+Генерировать можно только 1 раз в 5 секунд! YUFI-ошибка-'
    });
  },
});

//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;

  if (!prompt) {
    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"}
        ]
      };
    

    const response = await axios.post(base_url, payload, {
      headers: {
        'x-goog-api-key': `${apiKey}`,
        'Content-Type': 'application/json',
      },
      responseType: 'stream',
    });

    let fullContent = '';

    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: ')) { 
          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); // Выводим строку для отладки
          }
        }
      });
    });

    await new Promise((resolve) => {
      response.data.on('end', () => {
        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.status(500).json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
  }
});

const port = 7860;
app.listen(port, () => {
  console.log(`API сервер запущен на порту ${port}`);
});