fixed up privacy preserve

src/modal_app.py

@@ -227,7 +227,7 @@ def change_image_objects_lab(
     image=image,
     volumes={volume_path: volume},
 )
-def apply_mosaic_with_bool_mask(image, mask, intensity: int = 20):
+def apply_mosaic_with_bool_mask(image, mask, intensity: int = 50):
     h, w = image.shape[:2]
     block_size = max(1, min(intensity, min(h, w)))
 
@@ -246,16 +246,20 @@ def apply_mosaic_with_bool_mask(image, mask, intensity: int = 20):
     image=image,
     volumes={volume_path: volume},
 )
-def preserve_privacy_test(
+def preserve_privacy(
     image_pil: Image.Image,
     prompt: str,
 ) -> Image.Image:
+    os.environ["TORCH_HOME"] = TORCH_HOME
+    os.environ["HF_HOME"] = HF_HOME
+    os.makedirs(HF_HOME, exist_ok=True)
+    os.makedirs(TORCH_HOME, exist_ok=True)
 
     langsam_results = lang_sam_segment.remote(
         image_pil=image_pil,
         prompt=prompt,
         box_threshold=0.35,
-        text_threshold=0.30,
+        text_threshold=0.40,
     )
 
     img_array = np.array(image_pil)
@@ -280,24 +284,7 @@ def preserve_privacy_test(
                 continue
             print(f"Processing mask {i + 1}/{len(result['masks'])}")
             print(f"Mask score: {mask_score}")
-            box = result["boxes"][i]
-            cv2.rectangle(
-                img_array,
-                (int(box[0]), int(box[1])),
-                (int(box[2]), int(box[3])),
-                (255, 0, 0),  # Blue color in BGR
-                2,  # Thickness of the rectangle
-            )
-            label = result["labels"][i]
-            cv2.putText(
-                img_array,
-                label,
-                (int(box[0]), int(box[1] - 10)),
-                cv2.FONT_HERSHEY_SIMPLEX,
-                0.5,  # Font size
-                (255, 0, 0),  # Blue color in BGR
-                2,  # Thickness of the text
-            )
+
             mask_bool = mask.astype(bool)
 
             img_array = apply_mosaic_with_bool_mask.remote(img_array, mask_bool)

src/tools.py

@@ -26,8 +26,17 @@ def privacy_preserve_image(
     example:
         input_prompt = ["face", "license plate"]
     """
+    func = modal.Function.from_name("ImageAlfred", "preserve_privacy")
+    output_pil = func.remote(image_pil=input_img, prompt=input_prompt)
 
-    return input_img
+    if output_pil is None:
+        raise ValueError("Received None from modal remote function.")
+    if not isinstance(output_pil, Image.Image):
+        raise TypeError(
+            f"Expected Image.Image from modal remote function, got {type(output_pil)}"
+        )
+
+    return output_pil
 
 
 def change_color_objects_hsv(