fix multiple image return from api nodes (#7772)

comfy_api_nodes/nodes_api.py

@@ -31,15 +31,20 @@ def downscale_input(image):
     s = s.movedim(1,-1)
     return s
 
-def validate_and_cast_response (response):
+def validate_and_cast_response(response):
     # validate raw JSON response
     data = response.data
     if not data or len(data) == 0:
         raise Exception("No images returned from API endpoint")
 
-    # Get base64 image data
+    # Initialize list to store image tensors
-    image_url = data[0].url
+    image_tensors = []
-    b64_data = data[0].b64_json
+
+    # Process each image in the data array
+    for image_data in data:
+        image_url = image_data.url
+        b64_data = image_data.b64_json
+
         if not image_url and not b64_data:
             raise Exception("No image was generated in the response")
 
@@ -57,9 +62,12 @@ def validate_and_cast_response (response):
 
         # Convert to numpy array, normalize to float32 between 0 and 1
         img_array = np.array(img).astype(np.float32) / 255.0
+        img_tensor = torch.from_numpy(img_array)
 
-    # Convert to torch tensor and add batch dimension
+        # Add to list of tensors
-    return torch.from_numpy(img_array)[None,]
+        image_tensors.append(img_tensor)
+
+    return torch.stack(image_tensors, dim=0)
 
 class OpenAIDalle2(ComfyNodeABC):
     """