from typing import Any

from IPython.core.display_functions import display
from PIL.Image import Image
from PIL.ImageShow import UnixViewer


def show_file(self, path: str, **options: Any) -> int:
    # To prevent errors from trying to display image without any display
    return 0


UnixViewer.show_file = show_file
original_save = Image.save


def save(image, fp, format=None, **options):
    if isinstance(fp, str):
        display(image)

    original_save(image, fp, format, **options)


Image.save = save