Update server.js

server.js

@@ -1,19 +1,13 @@
 const express = require('express');
 const rateLimit = require('express-rate-limit');
 const axios = require('axios');
-const multer = require('multer'); // Импорт multer
-const ImageKit = require('imagekit-javascript'); // Импорт ImageKit
+const multer = require('multer'); // импорт multer
 
 const app = express();
 app.use(express.json({ limit: '10mb' })); // Увеличение лимита для base64 изображений
 
 const openai_key = process.env.OPENAI_KEY;
 const base_url = process.env.BASE_URL;
-const imagekit = new ImageKit({
-  publicKey: process.env.IMAGEKIT_PUBLIC_KEY,
-  privateKey: process.env.IMAGEKIT_PRIVATE_KEY,
-  urlEndpoint: process.env.IMAGEKIT_URL_ENDPOINT,
-});
 
 const limiter = rateLimit({
   windowMs: 40 * 1000, // 40 секунд
@@ -35,9 +29,18 @@ app.post('/gener', async (req, res) => {
   return res.json({ content: '+ошибка+❗ Проект закрыт и больше не функционирует из-за отсутствия финансирования!-ошибка-' });
 });
 
-app.post('/gen', async (req, res) => {
+const storage = multer.diskStorage({
+  destination: './uploads',
+  filename: (req, file, cb) => {
+    cb(null, Date.now() + '-' + file.originalname);
+  },
+});
+
+const upload = multer({ storage });
+
+app.post('/gen', upload.single('image'), async (req, res) => {
   const prompt = req.body.prompt;
-  const imageUrl = req.body.imageUrl;
+  const img = req.file ? req.file.path : undefined; // Проверка на существование файла перед доступом к его пути
   const apiKey = req.body.api || openai_key;
   let payload;
 
@@ -50,8 +53,8 @@ app.post('/gen', async (req, res) => {
   }
 
   // Создание тела запроса в зависимости от наличия изображения
-  if (imageUrl) {
-    // Если изображение предоставляется по URL
+  if (img) {
+    // Если изображение предоставлено
     payload = {
       "model": "gpt-4-vision-preview",
       "messages": [
@@ -73,7 +76,7 @@ app.post('/gen', async (req, res) => {
             {
               "type": "image_url",
               "image_url": {
-                "url": imageUrl,
+                "url": `http://${req.headers.host}/uploads/${img}`,
               },
             },
           ],
@@ -118,51 +121,6 @@ app.post('/gen', async (req, res) => {
   }
 });
 
-app.post('/translate', async (req, res) => {
-  const prompt = req.body.prompt;
-  const lang = req.body.lang;
-  const apiKey = req.body.api || openai_key;
-  let payload;
-
-  if (!prompt) {
-    return res.json({ content: '+ошибка+❌ Ошибка данных, повторите попытку.-ошибка-' });
-  }
-  payload = {
-      "model": "gpt-3.5-turbo-16k",
-      "messages": [
-        {
-          "role": "system",
-          "content": starttransl,
-        },
-        {
-          "role": "user",
-          "content": `На ${lang}: ${prompt}`,
-        }
-      ],
-      "max_tokens": 9000,
-    };
-
-  try {
-    const response = await axios.post(base_url, payload, {
-      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(content);
-      res.json({ content });
-    } else {
-      res.json({ content: '+ошибка+❌ Не удалось прочитать перевод.-ошибка-' });
-    }
-  } catch (error) {
-    console.error(error);
-    res.json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
-  };
-});
-
 const port = 7860;
 app.listen(port, () => {
   console.log(`API сервер запущен на порту ${port}`);