# Part of ImGui Bundle - MIT License - Copyright (c) 2022-2026 Pascal Thomet - https://github.com/pthom/imgui_bundle
# mypy: disable_error_code=no-untyped-call
from typing import Any, Callable, Tuple
from imgui_bundle import immapp, hello_imgui


GuiFunction = Callable[[], None]
ScreenSize = Tuple[int, int]


def _make_gui_with_light_theme(gui_function: GuiFunction) -> GuiFunction:
    @immapp.static(was_theme_set=False)
    def inner() -> None:
        static = inner
        if not static.was_theme_set:
            hello_imgui.apply_theme(hello_imgui.ImGuiTheme_.white_is_white)
            static.was_theme_set = True
        gui_function()
    return inner


def _run_app_function_and_display_image_in_notebook(
        app_function: GuiFunction,  # A function that runs the app entirely
        thumbnail_height: int = 0,
        thumbnail_ratio: float = 0.0,
) -> None:
    """ImguiBundle app runner for jupyter notebook

    This runner is able to:
        * run an ImGui app from a jupyter notebook
        * display a thumbnail of the app + a "run" button in the cell output

    The GUI will be rendered with a white theme.
    If window_size is left as its default value (0, 0), then the window will autosize.

    thumbnail_height and thumbnail_ratio control the size of the screenshot that is displayed after execution.
    """
    # pip install opencv-python or pip install opencv-contrib-python
    import cv2
    import PIL.Image  # pip install pillow
    from IPython.display import display

    def make_thumbnail(image: Any) -> Any:
        resize_ratio = 1.0
        if thumbnail_height > 0:
            resize_ratio = thumbnail_height / image.shape[0]
        if thumbnail_ratio != 0.0:
            resize_ratio *= thumbnail_ratio

        if resize_ratio != 1:
            thumbnail_image = cv2.resize(
                image,
                (0, 0),
                fx=resize_ratio,
                fy=resize_ratio,
                interpolation=cv2.INTER_AREA,
            )
        else:
            thumbnail_image = image
        return thumbnail_image

    # def display_image(image):
    #     pil_image = PIL.Image.fromarray(image)
    #     display(pil_image)

    def display_image(image: Any) -> None:
        from IPython.display import Image

        # Convert the input image to a PIL image
        pil_image = PIL.Image.fromarray(image)

        # Save the PIL image to a bytes buffer in PNG format
        import io
        buffer = io.BytesIO()
        pil_image.save(buffer, format="JPEG")
        buffer.seek(0)

        # Create an IPython display Image object and specify PNG format
        jpeg_image = Image(data=buffer.getvalue(), format="jpeg")
        display(jpeg_image)


    def run_app_and_display_thumb() -> None:
        nonlocal thumbnail_ratio
        from imgui_bundle import hello_imgui

        app_function()
        app_image = hello_imgui.final_app_window_screenshot()

        scale = hello_imgui.final_app_window_screenshot_framebuffer_scale()
        if thumbnail_ratio == 0.0:
            thumbnail_ratio = 1.0  / scale

        thumbnail = make_thumbnail(app_image)
        display_image(thumbnail)

    run_app_and_display_thumb()