from PIL import Image

def change_palette(image_path, output_path, new_colors):
    """
    Change the palette of a PNG image to a new set of colors.

    :param image_path: Path to the input PNG image.
    :param output_path: Path to save the modified PNG image.
    :param new_colors: List of RGB tuples representing the new colors for the palette.
    """
    # Open the image
    img = Image.open(image_path)
    
    # Ensure the image is in palette mode
    if img.mode != 'P':
        raise ValueError("Image is not in palette mode")

    # Flatten the new_colors list
    new_palette = [value for color in new_colors for value in color]

    # The palette should be 768 values (256 RGB tuples)
    # If new_palette is shorter, pad it with zeros
    if len(new_palette) < 768:
        new_palette.extend([0] * (768 - len(new_palette)))
    elif len(new_palette) > 768:
        raise ValueError("Too many colors provided. The palette can have a maximum of 256 colors.")

    # Apply the new palette
    img.putpalette(new_palette)
    
    # Save the modified image
    img.save(output_path)

# Example usage:
image_path = 'pals.png'  # Path to your input PNG image
output_path = 'output.png'  # Path to save the modified PNG image

# Define a new palette (example: a few basic colors)
new_colors = [
    (255, 0, 0),   # Red
    (0, 255, 0),   # Green
    (0, 0, 255),   # Blue
    (255, 255, 0), # Yellow
    (255, 0, 255), # Magenta
    (0, 255, 255), # Cyan
    # Add more colors as needed...
]

change_palette(image_path, output_path, new_colors)