Update plugins/faceswap.js

plugins/faceswap.js

@@ -30,6 +30,7 @@ async function faceSwap(img_url1, img_url2) {
     return {
       success: true,
       elapsedTime: swapResponse.data.elapsedTime,
+      result: swapResponse.data.result,
       url: imageUrl
     };
 
@@ -40,12 +41,12 @@ async function faceSwap(img_url1, img_url2) {
 
 const handler = async (req, res) => {
     try {
-        const { img1, img2 } = req.query;
+        const { img, img2, key } = req.query;
 
-        if (!img1) {
+        if (!img) {
             return res.status(400).json({
                 success: false,
-                error: 'Missing required parameter: img1'
+                error: 'Missing required parameter: img'
             });
         }
 
@@ -56,12 +57,23 @@ const handler = async (req, res) => {
             });
         }
 
-        const result = await faceSwap(img1, img2);
+        if (!key) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required parameter: key'
+            });
+        }
+
+        const result = await faceSwap(img, img2);
 
         return res.json({
             author: "Herza",
             success: true,
-            data: result.url
+            data: {
+                elapsedTime: result.elapsedTime,
+                result: result.result,
+                url: result.url
+            }
         });
 
     } catch (error) {
@@ -77,11 +89,11 @@ module.exports = {
     name: 'Face Swap',
     description: 'Swap faces between two images using AI',
     type: 'GET',
-    routes: ['api/ai/faceswap'],
+    routes: ['api/AI/faceswap'],
     tags: ['AI', 'tools'],
-    parameters: ['img1', 'img2', 'key'],
+    parameters: ['img', 'img2', 'key'],
     limit: 3,
     enabled: true,
-    main: ['ai', 'tools'],
+    main: ['AI', 'tools'],
     handler
 };