from __future__ import annotations

from . import response_helper

def handle_response(api_result, schema=None, loaded_client=None, response_url=None):
    generated_images = getattr(api_result, "generated_images", None)
    if not generated_images:
        return [None, None]

    image = getattr(generated_images[0], "image", None)
    image_bytes = getattr(image, "image_bytes", None)
    return [response_helper.bytes_to_tensor(image_bytes), image_bytes]