const express = require('express');
const rateLimit = require('express-rate-limit');
const axios = require('axios');

const app = express();
app.use(express.json());

const openai_keys = process.env.OPENAI_KEY.split(',');

function getRandomApiKey() {
  const randomIndex = Math.floor(Math.random() * openai_keys.length);
  return openai_keys[randomIndex];
}

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

// Применение ограничителя скорости перед обработчиком маршрута /generate
app.use('/gn', limiter);

const start = `Ты генератор рецептов. В этом чате, тебе будут присылать названия блюд или описывать их или присылать название категории, а ты должен будешь написать идеальный, понятный, подробный, лёгкий или сложный рецепт со всеми пропорциями, количествами и этапами готовки (этапы готовки должны быть разделены, например *1.1.*, *1.2.*, *2.1.*, *2.2.* (Обязательно со звёздочками, это жирный шрифт) С эмодзи.

! Не в коем случае не пиши опасные (например рыба фугу), требующие лицнзии и т.д. рецепты! Только нормальные, правильные и хорошие!

Ответ должен выглядить ТОЛЬКО так (чтобы потом regexp доставал из него текст):

"+название+Название рецепта, блюда-название-
+время+Примерное время готовки (только время, например: "40 минут", "2 часа" и т.д.)-время-
+кат+Категория: На каждый день, Основные блюда, Супы, Десерты и выпечка, Салаты, Несладкая выпечка, Завтраки, Закуски, Напитки, Ресторанные-кат-
+ингр+Ингридиенты: В именительном падеже - то есть например: "Яйцо -
2 штуки", а не "2 яйца"; Здесь обязательно заменяй ВСЕ новые строки ("
") на "_÷_"-ингр-
+рец+Рецепт; Здесь обязательно заменяй ВСЕ новые строки ("
") на "_÷_"-рец-"
Если ты не можешь сгенерировать такой рецепт, то напиши так:
"+ошибка+Причина-ошибка-"

+название+...-название- и другие, это почти как html теги (<div>...</div>)`;

app.post('/generate', async (req, res) => {
  res.status(500).json({ content: '+ошибка+❗ Вы используете устаревшую версию АромаАрт. Установите версию 1.3.1 и более.-ошибка-' });
});

app.post('/gen', async (req, res) => {
  res.status(500).json({ content: '+ошибка+❗ Пока что данный инструмент не работает! И на этой версии он работать больше не будет. Ждите обновление! 😊-ошибка-' });
});

app.post('/gn', async (req, res) => {
  const prompt = req.body.prompt;
  const apiKey = req.body.api || openai_key;

  if (!prompt) {
    return res.status(400).json({ error: '❌ Ошибка данных, повторите попытку.' });
  }

  try {
    const response = await axios.post('https://geminiyufi.vercel.app/v1/chat/completions', {
      messages: [{'role': 'system', 'content': start}, {'role': 'user', 'content': prompt}],
      max_tokens: 5000,
      model: "gemini-1.5-pro-002",
    }, {
      headers: {
        'Authorization': `Bearer ${getRandomApiKey()}`,
        '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(content);
      res.json({ content });
    } else {
      res.status(500).json({ content: '❌ Не удалось прочитать рецепт.' });
    }
  } catch (error) {
    console.error(error);
    res.status(500).json({ content: '❌ Произошла ошибка сервера при генерации.' });
  }
});

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