Update app.py

app.py

@@ -332,7 +332,48 @@ def convert_image_format(input_path, output_format="png"):
     return output_path
 
 
+def mosaic_image_with_mask(image_path, mask_path, output_path, mosaic_level=15):
+    """
+    マスク画像を使用して元画像の指定領域にモザイクをかける関数。
+    Parameters:
+    - image_path: 元画像のパス
+    - mask_path: モザイクをかけたい領域を白、その他を黒としたマスク画像のパス
+    - output_path: モザイク処理結果の出力パス
+    - mosaic_level: モザイクの強さ（値が大きいほど粗いモザイクになる）
+    Returns:
+    - output_path: モザイク処理された画像の出力パス
+    """
+    # 画像とマスクを読み込み
+    image = cv2.imread(image_path)
+    mask = cv2.imread(mask_path, cv2.IMREAD_GRAYSCALE)
+
+    # 画像とマスクの読み込みチェック
+    if image is None:
+        raise ValueError(f"元画像が見つかりません: {image_path}")
+    if mask is None:
+        raise ValueError(f"マスク画像が見つかりません: {mask_path}")
+
+    # マスク画像が元画像と同じサイズでない場合、リサイズ
+    if image.shape[:2] != mask.shape[:2]:
+        print(f"マスク画像のサイズを元画像に合わせてリサイズします: {mask.shape} -> {image.shape[:2]}")
+        mask = cv2.resize(mask, (image.shape[1], image.shape[0]))
+
+    # モザイクをかける領域を抽出
+    mosaic_area = cv2.bitwise_and(image, image, mask=mask)
+
+    # モザイク処理
+    small = cv2.resize(mosaic_area, (image.shape[1] // mosaic_level, image.shape[0] // mosaic_level), interpolation=cv2.INTER_LINEAR)
+    mosaic = cv2.resize(small, (image.shape[1], image.shape[0]), interpolation=cv2.INTER_NEAREST)
 
+    # マスクを使って元画像にモザイク部分を合成
+    mosaic_result = cv2.bitwise_and(mosaic, mosaic, mask=mask)
+    image_no_mosaic = cv2.bitwise_and(image, image, mask=cv2.bitwise_not(mask))
+    result_image = cv2.add(image_no_mosaic, mosaic_result)
+
+    # モザイク処理結果を保存
+    cv2.imwrite(output_path, result_image)
+
+    return output_path
 
 
 #この下は、openCV
@@ -498,6 +539,18 @@ async def create_mask_and_inpaint_opencv(image: UploadFile = File(...), risk_lev
     # OpenCVでインペイント
     inpaint_image_with_mask(input_path, mask_path, output_path)
 
+    return FileResponse(output_path)
+@app.post("/create-mask-and-inpaint-mosaic")
+async def create_mask_and_inpaint_opencv(image: UploadFile = File(...), risk_level: int = Form(...)):
+    point1 = (0.00000000000002, 0.00000000000002)
+    point2 = (0.00000000000001, 0.00000000000001)
+    input_path = save_image(image.file, "input.jpg")
+    mask_path =  special_process_image_yolo(risk_level, input_path, point1, point2, thresholds)
+
+    output_path = SAVE_DIR / "output_opencv.jpg"
+    # OpenCVでインペイント
+    mosaic_image_with_mask(input_path, mask_path, output_path)
+
     return FileResponse(output_path)
 @app.post("/create-mask-and-inpaint-stamp")
 async def create_mask_and_inpaint_opencv(image: UploadFile = File(...), risk_level: int = Form(...)):
@@ -736,7 +789,7 @@ async def create_mask_sum(image: UploadFile = File(...), risk_level: int = Form(
     output_path = f"./output_simple_lama_{timestamp}_{unique_id}.jpg"
 
     # OpenCVでインペイント
-    inpaint_image_with_mask(input_path, mask_path, output_path)
+    mosaic_image_with_mask(input_path, mask_path, output_path)
 
     return FileResponse(output_path)
 
@@ -831,8 +884,8 @@ async def read_root():
             <select id="processingType" class="custom-select">
                 <option value="opencv">OpenCVインペイント</option>
                 <option value="simple_lama">Simple Lamaインペイント</option>
-                <option value="stable_diffusion">Stable Diffusionインペイント</option>
-                <option value="deep_fill_v2">DeepFillv2インペイント</option>
+                <option value="stamp">stampインペイント</option>
+                <option value="mosaic">mosaicインペイント</option>
             </select>
         </div>
 
@@ -907,10 +960,10 @@ async def read_root():
                 apiEndpoint = "/create-mask-and-inpaint-opencv";
             } else if (processingType === "simple_lama") {
                 apiEndpoint = "/create-mask-and-inpaint-simple-lama";
-            } else if (processingType === "stable_diffusion") {
-                apiEndpoint = "/create-mask-and-inpaint-stable-diffusion";
-            } else if (processingType=="deep_fill_v2"){
-                apiEndpoint = "/create-mask-and-inpaint-deepfillv2";
+            } else if (processingType === "stamp") {
+                apiEndpoint = "/create-mask-and-inpaint-stamp";
+            } else if (processingType=="mosaic"){
+                apiEndpoint = "/create-mask-and-inpaint-mosaic";
             }
 
             const url = "https://rein0421-aidentify.hf.space" + apiEndpoint;