import cv2
import os
import json
import base64
from PIL import Image
import io

if __name__ == '__main__':
    with open('request_json.json', 'r') as f:
        request_data = json.load(f)
    request_json = request_data
    img_base64 = request_json['init_images'][0]
    mask_base64 = request_json['mask']
    image = Image.open(io.BytesIO(base64.b64decode(img_base64)))
    image.save('image.png')
    mask = Image.open(io.BytesIO(base64.b64decode(mask_base64)))
    mask.save('mask.png')