Merge pull request #14 from wang-bam-bbang/masking

feat: apply ocr masking image feature in image upload process

package.json

@@ -37,6 +37,7 @@
     "class-validator": "^0.14.1",
     "cookie-parser": "^1.4.7",
     "express-basic-auth": "^1.2.1",
+    "form-data": "^4.0.1",
     "passport": "^0.7.0",
     "passport-http-bearer": "^1.0.1",
     "passport-oauth2": "^1.8.0",

src/comment/comment.service.ts

@@ -50,6 +50,12 @@ export class CommentService {
   }
 
   async getPostComments(postId: number): Promise<CommentResponseDto[]> {
+    const post = this.postService.getPostById(postId);
+
+    if (!post) {
+      throw new NotFoundException('Post not found');
+    }
+
     return this.commentRepository.getPostComments(postId);
   }
 

src/image/image.service.ts

@@ -13,7 +13,8 @@ import {
   NotFoundException,
 } from '@nestjs/common';
 import { ConfigService } from '@nestjs/config';
-
+import axios from 'axios';
+import FormData from 'form-data';
 import sharp from 'sharp';
 
 @Injectable()
@@ -53,7 +54,13 @@ export class ImageService {
   private async uploadImage(file: Express.Multer.File): Promise<string> {
     const key = `${Date.now()}-${Math.random().toString(36).substring(2)}${file.originalname}`;
 
-    const webpFile = await this.convertToWebp(file);
+    // OCR Masking
+    const maskedImageBuffer = await this.maskingImageWithFlask(file);
+
+    const webpFile = await this.convertToWebp({
+      ...file,
+      buffer: maskedImageBuffer,
+    });
 
     const command = new PutObjectCommand({
       Bucket: this.bucketName,
@@ -172,4 +179,40 @@ export class ImageService {
       throw new InternalServerErrorException('Failed to generate signed URLs');
     }
   }
+
+  /**
+   * AI feature 서버에 이미지를 보내서 OCR 마스킹 처리
+   * @param file Express.Multer.File
+   * @returns Buffer (마스킹된 이미지 데이터)
+   */
+  private async maskingImageWithFlask(
+    file: Express.Multer.File,
+  ): Promise<Buffer> {
+    const flaskApiUrl = this.configService.get<string>('FLASK_API_URL'); // Flask API URL
+    const flaskApiKey = this.configService.get<string>('FLASK_API_KEY'); // Flask API Key
+
+    try {
+      const formData = new FormData();
+      formData.append('image', file.buffer, file.originalname);
+
+      const response = await axios.post(
+        `${flaskApiUrl}/process_image`,
+        formData,
+        {
+          headers: {
+            'X-API-KEY': flaskApiKey,
+            ...formData.getHeaders(),
+          },
+          responseType: 'arraybuffer', // 이미지 데이터를 Buffer로 받기
+        },
+      );
+
+      return Buffer.from(response.data); // 마스킹된 이미지 데이터 반환
+    } catch (error) {
+      console.error('Error calling Flask API:', error.message);
+      throw new InternalServerErrorException(
+        'Failed to process image with Flask API',
+      );
+    }
+  }
 }