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

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

const openai_key = process.env.OPENAI_KEY;

// Определяем лимитер с задержкой 30 секунд на IP
const generateImageLimiter = rateLimit({
  windowMs: 45 * 1000, // 30 секунд
  max: 1, // лимит: 1 запрос на IP
  handler: function (req, res) {
    return res.status(429).json({
      error: "Слишком много запросов с этого IP, пожалуйста, попробуйте позже. (Лимит 1 генерация изображения в 45 секунд)"
    });
  }
});

app.use('/generate-image', generateImageLimiter);

// Применяем лимитер только к определенному маршруту
app.post('/generate-image', async (req, res) => {
  const prompt = req.body.prompt;
  const apiKey = req.body.api || openai_key;

  if (!prompt) {
    return res.status(400).json({ error: 'Требуется параметр "prompt".' });
  }

  try {
    const response = await axios.post('https://api.openai.com/v1/images/generations', {
      prompt: prompt,
      n: 1,
      size: "1024x1024",
      quality: "hd",
      model: "dall-e-3",
    }, {
      headers: {
        'Authorization': `Bearer ${apiKey}`,
        'Content-Type': 'application/json',
      },
    });

    res.json(response.data);
  } catch (error) {
    console.error(error);
    res.status(500).json({ error: 'Произошла ошибка сервера при генерации изображения.' });
  }
});

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