Added beautiful output image

app.py

@@ -44,49 +44,79 @@ def load_images_from_zip(zip_path):
     with zipfile.ZipFile(zip_path, 'r') as zip_file:
         for file_name in zip_file.namelist():
             with zip_file.open(file_name) as file:
-                img_bytes = np.asarray(bytearray(file.read()), dtype=np.uint8)
-                img = cv2.imdecode(img_bytes, cv2.IMREAD_COLOR)
-                images.append(img)
-    return [img for img in images if img is not None]
+                img_bytes = file.read()
+                img = cv2.imdecode(np.frombuffer(img_bytes, np.uint8), cv2.IMREAD_COLOR)
+                if img is not None:
+                    images.append(img)
+    return images
 
 def create_image(images):
-    if not images:
-        return np.zeros((1, 1, 3), dtype=np.uint8)
-    
-    max_width = max([img.shape[1] for img in images])
-    total_height = sum([img.shape[0] for img in images])
+    table_width = 800
+    row_height = 100
+    margin = 10
+    text_margin = 20
 
-    new_image = np.zeros((total_height, max_width, 3), dtype=np.uint8)
+    table_height = text_margin + margin + (row_height + margin) * len(images)
 
-    y_offset = 0
-    for img in images:
-        new_image[y_offset:y_offset+img.shape[0], :img.shape[1], :] = img
-        y_offset += img.shape[0]
+    font = cv2.FONT_HERSHEY_SIMPLEX
+    font_scale = 0.5
+    color = (255, 255, 255)
+    thickness = 2
 
-    new_image = cv2.resize(new_image, (0, 0), fx=0.5, fy=0.5)
+    table = np.zeros((table_height, table_width, 3), np.uint8)
+    id_col_width = 100
 
-    return new_image
+    id_x = 10
+    img_x = id_col_width + 10
+    y = text_margin
+
+    cv2.putText(table, "Image ID", (id_x, y), font, font_scale, color, thickness)
+    cv2.putText(table, "Face", (img_x, y), font, font_scale, color, thickness)
+
+    y += margin
+
+    for i, img in enumerate(images):
+        height, width = img.shape[:2]
+        new_width = int(width * row_height / height)
+        if img_x + new_width > table_width:
+            new_width = table_width - img_x
+        img_resized = cv2.resize(img, (new_width, row_height))
+        cv2.putText(table, str(i), (id_x, y + margin), font, font_scale, color, thickness)
+        table[y:y+row_height, img_x:img_x+new_width] = img_resized
+
+        y += row_height + margin
+
+    for col in range(table.shape[1]-1, -1, -1):
+        if not np.any(table[:, col]):
+            continue
+        else:
+            break
+
+    table_cropped = table[:, :col+1+id_x]
+    return table_cropped
 
 def check(avatars_zip, photos_zip):
-  avatars = load_images_from_zip(avatars_zip.name)
-  photos = load_images_from_zip(photos_zip.name)
-  not_found_faces = []
-  for photo in photos:
-    photo = cv2.cvtColor(photo, cv2.COLOR_RGB2BGR)
-    input_faces = find_faces(photo)
-    for input_face in input_faces:
-        avatars_checked = 0
-        for avatar in avatars:
-            avatar = cv2.cvtColor(avatar, cv2.COLOR_RGB2BGR)
-            distance = find_distance(avatar, input_face)
-            if distance > FACE_DIST_TRESH:
-                break
-            else:
-                avatars_checked+=1
-            if avatars_checked == len(avatars):
-                not_found_faces.append(input_face)
-
-  return create_image(not_found_faces)
+    avatars = [cv2.cvtColor(img, cv2.COLOR_RGB2BGR) for img in load_images_from_zip(avatars_zip.name)]
+    photos = [cv2.cvtColor(img, cv2.COLOR_RGB2BGR) for img in load_images_from_zip(photos_zip.name)]
+
+    input_avatars_faces = [find_faces(avatar) for avatar in avatars]
+    input_avatars_faces = [face for faces in input_avatars_faces for face in faces]
+
+    avatars_faces_count = len(input_avatars_faces)
+
+    not_found_faces = []
+
+    for photo in photos:
+        input_faces = find_faces(photo)
+        for input_face in input_faces:
+            for i in range(avatars_faces_count):
+                distance = find_distance(input_avatars_faces[i], input_face)
+                if distance <= FACE_DIST_TRESH:
+                    break
+                elif i + 1 == avatars_faces_count:
+                    not_found_faces.append(input_face)
+
+    return create_image(not_found_faces)
 
 title = '<h1 style="text-align:center">FootballChecker</h1>'