Update plugins/faceswap.js

plugins/faceswap.js

@@ -1,8 +1,38 @@
 const axios = require('axios');
-const upload = require('../lib/uploadImage');
+const FormData = require('form-data');
+const fs = require('fs');
+const path = require('path');
+
+async function uploadToTmpFiles(buffer, filename) {
+    const tempDir = path.join(__dirname, '..', 'temp');
+    
+    if (!fs.existsSync(tempDir)) {
+        fs.mkdirSync(tempDir, { recursive: true });
+    }
+    
+    const tempFile = path.join(tempDir, filename);
+    fs.writeFileSync(tempFile, buffer);
+    
+    const form = new FormData();
+    form.append('file', fs.createReadStream(tempFile));
+    
+    const response = await axios.post('https://tmpfiles.org/api/v1/upload', form, {
+        headers: {
+            ...form.getHeaders()
+        }
+    });
+    
+    if (fs.existsSync(tempFile)) {
+        fs.unlinkSync(tempFile);
+    }
+    
+    const uploadUrl = response.data.data.url;
+    const directUrl = uploadUrl.replace('tmpfiles.org/', 'tmpfiles.org/dl/');
+    
+    return directUrl;
+}
 
 async function faceSwap(img_url1, img_url2) {
-  try {
     const response1 = await axios.get(img_url1, { responseType: 'arraybuffer' });
     const base64_1 = Buffer.from(response1.data).toString('base64');
     
@@ -10,33 +40,29 @@ async function faceSwap(img_url1, img_url2) {
     const base64_2 = Buffer.from(response2.data).toString('base64');
 
     const swapResponse = await axios.post('https://api.faceswapper.ai/swap', {
-      target: base64_1,
-      source: base64_2,
-      type: 'invisible',
-      id: 'faceswapper'
+        target: base64_1,
+        source: base64_2,
+        type: 'invisible',
+        id: 'faceswapper'
     }, {
-      headers: {
-        'Accept': 'application/json, text/plain, */*',
-        'Content-Type': 'application/json',
-        'type': 'rapid'
-      }
+        headers: {
+            'Accept': 'application/json, text/plain, */*',
+            'Content-Type': 'application/json',
+            'type': 'rapid'
+        }
     });
 
     const base64Data = swapResponse.data.result.replace(/^data:image\/\w+;base64,/, '');
     const buffer = Buffer.from(base64Data, 'base64');
     
-    const imageUrl = await upload(buffer);
+    const imageUrl = await uploadToTmpFiles(buffer, `faceswap_${Date.now()}.png`);
 
     return {
-      success: true,
-      elapsedTime: swapResponse.data.elapsedTime,
-      result: swapResponse.data.result,
-      url: imageUrl
+        success: true,
+        elapsedTime: swapResponse.data.elapsedTime,
+        result: swapResponse.data.result,
+        url: imageUrl
     };
-
-  } catch (error) {
-    throw error;
-  }
 }
 
 const handler = async (req, res) => {
@@ -45,6 +71,7 @@ const handler = async (req, res) => {
 
         if (!img) {
             return res.status(400).json({
+                author: "Herza",
                 success: false,
                 error: 'Missing required parameter: img'
             });
@@ -52,6 +79,7 @@ const handler = async (req, res) => {
 
         if (!img2) {
             return res.status(400).json({
+                author: "Herza",
                 success: false,
                 error: 'Missing required parameter: img2'
             });
@@ -59,6 +87,7 @@ const handler = async (req, res) => {
 
         if (!key) {
             return res.status(400).json({
+                author: "Herza",
                 success: false,
                 error: 'Missing required parameter: key'
             });
@@ -78,6 +107,7 @@ const handler = async (req, res) => {
 
     } catch (error) {
         res.status(500).json({
+            author: "Herza",
             success: false,
             error: error.message,
             timestamp: new Date().toISOString()