Bjornulf_custom_nodes/select_image_from_list.py

import torch

class Everything(str):
    def __ne__(self, __value: object) -> bool:
        return False

class SelectImageFromList:
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "all_images": ("IMAGE", {}),
                "selection": ("INT", {"default": 1, "min": -999999, "max": 999999, "step": 1}),  # Updated to allow negative values
            }
        }

    RETURN_TYPES = ("IMAGE",)
    RETURN_NAMES = ("selected_image",)
    FUNCTION = "select_an_image"
    CATEGORY = "Bjornulf"

    def select_an_image(self, all_images, selection):
        num_images = all_images.shape[0]
        
        # Convert selection to 0-based index
        if selection > 0:
            index = selection - 1
        else:
            # Handle negative indices directly
            index = selection
        
        # Ensure the index is within bounds
        if index >= num_images:
            index = num_images - 1
        elif index < -num_images:
            index = 0
            
        # Select the image at the specified index
        selected_image = all_images[index].unsqueeze(0)
        
        return (selected_image,)