Ajustes no crop, tranpose e gitignore

.gitignore

@@ -1,4 +1,7 @@
 __pycache__
 flagged
 .env
-venv/
+venv/
+test-images/
+test.py
+temp/*

app.py

@@ -5,6 +5,7 @@ from dotenv import load_dotenv
 from image_crop import crop_faces
 import cv2
 import numpy as np
+from PIL import ImageOps
 
 load_dotenv()
 
@@ -14,10 +15,10 @@ def crop_photoshoot_images(photo_shoot_id):
 
   for file in files:
     s3_object = AwsService.get_image_from_s3(os.environ.get('AWS_S3_BUCKET'), file['Key'])
-    image = s3_object["pil"]
+    image = ImageOps.exif_transpose(s3_object["pil"])
     key = s3_object["key"]
-    cv2_img = np.array(image)
-    cv2_img = cv2_img[:, :, ::-1].copy()
+    image.save(f"temp/{key}.jpg")
+    cv2_img = cv2.imread(f"temp/{key}.jpg")
     faces = crop_faces(cv2_img)
     i = 1
     for face in faces:
@@ -26,6 +27,7 @@ def crop_photoshoot_images(photo_shoot_id):
       AwsService.send_image_to_s3(filename, os.environ.get('AWS_S3_BUCKET'), f'PhotoShoots/{photo_shoot_id}/Croppeds/{key}-{i}.jpg')
       os.remove(filename)
       i += 1
+    os.remove(f"temp/{key}.jpg")
 
   return {"message": f"{len(files)} images cropped successfully"}
 

image_crop.py

@@ -8,7 +8,7 @@ def discover_square_crop_points(points, image, step=3, maximum_step=30):
   h = points["h"]
 
   if step >= maximum_step:
-    return {"initial_x": x, "initial_y": y, "final_x": x + w, "final_y": x + h}
+    return {"initial_x": x, "initial_y": y, "final_x": x + w, "final_y": y + h}
 
   img_height = image.shape[0]
   img_width = image.shape[1]