Update avatar.py

avatar.py

@@ -202,16 +202,25 @@ class Avatar:
     face_detect_results = []
     pady1, pady2, padx1, padx2 = [0, 10, 0, 0]
     for rect, image in zip(predictions, images):
-      if rect is None:
-        cv2.imwrite('temp_faulty_frame.jpg', image) # check this frame where the face was not detected.
-        raise ValueError('Face not detected! Ensure the video contains a face in all the frames.')
+        # Kiểm tra nếu rect là None hoặc không đủ phần tử
+        if rect is None or len(rect) < 4:
+            # Bỏ qua frame này và tiếp tục với frame tiếp theo
+            face_detect_results.append(None)
+            continue
 
-      y1 = max(0, rect[1] - pady1)
-      y2 = min(image.shape[0], rect[3] + pady2)
-      x1 = max(0, rect[0] - padx1)
-      x2 = min(image.shape[1], rect[2] + padx2)
+        try:
+            y1 = max(0, rect[1] - pady1)
+            y2 = min(image.shape[0], rect[3] + pady2)
+            x1 = max(0, rect[0] - padx1)
+            x2 = min(image.shape[1], rect[2] + padx2)
+            
+            face_detect_results.append([x1, y1, x2, y2])
+        except (IndexError, TypeError) as e:
+            print(f"Error processing face detection result: {e}")
+            face_detect_results.append(None)
+            continue
 
-      face_detect_results.append([x1, y1, x2, y2])
+    return face_detect_results
     # print("\n")
     # print("face_detect_results length = " + str(len(face_detect_results)))
     # print("face_detect_results[2]="+str(face_detect_results[2]))