Update helper_fns.py

helper_fns.py

@@ -1,54 +1,36 @@
-import io
-import cv2
-import time
-from PIL import Image
-
-
-# Function to convert PIL image to binary
-def pil_to_binary(img):
-    with io.BytesIO() as output:
-        img.save(output, format="PNG")
-        return output.getvalue()
-
-
-def binary_to_pil(binary):
-    return Image.open(io.BytesIO(binary))
-
-
-def crop_image(image_array, x, y, w, h):
-    return image_array[y: y + h, x: x + w]
-
-
-def extract_faces(image_one, image_two, results):
-    image_one_results = results['facial_areas']['img1']
-    cropped_image_one = crop_image(image_one, 
-                                    image_one_results['x'],
-                                    image_one_results['y'],
-                                    image_one_results['w'],
-                                    image_one_results['h'],)
-    
-    image_two_results = results['facial_areas']['img2']
-    cropped_image_two = crop_image(image_two, 
-                                    image_two_results['x'],
-                                    image_two_results['y'],
-                                    image_two_results['w'],
-                                    image_two_results['h'],)
-    
-    return Image.fromarray(cropped_image_one).resize((480, 480)), Image.fromarray(cropped_image_two).resize((480, 480))
-
-def read_image_from_webcam():
-    #initialize webcam
-    cap = cv2.VideoCapture(0)
-
-    if not cap.isOpened():
-        print(f'Error could not open webcam')
-        return None
-    else:
-        print(f'Webcam opened successfully!')
-        #wait for 5 seconds
-        time.sleep(5)
-        _, frame = cap.read()
-        #Image.fromarray
-        rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        cap.release()
-        return  rgb_frame
+import io
+import time
+from PIL import Image
+
+
+# Function to convert PIL image to binary
+def pil_to_binary(img):
+    with io.BytesIO() as output:
+        img.save(output, format="PNG")
+        return output.getvalue()
+
+
+def binary_to_pil(binary):
+    return Image.open(io.BytesIO(binary))
+
+
+def crop_image(image_array, x, y, w, h):
+    return image_array[y: y + h, x: x + w]
+
+
+def extract_faces(image_one, image_two, results):
+    image_one_results = results['facial_areas']['img1']
+    cropped_image_one = crop_image(image_one, 
+                                    image_one_results['x'],
+                                    image_one_results['y'],
+                                    image_one_results['w'],
+                                    image_one_results['h'],)
+    
+    image_two_results = results['facial_areas']['img2']
+    cropped_image_two = crop_image(image_two, 
+                                    image_two_results['x'],
+                                    image_two_results['y'],
+                                    image_two_results['w'],
+                                    image_two_results['h'],)
+    
+    return Image.fromarray(cropped_image_one).resize((480, 480)), Image.fromarray(cropped_image_two).resize((480, 480))