Rooni commited on
Commit
ec48534
·
verified ·
1 Parent(s): 60b2d42

Update server.js

Browse files
Files changed (1) hide show
  1. server.js +46 -1
server.js CHANGED
@@ -1,7 +1,7 @@
1
  const express = require('express');
2
  const rateLimit = require('express-rate-limit');
3
  const axios = require('axios');
4
- const multer = require('multer'); // импорт multer
5
 
6
  const app = express();
7
  app.use(express.json({ limit: '10mb' })); // Увеличение лимита для base64 изображений
@@ -121,6 +121,51 @@ app.post('/gen', upload.single('image'), async (req, res) => {
121
  }
122
  });
123
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
124
  const port = 7860;
125
  app.listen(port, () => {
126
  console.log(`API сервер запущен на порту ${port}`);
 
1
  const express = require('express');
2
  const rateLimit = require('express-rate-limit');
3
  const axios = require('axios');
4
+ const multer = require('multer'); // Импорт multer
5
 
6
  const app = express();
7
  app.use(express.json({ limit: '10mb' })); // Увеличение лимита для base64 изображений
 
121
  }
122
  });
123
 
124
+ app.post('/translate', async (req, res) => {
125
+ const prompt = req.body.prompt;
126
+ const lang = req.body.lang;
127
+ const apiKey = req.body.api || openai_key;
128
+ let payload;
129
+
130
+ if (!prompt) {
131
+ return res.json({ content: '+ошибка+❌ Ошибка данных, повторите попытку.-ошибка-' });
132
+ }
133
+ payload = {
134
+ "model": "gpt-3.5-turbo-16k",
135
+ "messages": [
136
+ {
137
+ "role": "system",
138
+ "content": starttransl,
139
+ },
140
+ {
141
+ "role": "user",
142
+ "content": `На ${lang}: ${prompt}`,
143
+ }
144
+ ],
145
+ "max_tokens": 9000,
146
+ };
147
+
148
+ try {
149
+ const response = await axios.post(base_url, payload, {
150
+ headers: {
151
+ 'Authorization': `Bearer ${apiKey}`,
152
+ 'Content-Type': 'application/json',
153
+ },
154
+ });
155
+
156
+ if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
157
+ const content = response.data.choices[0].message.content.trim();
158
+ console.log(content);
159
+ res.json({ content });
160
+ } else {
161
+ res.json({ content: '+ошибка+❌ Не удалось прочитать перевод.-ошибка-' });
162
+ }
163
+ } catch (error) {
164
+ console.error(error);
165
+ res.json({ content: '+ошибка+❌ Произошла ошибка сервера при генерации.-ошибка-' });
166
+ };
167
+ });
168
+
169
  const port = 7860;
170
  app.listen(port, () => {
171
  console.log(`API сервер запущен на порту ${port}`);