Update egg-api.js

egg-api.js

@@ -5,52 +5,63 @@ const ffmpeg = require('fluent-ffmpeg');
 const axios = require('axios');
 const fs = require('fs');
 const app = express();
-const PORT = 7860; // 将端口改为 7860
-
+const PORT = 7860;
 
 app.use(express.json());
-// GIF转换为视频
-async function gifToVideo(gifStream,tempVideoPath) {
 
+// GIF转换为视频
+async function gifToVideo(gifStream, tempVideoPath) {
+    console.log('开始将GIF转换为视频...');
     await new Promise((resolve, reject) => {
-        // 将内存流传递给 ffmpeg
         ffmpeg(gifStream)
             .inputFormat('gif')
-            .on('end', resolve)
-            .on('error', reject)
-            .output(tempVideoPath) // 将输出指定为内存流
+            .on('end', () => {
+                console.log('GIF成功转换为视频');
+                resolve();
+            })
+            .on('error', (err) => {
+                console.error('GIF转换为视频时出错:', err);
+                reject(err);
+            })
+            .output(tempVideoPath)
             .run();
     });
-
-
 }
 
 // 添加将输出视频转换为 GIF 的函数
 async function videoToGif(inputVideoPath, outputGifPath) {
+    console.log('开始将视频转换为GIF...');
     await new Promise((resolve, reject) => {
         ffmpeg()
             .input(inputVideoPath)
             .output(outputGifPath)
-            .on('end', resolve)
-            .on('error', reject)
-            .run()
- 
+            .on('end', () => {
+                console.log('视频成功转换为GIF');
+                resolve();
+            })
+            .on('error', (err) => {
+                console.error('视频转换为GIF时出错:', err);
+                reject(err);
+            })
+            .run();
     });
 }
 
 // 从文件夹加载所有分类器
 async function loadClassifiersFromFolder(folderPath) {
-    const classifiers = []
+    console.log(`开始从文件夹 ${folderPath} 加载分类器...`);
+    const classifiers = [];
 
-    // 读取文件夹中的所有文件
-    const files = fs.readdirSync(folderPath)
+    const files = fs.readdirSync(folderPath);
     for (const file of files) {
-        const filePath = `${folderPath}/${file}`
-        const classifier = new cv.CascadeClassifier(filePath)
-        classifiers.push(classifier)
+        const filePath = `${folderPath}/${file}`;
+        const classifier = new cv.CascadeClassifier(filePath);
+        classifiers.push(classifier);
+        console.log(`加载分类器: ${file}`);
     }
 
-    return classifiers
+    console.log('分类器加载完成');
+    return classifiers;
 }
 
 // 处理 GIF 的 Express 接口
@@ -62,23 +73,31 @@ app.get('/processGif', async (req, res) => {
         const replacementImagePath = './replacement_face.png';
         const classifiersFolder = './classifiers';
 
-        // 修改下载 GIF 到本地的部分
+        console.log('接收到GIF处理请求:', gifUrl);
+
+        // 下载 GIF 到本地
+        console.log('开始下载GIF...');
         const gifData = await axios.get(gifUrl, { responseType: 'stream' });
         const gifStream = gifData.data;
+        console.log('GIF下载完成');
+
         // 从文件夹动态加载分类器
         const classifiers = await loadClassifiersFromFolder(classifiersFolder);
 
         // 将GIF转换为视频
-      await gifToVideo(gifStream,tempVideoPath);
+        await gifToVideo(gifStream, tempVideoPath);
 
         // 读取视频文件
+        console.log('开始读取视频文件...');
         const videoCapture = new cv.VideoCapture(tempVideoPath);
         const frameWidth = videoCapture.get(cv.CAP_PROP_FRAME_WIDTH);
         const frameHeight = videoCapture.get(cv.CAP_PROP_FRAME_HEIGHT);
         const originalFps = videoCapture.get(cv.CAP_PROP_FPS);
         const videoWriter = new cv.VideoWriter('./out.mp4', cv.VideoWriter.fourcc('avc1'), originalFps, new cv.Size(frameWidth, frameHeight));
+        console.log('视频文件读取完成');
 
         // 处理每一帧
+        console.log('开始处理每一帧...');
         while (true) {
             const frame = videoCapture.read();
             if (frame.empty) break;
@@ -93,6 +112,7 @@ app.get('/processGif', async (req, res) => {
             const replacementFace = await Jimp.read(replacementImagePath);
 
             if (Array.isArray(faces) && faces.length > 0) {
+                console.log(`检测到 ${faces.length} 张脸`);
                 for (const rect of faces) {
                     const resizedSubstituteImage = await replacementFace.clone().resize(rect.width, rect.height);
                     const substituteRegion = frame.getRegion(rect);
@@ -110,7 +130,7 @@ app.get('/processGif', async (req, res) => {
                     substituteMat.copyTo(frame.getRegion(rect));
                 }
             } else {
-                console.error('Error: No faces detected or faces is not a non-empty iterable array');
+                console.error('未检测到人脸或faces数组为空');
             }
 
             videoWriter.write(frame);
@@ -119,16 +139,18 @@ app.get('/processGif', async (req, res) => {
         videoCapture.release();
         videoWriter.release();
         cv.destroyAllWindows();
+        console.log('帧处理完成');
+
         await videoToGif('./out.mp4', outputGifPath);
 
         res.set('Content-Type', 'image/gif');
         res.sendFile(outputGifPath, { root: '.' });
     } catch (error) {
-        console.error('Error in processing GIF:', error);
-        res.status(500).send('Internal Server Error');
+        console.error('处理GIF时出错:', error);
+        res.status(500).send('内部服务器错误');
     }
 });
 
 app.listen(PORT, () => {
-    console.log(`Server is running on port ${PORT}`);
-});
+    console.log(`服务器运行在端口 ${PORT}`);
+});