Update vision.py

mxrch · Jan 15, 2024 · 7d117a3 · 7d117a3
1 parent e873469
commit 7d117a3
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/ghunt/parsers/vision.py b/ghunt/parsers/vision.py
@@ -88,7 +88,10 @@ def __init__(self):
         self.face_annotations: List[VisionFaceAnnotation] = []
 
     def _scrape(self, vision_data: Dict[str, any]):
-        for face_data in vision_data["faceAnnotations"]:
-            face_annotation = VisionFaceAnnotation()
-            face_annotation._scrape(face_data)
-            self.face_annotations.append(face_annotation)
+        if 'faceAnnotations' in vision_data:
+            for face_data in vision_data["faceAnnotations"]:
+                face_annotation = VisionFaceAnnotation()
+                face_annotation._scrape(face_data)
+                self.face_annotations.append(face_annotation)
+        else:
+            print("No face annotations found in the vision data.")