Update egg-api.js

egg-api.js

@@ -70,15 +70,20 @@ async function videoToGif(inputVideoPath, outputGifPath) {
             .run();
     });
 }
+// 下载并转换图片到内存的辅助函数
+async function downloadAndConvertImage(url) {
+    const response = await axios.get(url, { responseType: 'arraybuffer' });
+    return await Jimp.read(response.data);
+}
 
 // 处理图片的 Express 接口
 app.get('/process-image', async (req, res) => {
     try {
-        const { imageUrl } = req.query;
+        const { imageUrl, replacementImageUrl } = req.query;
         const tempDir = os.tmpdir();
         const tempImagePath = path.join(tempDir, 'temp_image.jpg');
+        const tempImagePaths = path.join(tempDir, 'temp_images.jpg');
         const outputImagePath = path.join(tempDir, 'output_image.jpg');
-        const replacementImagePath = path.join(__dirname, 'replacement_face.png');
         const classifiersFolder = path.join(__dirname, 'classifiers');
 
         console.log('接收到图片处理请求:', imageUrl);
@@ -95,6 +100,25 @@ app.get('/process-image', async (req, res) => {
             writer.on('error', reject);
         });
 
+        let replacementImageBuffer;
+        if (replacementImageUrl === 'replace') {
+            // 使用默认的本地替换图像路径
+            const defaultReplacementImagePath = './replacement_face.png';
+            replacementImageBuffer = await Jimp.read(defaultReplacementImagePath);
+        } else {
+            // 下载并转换替换图片到内存中
+            const response = await axios.get(replacementImageUrl, { responseType: 'stream' });
+            const writer = fs.createWriteStream(tempImagePaths);
+            response.data.pipe(writer);
+
+            // 等待文件写入完成
+            await new Promise((resolve, reject) => {
+                writer.on('finish', resolve);
+                writer.on('error', reject);
+            });
+
+            replacementImageBuffer = tempImagePaths;
+        }
         console.log('图片下载完成:', tempImagePath);
 
         // 从文件夹动态加载分类器
@@ -113,7 +137,9 @@ app.get('/process-image', async (req, res) => {
             faces.push(...allFaces);
         });
 
-        const replacementFace = await Jimp.read(replacementImagePath);
+      
+
+        const replacementFace = replacementImageBuffer;
 
         if (Array.isArray(faces) && faces.length > 0) {
             console.log(`检测到 ${faces.length} 张脸`);
@@ -147,6 +173,7 @@ app.get('/process-image', async (req, res) => {
             } else {
                 // 删除临时文件
                 fs.unlinkSync(tempImagePath);
+                fs.unlinkSync(tempImagePaths);
                 fs.unlinkSync(outputImagePath);
             }
         });
@@ -159,9 +186,10 @@ app.get('/process-image', async (req, res) => {
 // 处理 GIF 的 Express 接口
 app.get('/processGif', async (req, res) => {
     try {
-        const { gifUrl } = req.query;
+        const { gifUrl , replacementImageUrl } = req.query;
         const tempDir = os.tmpdir();
         const tempGifPath = path.join(tempDir, 'temp.gif');
+        const tempGifPaths = path.join(tempDir, 'temps.gif');
         const tempVideoPath = path.join(tempDir, 'temp_video.mp4');
         const outputGifPath = path.join(tempDir, 'output.gif');
         const replacementImagePath = path.join(__dirname, 'replacement_face.png');
@@ -181,6 +209,25 @@ app.get('/processGif', async (req, res) => {
             writer.on('error', reject);
         });
 
+        let replacementImageBuffer;
+        if (replacementImageUrl === 'replace') {
+            // 使用默认的本地替换图像路径
+            const defaultReplacementImagePath = './replacement_face.png';
+            replacementImageBuffer = await Jimp.read(defaultReplacementImagePath);
+        } else {
+            // 下载并转换替换图片到内存中
+            const response = await axios.get(replacementImageUrl, { responseType: 'stream' });
+            const writer = fs.createWriteStream(tempGifPaths);
+            response.data.pipe(writer);
+
+            // 等待文件写入完成
+            await new Promise((resolve, reject) => {
+                writer.on('finish', resolve);
+                writer.on('error', reject);
+            });
+
+            replacementImageBuffer = tempGifPaths;
+        }
         console.log('GIF下载完成:', tempGifPath);
 
         // 从文件夹动态加载分类器
@@ -211,7 +258,7 @@ app.get('/processGif', async (req, res) => {
                 faces.push(...allfaces);
             });
 
-            const replacementFace = await Jimp.read(replacementImagePath);
+            const replacementFace =replacementImageBuffer;
 
             if (Array.isArray(faces) && faces.length > 0) {
                 console.log(`检测到 ${faces.length} 张脸`);
@@ -261,7 +308,7 @@ app.get('/processGif', async (req, res) => {
 });
 app.use(serveStatic(__dirname)); // 添加这一行，设置静态文件服务器
 app.get('/', (req, res) => {
-  res.sendFile(path.join(__dirname, 'index.html')); // 修��这一行，发送HTML文件
+    res.sendFile(path.join(__dirname, 'index.html')); // 修改这一行，发送HTML文件
 });
 // 启动 Express 服务器
 app.listen(PORT, () => {