Update demo.py

demo.py

@@ -3,6 +3,7 @@ import requests
 import datadog_api_client
 import json
 import io
+import base64
 
 from PIL import Image
 
@@ -32,6 +33,22 @@ def face_crop(image, face_rect):
     face_image = face_image.resize((int(resized_w), int(resized_h)))
     return face_image
 
+def pil_image_to_base64(image, format="PNG"):
+    """
+    Converts a PIL.Image object to a Base64-encoded string.
+    :param image: PIL.Image object
+    :param format: Format to save the image, e.g., "PNG", "JPEG"
+    :return: Base64-encoded string
+    """
+    # Save the image to a BytesIO buffer
+    buffer = io.BytesIO()
+    image.save(buffer, format=format)
+    buffer.seek(0)  # Rewind the buffer
+
+    # Convert the buffer's contents to a Base64 string
+    base64_string = base64.b64encode(buffer.getvalue()).decode('utf-8')
+    return base64_string
+    
 def compare_face(frame1, frame2):
     url = "http://127.0.0.1:8080/compare_face"
     files = {'file1': open(frame1, 'rb'), 'file2': open(frame2, 'rb')}