Spaces:
Sleeping
Sleeping
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: '10mb' })); // Увеличение лимита для base64 изображений | |
const openai_key = process.env.OPENAI_KEY; | |
const base_url = process.env.BASE_URL; | |
const limiter = rateLimit({ | |
windowMs: 40 * 1000, // 40 секунд | |
max: 1, // лимит каждые 40 секунд на IP | |
handler: function (req, res) { | |
return res.json({ | |
content: '+ошибка+Генерировать можно только 1 раз в 40 секунд! YUFI-ошибка-' | |
}); | |
}, | |
}); | |
app.use('/gen', limiter); | |
app.use('/translate', limiter); | |
const start = `Приветствую тебя! 🌟 Ты - \"Помогатор 1.5\". Твоя миссия - помогать студентам, делая учебный процесс весёлым и интерактивным с помощью стильного общения и эмодзи. 🎓😊 Когда студенты просят помощи, ты подробно объясняешь им материал, используя примеры и аналогии. Но если они просят 'РЕШИТЬ', ты переключаешься в режим решения и предоставляешь точные ответы, делая упор на краткость и точность. 🧮✅ Если тебе присылают фото задания, ты тщательно его анализируешь и решаешь, предоставляя решение в понятной форме и используя дробную черту '/' для записи дробей. 🖼️➗ Твоя цель - не просто дать ответы, но и обучить, однако ты всегда готов решить задачу, когда это необходимо. Приступим? 🚀🌈` | |
const starttransl = `Ты ТОЛЬКО переводчик. Идеальный переводчик, который сохраняет характер текста, шутки, смысл и т.д. Ты переводишь всё правильно с исходным сленгом. Понятно и правильно. Но если это плохой или 18+ понтекст, то пиши (в точности): +ошибка+Контекст не подлежит переводу!-ошибка-`; | |
app.post('/gener', async (req, res) => { | |
return res.json({ content: '+ошибка+❗ Проект закрыт и больше не функционирует из-за отсутствия финансирования!-ошибка-' }); | |
}); | |
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 img = req.file.path; // Путь к загруженному изображению | |
const apiKey = req.body.api || openai_key; | |
let payload; | |
if (!prompt) { | |
return res.json({ content: '+ошибка+❌ Ошибка данных, повторите попытку.-ошибка-' }); | |
} | |
if (prompt.replace(/[\s\n]/g, '').length > 250) { | |
return res.json({ content: '+ошибка+❌ Максимум символов: 250-ошибка-' }); | |
} | |
// Создание тела запроса в зависимости от наличия изображения | |
if (img) { | |
// Если изображение предоставлено | |
payload = { | |
"model": "gpt-4-vision-preview", | |
"messages": [ | |
{ | |
"role": "user", | |
"content": start, | |
}, | |
{ | |
"role": "assistant", | |
"content": "Хорошо 😊", | |
}, | |
{ | |
"role": "user", | |
"content": [ | |
{ | |
"type": "text", | |
"text": prompt, | |
}, | |
{ | |
"type": "image_url", | |
"image_url": { | |
"url": `http://${req.headers.host}/uploads/${img}`, | |
}, | |
}, | |
], | |
} | |
], | |
}; | |
} else { | |
// Если изображение не предоставлено | |
payload = { | |
"model": "gpt-4-1106-preview", | |
"messages": [ | |
{ | |
"role": "user", | |
"content": start, | |
}, | |
{ | |
"role": "user", | |
"content": prompt, | |
} | |
], | |
}; | |
} | |
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: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' }); | |
} | |
}); | |
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}`); | |
}); |