ws = websocket.WebSocket()
ws.connect("ws://{}/ws?clientId={}".format(server_address, client_id))
images = get_images(ws, prompt)
for node_id in images:
    for idx, image_data in enumerate(images[node_id]):
        image = Image.open(io.BytesIO(image_data))
        image_path = os.path.join(info['expo_raw_sd_dir'], f"{node_id}_{idx}.png")
        image.save(image_path)