Merge pull request #86 from regulaforensics/fix_identification_module

Fix identification module
regulaforensics · Dec 12, 2023 · 616727a · 616727a
2 parents 4a0bea9 + 6674fe6
commit 616727a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/regula/facesdk/webclient/ext/api/person_api.py b/regula/facesdk/webclient/ext/api/person_api.py
@@ -16,10 +16,12 @@ def __init__(self, api_client: ApiClient):
 
     def add_image_to_person(self, person_id: str, content: bytes, content_type: str = None, **kwargs) -> Image:
         base_image = base64.b64encode(content).decode("UTF-8")
-        image = ImageFields(ImageFieldsImage(content_type, base_image))
+        image = ImageFields(image=ImageFieldsImage(content_type, base_image))
         return super().add_image_to_person(person_id, image, **kwargs)
 
     def create_person(self, person_fields: PersonFields, **kwargs) -> Person:
+        if person_fields.metadata is None:
+            person_fields.metadata = {}
         return super().create_person(person_fields, **kwargs)
 
     def delete_image_of_person(self, image_id: str, person_id: str, **kwargs) -> None:

diff --git a/regula/facesdk/webclient/ext/api/search_api.py b/regula/facesdk/webclient/ext/api/search_api.py
@@ -13,5 +13,5 @@ def __init__(self, api_client: ApiClient):
     def search(self, search_request: SearchRequest, **kwargs) -> SearchResult:
         image = search_request.image
         if image is not None:
-            search_request.image.content = base64.b64encode(image.content).decode("utf-8")
+            search_request.image.content = base64.b64encode(image.content).decode("utf-8") if image.content else None
         return super().search(search_request, **kwargs)