Update apply_mask.py

apply_mask.py

@@ -33,7 +33,7 @@ def apply_mask_and_crop(input_folder, mask, output_folder):
         return
     
     print(f"Found {len(image_files)} images to process")
-    
+    i = 0
     for img_path in image_files:
         # Load image
         img = cv2.imread(img_path)
@@ -68,6 +68,11 @@ def apply_mask_and_crop(input_folder, mask, output_folder):
         
         # Save cropped image
         cv2.imwrite(output_path, cropped_img)
+        if i = 0:
+            preview_path = os.path.join(output_folder, f"preview_{filename}")
+            cv2.imwrite(preview_path, cropped_img)
+            return preview_path
+        i= i+1
         # print(f"Processed and saved: {output_path}")
     
     print(f"\nProcessing complete! Saved {len(image_files)} cropped images to {output_folder}")