Update egg-api.js

egg-api.js

@@ -9,11 +9,14 @@ import sharp from 'sharp';
 const app = express();
 const port = 7860;
 const outputDir = './output';
+const classifiersFolder = './classifiers';
+const replacementImagePath = './replacement_face.png';
 
 // 确保输出目录存在
 if (!fs.existsSync(outputDir)){
     fs.mkdirSync(outputDir);
 }
+
 // 从文件夹加载所有分类器
 async function loadClassifiersFromFolder(folderPath) {
   const classifiers = [];
@@ -26,57 +29,22 @@ async function loadClassifiersFromFolder(folderPath) {
   return classifiers;
 }
 
-
+// 下载图片并转换格式
 async function downloadAndConvertImage(url, dest) {
-  try {
-    const response = await axios({
-      method: 'get',
-      url: url,
-      responseType: 'arraybuffer'
-    });
-
-    await sharp(response.data)
-      .toFormat('jpeg')
-      .toFile(dest);
-
-    console.log('Image downloaded and converted successfully.');
-  } catch (error) {
-    console.error('Error downloading or converting image:', error);
-    throw error;
-  }
+  const response = await axios.get(url, { responseType: 'arraybuffer' });
+  await sharp(response.data).toFormat('jpeg').toFile(dest);
 }
 
 // 图像处理函数
-async function processImage(inputImagePath, outputImagePath, classifiersFolder, replacementImagePath) {
-  try {
-    const classifiers = await loadClassifiersFromFolder(classifiersFolder);
-    
-    if (!fs.existsSync(inputImagePath)) {
-      throw new Error(`Input image not found at ${inputImagePath}`);
-    }
-
-    // 使用 Jimp 验证图片文件是否有效
-    const jimpImage = await Jimp.read(inputImagePath);
-    if (!jimpImage) {
-      throw new Error('Failed to read the image with Jimp. The image file might be corrupted or not supported.');
-    }
-
-    const img = cv.imread(inputImagePath);
-    if (img.empty) {
-      throw new Error('Failed to read the image with OpenCV. The image file might be corrupted or not supported.');
-    }
-
-    const grayImg = img.bgrToGray();
-    const allFaces = [];
-    classifiers.forEach(classifier => {
-      const faces = classifier.detectMultiScale(grayImg).objects;
-      allFaces.push(...faces);
-    });
-    
-
-    
-    const replacementFace = await Jimp.read(replacementImagePath);
-    allFaces.forEach((faceRect, i) => {
+async function processImage(inputImagePath, outputImagePath) {
+  const classifiers = await loadClassifiersFromFolder(classifiersFolder);
+  const img = cv.imread(inputImagePath);
+  const grayImg = img.bgrToGray();
+  const replacementFace = await Jimp.read(replacementImagePath);
+
+  classifiers.forEach(classifier => {
+    const faces = classifier.detectMultiScale(grayImg).objects;
+    faces.forEach(faceRect => {
       const resizedReplacementFace = replacementFace.resize(faceRect.width, faceRect.height);
       const faceRegion = img.getRegion(faceRect);
       const replacementBuffer = resizedReplacementFace.bitmap.data;
@@ -90,35 +58,27 @@ async function processImage(inputImagePath, outputImagePath, classifiersFolder,
         }
       }
     });
-    cv.imwrite(outputImagePath, img);
-  } catch (error) {
-    console.error('Error during image processing:', error);
-    throw error;
-  }
+  });
+
+  cv.imwrite(outputImagePath, img);
 }
 
 // Express接口
 app.get('/process-image', async (req, res) => {
-  const { imageUrl } = req.query;
-
-  if (!imageUrl) {
-    return res.status(400).send('Image URL is required');
-  }
+  try {
+    const { imageUrl } = req.query;
+    if (!imageUrl) {
+      return res.status(400).send('Image URL is required');
+    }
 
-  const inputImagePath = path.join(outputDir, 'test.jpg'); // 使用jpg格式保存
-  const outputImagePath = path.join(outputDir, 'image_with_replaced_faces.jpg');
-  const replacementImagePath = './replacement_face.png';
-  const classifiersFolder = './classifiers';
+    const inputImagePath = path.join(outputDir, 'test.jpg'); // 使用jpg格式保存
+    const outputImagePath = path.join(outputDir, 'image_with_replaced_faces.jpg');
 
-  try {
     // 下载并转换输入图片
     await downloadAndConvertImage(imageUrl, inputImagePath);
-    // 检查下载的图片文件是否存在且有效
-    if (!fs.existsSync(inputImagePath) || fs.statSync(inputImagePath).size === 0) {
-      throw new Error('Downloaded image file is not found or is empty.');
-    }
+
     // 处理图片
-    await processImage(inputImagePath, outputImagePath, classifiersFolder, replacementImagePath);
+    await processImage(inputImagePath, outputImagePath);
 
     // 返回处理后的图片链接
     res.sendFile(outputImagePath, { root: '.' });
@@ -130,4 +90,4 @@ app.get('/process-image', async (req, res) => {
 
 app.listen(port, () => {
   console.log(`Server is running on http://localhost:${port}`);
-});
+});