26 lines
796 B
Python
26 lines
796 B
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
from . import response_helper
|
|
|
|
def handle_response(api_result, schema=None, loaded_client=None, response_url=None):
|
|
data_items = getattr(api_result, "data", None)
|
|
if not data_items:
|
|
return [None, None]
|
|
|
|
image_tensors = []
|
|
first_bytes = None
|
|
for item in data_items:
|
|
image_base64 = getattr(item, "b64_json", None)
|
|
if not image_base64:
|
|
continue
|
|
|
|
image_bytes = base64.b64decode(image_base64)
|
|
tensor = response_helper.bytes_to_tensor(image_bytes)
|
|
if tensor is not None:
|
|
image_tensors.append(tensor)
|
|
if first_bytes is None:
|
|
first_bytes = image_bytes
|
|
|
|
return [response_helper.stack_image_tensors(image_tensors), first_bytes]
|