Update app.py

app.py

@@ -1,10 +1,33 @@
 import gradio as gr
 from nudenet import NudeDetector
+import cv2
+
+def blur_image(image_path, detections, parts_to_blur):
+    image = cv2.imread(image_path)
+    
+    for detection in detections:
+        label = detection['label']
+        if label in parts_to_blur:
+            x1, y1, x2, y2 = map(int, detection['box'])
+            roi = image[y1:y2, x1:x2]
+            roi = cv2.GaussianBlur(roi, (51, 51), 30)
+            image[y1:y2, x1:x2] = roi
+    
+    blurred_image_path = 'blurred_' + image_path.split('/')[-1]
+    cv2.imwrite(blurred_image_path, image)
+    return blurred_image_path
 
 def process(input_img):
     detector = NudeDetector(model_path="640m.onnx", inference_resolution=640)
-    converted = detector.censor(input_img)
-    return converted
+    detections = detector.detect(input_img)
+    print(detections)
+    parts_to_blur = [
+        'EXPOSED_GENITALIA_F', 'EXPOSED_GENITALIA_M', 
+        'EXPOSED_BREAST_F', 'EXPOSED_BUTTOCKS_F', 'EXPOSED_BUTTOCKS_M'
+    ]
+    
+    blurred_image_path = blur_image(input_img, detections, parts_to_blur)
+    return blurred_image_path
 
 iface = gr.Interface(process, gr.components.Image(type='filepath'), gr.components.Image(type="filepath"))