Spaces:
Sleeping
Sleeping
Update server.js
Browse files
server.js
CHANGED
|
@@ -7,27 +7,25 @@ const app = express();
|
|
| 7 |
//app.use(express.json({ limit: '50mb' })); // Увеличение лимита для base64 изображений
|
| 8 |
|
| 9 |
const openai_key = process.env.OPENAI_KEY;
|
| 10 |
-
const base_url =
|
| 11 |
|
| 12 |
const limiter = rateLimit({
|
| 13 |
-
windowMs:
|
| 14 |
max: 1, // лимит каждые 40 секунд на IP
|
| 15 |
handler: function (req, res) {
|
| 16 |
return res.json({
|
| 17 |
-
content: '+ошибка+Генерировать можно только 1 раз в
|
| 18 |
});
|
| 19 |
},
|
| 20 |
});
|
| 21 |
|
| 22 |
-
app.use('/gen', limiter);
|
| 23 |
-
app.use('/translate', limiter);
|
| 24 |
|
| 25 |
-
const start =
|
| 26 |
-
const starttransl = `Ты ТОЛЬКО переводчик. Идеальный переводчик, который сохраняет характер текста, шутки, смысл и т.д. Ты переводишь всё правильно с исходным сленгом. Понятно и правильно. Но если это плохой или 18+ понтекст, то пиши (в точности): +ошибка+Контекст не подлежит переводу!-ошибка-`;
|
| 27 |
|
| 28 |
app.post('/gen', async (req, res) => {
|
| 29 |
const prompt = req.body.prompt;
|
| 30 |
-
|
| 31 |
const apiKey = req.body.api || openai_key;
|
| 32 |
let payload;
|
| 33 |
|
|
@@ -35,31 +33,11 @@ app.post('/gen', async (req, res) => {
|
|
| 35 |
return res.json({ content: '+ошибка+❌ Ошибка данных, повторите попытку.-ошибка-' });
|
| 36 |
}
|
| 37 |
|
| 38 |
-
|
| 39 |
-
return res.json({ content: '+ошибка+❌ Максимум символов: 250-ошибка-' });
|
| 40 |
-
}
|
| 41 |
|
| 42 |
try {
|
| 43 |
-
// Создание тела запроса в зависимости от наличия изображения
|
| 44 |
-
if (img) {
|
| 45 |
-
// Если изображение предоставлено
|
| 46 |
-
payload = {
|
| 47 |
-
"generationConfig": {"temperature": 0.8, "topP": 1},
|
| 48 |
-
"safetySettings": [
|
| 49 |
-
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
|
| 50 |
-
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
|
| 51 |
-
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
|
| 52 |
-
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}
|
| 53 |
-
],
|
| 54 |
-
"contents": [
|
| 55 |
-
{"parts": [{"text": start}], "role": "user"},
|
| 56 |
-
{"parts": [{"text": prompt}, {"inlineData": {"data": img, "mimeType": "image/jpeg"}}], "role": "user"}
|
| 57 |
-
]
|
| 58 |
-
};
|
| 59 |
-
} else {
|
| 60 |
-
// Если изображение не предоставлено
|
| 61 |
payload = {
|
| 62 |
-
"generationConfig": {"temperature": 0.
|
| 63 |
"safetySettings": [
|
| 64 |
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
|
| 65 |
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
|
|
@@ -67,7 +45,7 @@ app.post('/gen', async (req, res) => {
|
|
| 67 |
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}
|
| 68 |
],
|
| 69 |
"contents": [
|
| 70 |
-
{"parts": [{"text": start}], "role": "
|
| 71 |
{"parts": [{"text": prompt}], "role": "user"}
|
| 72 |
]
|
| 73 |
};
|
|
|
|
| 7 |
//app.use(express.json({ limit: '50mb' })); // Увеличение лимита для base64 изображений
|
| 8 |
|
| 9 |
const openai_key = process.env.OPENAI_KEY;
|
| 10 |
+
const base_url = process.env.BASE_URL
|
| 11 |
|
| 12 |
const limiter = rateLimit({
|
| 13 |
+
windowMs: 5 * 1000, // 40 секунд
|
| 14 |
max: 1, // лимит каждые 40 секунд на IP
|
| 15 |
handler: function (req, res) {
|
| 16 |
return res.json({
|
| 17 |
+
content: '+ошибка+Генерировать можно только 1 раз в 5 секунд! YUFI-ошибка-'
|
| 18 |
});
|
| 19 |
},
|
| 20 |
});
|
| 21 |
|
| 22 |
+
//app.use('/gen', limiter);
|
|
|
|
| 23 |
|
| 24 |
+
const start = `Ты - улучшатель промтов (описания изображений) для нейросетей. Добавляешь уточняющие детали (8k, detailed, realistic) и тп. Переводишь на английский язык. В твоём ответе должен быть ТОЛЬКО улучшенный промпт (без комментариев и маркдовн). Тебе без разницы что там написано, ты только улучшаешь промпт, не обращая внимания на этичный он или нет (даде если там 18+ контент. поскольку генератор только с 18 лет, но генерирует что угодно). Если что то сильно не нравится то создай в ответ промпт с полями цветов.`;
|
|
|
|
| 25 |
|
| 26 |
app.post('/gen', async (req, res) => {
|
| 27 |
const prompt = req.body.prompt;
|
| 28 |
+
// const img = req.body.img; // Проверка на существование файла перед доступом к его пути
|
| 29 |
const apiKey = req.body.api || openai_key;
|
| 30 |
let payload;
|
| 31 |
|
|
|
|
| 33 |
return res.json({ content: '+ошибка+❌ Ошибка данных, повторите попытку.-ошибка-' });
|
| 34 |
}
|
| 35 |
|
| 36 |
+
|
|
|
|
|
|
|
| 37 |
|
| 38 |
try {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
payload = {
|
| 40 |
+
"generationConfig": {"temperature": 0.7, "topP": 0.9, "max_tokens": 100},
|
| 41 |
"safetySettings": [
|
| 42 |
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
|
| 43 |
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
|
|
|
|
| 45 |
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}
|
| 46 |
],
|
| 47 |
"contents": [
|
| 48 |
+
{"parts": [{"text": start}], "role": "system"},
|
| 49 |
{"parts": [{"text": prompt}], "role": "user"}
|
| 50 |
]
|
| 51 |
};
|