Add option to return face locations, so we can highlight them on the image in the future

app.py

@@ -16,7 +16,7 @@ import torch
 #     return gr.Image(outputs[0])
 
 def process_image_str(image : str):
-    face_detection.createCroppedSetFromImage(image, "outputs", "imgs")
+    locations, paths = face_detection.getFaceLocationsAndFiles(image, "outputs", "imgs")
     # do AI stuff here
     
     return gr.Image(image)

face_detection.py

@@ -22,22 +22,42 @@ def getCroppedImages(image: Image.Image, cap = -1):
         outputs.append(image.crop([f[3], f[0], f[1], f[2]]))
         num += 1
         
-    return outputs
+    return face_locations, outputs
+
     
+def getFaceLocationsAndFiles(input_path: str, output_dir: str, output_name: str,):
+    """
+    input_path: local or absolute path to the input image\n
+    output_dir: folder (local or absolute path) to save the output image to\n
+    output_name: name for the new image (do not include extension)\n
+    returns: list of face locations and list of paths to the cropped face images 
+    """
+    locs, imgs = getCroppedImages(Image.open(input_path))
+    output_paths = []
+    num = 0
+    for i in imgs:
+        path = p.join(output_dir, f"{output_name}_{num}.png")
+        i.save(path)
+        output_paths.append(p.abspath(path))
+        num += 1
+        
+    return locs, output_paths
+
 
 def createCroppedSetFromImage(input_path: str, output_dir: str, output_name: str, cap = -1):
     """
-    input_path: path to the input image
-    output_dir: folder to save the output image to
-    output_name: name for the new image (do not include extension)
-    cap: set to -1 to process all faces detected, otherwise will limit faces to value
+    input_path: local or absolute path to the input image\n
+    output_dir: folder (local or absolute path) to save the output image to\n
+    output_name: name for the new image (do not include extension)\n
+    cap: set to -1 to process all faces detected, otherwise will limit faces to value\n
+    returns: list of absolute paths to the images
     """
-    imgs = getCroppedImages(Image.open(input_path))
+    imgs = getCroppedImages(Image.open(input_path))[1]
+    
     num = 0
     for i in imgs:
         path = p.join(output_dir, f"{output_name}_{num}.png")
         i.save(path)
-        print(p.abspath(path))
         num += 1