Fix base64 decoding

laurynas · May 5, 2024 · a462d5e · a462d5e
1 parent bf25e70
commit a462d5e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/app.py b/app.py
@@ -99,8 +99,13 @@ def reset_meter(meter_id):
 def __request_image():
     image_data = request.get_data()
 
-    # if 'base64' in image_data.decode('utf-8'):
-        # image_data = image_data.decode('utf-8').split('base64,')[1]
-        # image_data = base64.b64decode(image_data)
+    try:
+        image_data_str = image_data.decode('utf-8')
+
+        if 'base64' in image_data_str:
+            image_data_b64 = image_data_str.split('base64,')[1]
+            image_data = base64.b64decode(image_data_b64)
+    except UnicodeDecodeError:
+        pass
 
     return Image.open(BytesIO(image_data))