Error calling the generated function: name 'Image' is not defined

[mythicalBOB]

Persistent error when code imports Image from Pillow library for converting tensor to image ETC.

REFERENCE:

Here's the fixed version:
Imports in code: ['from PIL import Image, ImageOps, ImageEnhance, ImageFilter', 'import numpy as np', 'import torch']
Stored script:
def generated_function(input_data_1=None, input_data_2=None):
def tensor_to_pil(tensor):
# Convert tensor to PIL Image
return Image.fromarray((tensor.squeeze(0).numpy() * 255).astype('uint8'))

def pil_to_tensor(image):
    # Convert PIL Image back to tensor
    return torch.tensor(np.array(image) / 255.0).unsqueeze(0).float()

def vibrant_color_soft_sketch(image):
    # Convert to grayscale
    gray = image.convert('L')

    # Invert image
    inverted = ImageOps.invert(gray)

    # Blur image to create the "sketch base"
    blurred = inverted.filter(ImageFilter.GaussianBlur(radius=10))

    # Merge gray and blurred images to create base sketch effect
    blended = Image.blend(gray, blurred, alpha=0.5)

    # Enhance color
    enhancer = ImageEnhance.Color(image)
    vibrant_image = enhancer.enhance(2)

    # Combine with sketch
    final_image = Image.blend(vibrant_image, blended.convert('RGB'), alpha=0.3)

    return final_image

if input_data_1 is not None and isinstance(input_data_1, torch.Tensor):
    # Convert tensor to PIL image
    pil_image = tensor_to_pil(input_data_1)

    # Apply vibrant color soft sketch effect
    sketch_image = vibrant_color_soft_sketch(pil_image)

    # Convert back to tensor
    output_tensor = pil_to_tensor(sketch_image)

    # Ensure the tensor is the same shape as the input
    return output_tensor.view(1, *input_data_1.shape[1:])

return None

Error calling the generated function: name 'Image' is not defined
Traceback (most recent call last):
File "/workspace/ComfyUI/custom_nodes/anynode/nodes/any.py", line 388, in go
result = locals_dict[function_name](any, input_data_2=any2)
File "", line 34, in generated_function_98fd3fc746254fc1bfa2f4405e37e79e
File "", line 4, in tensor_to_pil
NameError: name 'Image' is not defined
!!! Exception during processing!!! name 'Image' is not defined
Traceback (most recent call last):
File "/workspace/ComfyUI/execution.py", line 151, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
File "/workspace/ComfyUI/execution.py", line 81, in get_output_data
return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
File "/workspace/ComfyUI/execution.py", line 74, in map_node_over_list
results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
File "/workspace/ComfyUI/custom_nodes/anynode/nodes/any.py", line 395, in go
raise e
File "/workspace/ComfyUI/custom_nodes/anynode/nodes/any.py", line 388, in go
result = locals_dict[function_name](any, input_data_2=any2)
File "", line 34, in generated_function_98fd3fc746254fc1bfa2f4405e37e79e
File "", line 4, in tensor_to_pil
NameError: name 'Image' is not defined