Update server.js

server.js

@@ -1,6 +1,7 @@
+require('dotenv').config();
 const express = require('express');
 const rateLimit = require('express-rate-limit');
-const axios = require('axios');
+const { HfInference } = require('@huggingface/inference');
 
 const app = express();
 app.use(express.json());
@@ -8,8 +9,6 @@ app.use(express.json());
 // Доверие к одному прокси (например, Heroku)
 app.set('trust proxy', 1);
 
-const openai_keys = process.env.OPENAI_KEY.split(',');
-
 function getRandomApiKey() {
   const randomIndex = Math.floor(Math.random() * openai_keys.length);
   return openai_keys[randomIndex];
@@ -35,50 +34,35 @@ app.post('/update', async (req, res) => {
 });
 
 async function sendRequest(prompt, prs) {
-  const firstUrl = 'https://text.pollinations.ai/openai';
-  const secondUrl = 'https://text.pollinations.ai/openai';
-
-  try {
-    const response = await axios.post(firstUrl, {
-      messages: [{'role': 'system', 'content': prs}, {'role': 'user', 'content': prompt}],
-      max_tokens: 1200,
-      temperature: 0.19,
-    }, {
-      headers: {
-        'Content-Type': 'application/json',
-      },
-    });
-
-    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
-      return response.data.choices[0].message.content.trim();
-    } else {
-      throw new Error("Ошибка прочтения на первой ссылке");
-    }
-  } catch (error) {
-    console.error("Первая попытка не удалась:", error);
-    // Попытка с другой ссылкой
+    const hf_api_key = getRandomApiKey();
+    const client = new HfInference(hf_api_key);
+  
     try {
-      const response = await axios.post(secondUrl, {
-        messages: [{'role': 'system', 'content': prs}, {'role': 'user', 'content': prompt}],
+      const chatCompletion = await client.chatCompletion({
+        model: "Qwen/Qwen2.5-72B-Instruct",
+        messages: [
+          {
+            role: "system",
+            content: prs
+          },
+          {
+            role: "user",
+            content: prompt
+          }
+        ],
         max_tokens: 1200,
-        temperature: 0.19,
-      }, {
-        headers: {
-          'Content-Type': 'application/json',
-        },
       });
-
-      if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message) {
-        return response.data.choices[0].message.content.trim();
+  
+      if (chatCompletion && chatCompletion.choices && chatCompletion.choices.length > 0 && chatCompletion.choices[0].message) {
+        return chatCompletion.choices[0].message.content.trim();
       } else {
-        throw new Error("Ошибка прочтения на второй ссылке");
+        throw new Error("Ошибка прочтения ответа");
       }
-    } catch (secondError) {
-      console.error("Вторая попытка не удалась:", secondError);
+    } catch (error) {
+      console.error("Ошибка при обращении к Hugging Face:", error);
       throw new Error("Ошибка при генерации");
     }
   }
-}
 
 app.post('/pl', async (req, res) => {
   const prompt = req.body.prompt;
@@ -98,7 +82,7 @@ app.post('/pl', async (req, res) => {
     const content = await sendRequest(prompt, prs);
     res.json({ content });
   } catch (error) {
-    res.json({ content: `{"error":"", "title":"Ошибка", "text":"Произошла ошибка на сервере. (${error.message})", "okb":"Ок", "oklink":"", "cancelable":"true"}` });
+        res.json({ content: `{"error":"", "title":"Ошибка", "text":"Произошла ошибка на сервере. (${error.message})", "okb":"Ок", "oklink":"", "cancelable":"true"}` });
   }
 });
 
@@ -128,4 +112,4 @@ app.post('/plbeta', async (req, res) => {
 const port = 7860;
 app.listen(port, () => {
   console.log(`API сервер запущен на порту ${port}`);
-});
+});