Update server.js

server.js

@@ -1,86 +1,46 @@
 const express = require('express');
-const rateLimit = require('express-rate-limit');
-const axios = require('axios');
+const fetch = require('node-fetch');
+const bodyParser = require('body-parser');
 
 const app = express();
-app.use(express.json());
+const port = 3000;
 
-const openai_key = process.env.OPENAI_KEY;
+// Используем body-parser для обработки JSON данных
+app.use(bodyParser.json());
 
-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.* (Обязательно со звёздочками, это жирный шрифт) С эмодзи.
-
-! Не в коем случае не пиши опасные (например рыба фугу), требующие лицнзии и т.д. рецепты! Только нормальные, правильные и хорошие!
+app.post('/generate-image', async (req, res) => {
+    const { prompt } = req.body;
 
-Ответ должен выглядить ТОЛЬКО так (чтобы потом 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: 2000,
-      model: "gemini-1.5-pro-latest",
-    }, {
-      headers: {
-        'Authorization': `Bearer ${apiKey}`,
-        'Content-Type': 'application/json',
-      },
-    });
+    if (!prompt) {
+        return res.status(400).send('Missing "prompt" in request body');
+    }
 
-    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: '❌ Не удалось прочитать рецепт.' });
+    try {
+        const response = await fetch('https://api-inference.huggingface.co/models/your-model-name', {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${process.env.KEY}`,
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({
+                inputs: prompt
+            })
+        });
+
+        if (!response.ok) {
+            throw new Error(`Error from Hugging Face API: ${response.statusText}`);
+        }
+
+        const imageBuffer = await response.buffer();
+        const base64Image = imageBuffer.toString('base64');
+
+        res.json({ image: base64Image });
+    } catch (error) {
+        console.error('Error generating image:', error);
+        res.status(500).send('Error generating image');
     }
-  } catch (error) {
-    console.error(error);
-    res.status(500).json({ content: '❌ Произошла ошибка сервера при генерации.' });
-  }
 });
 
-const port = 7860;
 app.listen(port, () => {
-  console.log(`API сервер запущен на порту ${port}`);
-});
+    console.log(`Server is running on http://localhost:${port}`);
+});