label end_luna_event:

    call lun_chibi("hide")
    hide luna_main
    with d3
    pause.5

    call update_luna

    $ states.last_girl = states.active_girl
    $ states.active_girl = None
    $ states.lun.busy = True
    $ luna.wear("all")
    $ luna.set_cum(None)
    $ luna.set_face(tears=False, cheeks=False)

    call music_block
    jump main_room_menu

label update_luna:
    # Chibi Update
    $ luna_chibi.update()
    $ luna_chibi.position(flip=False)
    $ luna.xzoom = 1
    hide screen luna_cloth_pile

    return

define character.luna_say = Character("name_luna_genie", show_icon="luna", dynamic=True)

init python:
    def lun(what, mouth=None, eyes=None, eyebrows=None, pupils=None, cheeks=None, tears=None,
            emote=None, face=None, xpos=None, ypos=None, pos=None, flip=None, trans=None,
            animation=False, **kwargs):

        def update_face(face_attrs):
            """Update Luna's face with provided attributes."""
            if any(face_attrs.values()):
                luna.set_face(**face_attrs)
                return True
            return False

        def update_position(xpos, ypos):
            """Update Luna's position based on xpos and ypos."""
            if xpos is not None or ypos is not None:
                xpos = luna.pos[0] if xpos is None else sprite_pos.get("x", {}).get(xpos, xpos)
                ypos = luna.pos[1] if ypos is None else sprite_pos.get("y", {}).get(ypos, ypos)
                luna.pos = (xpos, ypos)
                return True
            return False

        def handle_temp_attributes(temp_face):
            """Handle temporary face attributes."""
            if temp_face:
                last_face = luna.get_face()
                temp_data = dict(zip(temp_face[::2], temp_face[1::2]))

                luna.set_face(**temp_data)
                return last_face
            return None

        def show_luna(redraw, head_only):
            """Show or hide Luna depending on the state."""
            if redraw and not head_only:
                luna.show(force=True)
                renpy.with_statement({"dolls": trans or d1})
            elif head_only:
                luna.hide()

        def reset_attributes(last_face):
            """Reset temporary face attributes."""
            if last_face:
                luna.set_face(**last_face)

        # Determine name prefix and suffix
        name_prefix = f'Luna {{size=-20}}"' if name_luna_genie != "Luna" else ""
        name_suffix = '"{/size}' if name_prefix else ""

        # If no visual changes, just make Luna speak
        if not any((mouth, eyes, eyebrows, pupils, cheeks, tears, emote, xpos, ypos, flip, trans, animation)):
            character.luna_say(what, who_prefix=name_prefix, who_suffix=name_suffix, **kwargs)
            return

        # Handle position update
        redraw = update_position(xpos, ypos)

        # Handle emote, animation, and flipping
        if emote:
            luna.set_emote(emote)
            redraw = True

        if animation:
            luna.animation = animation
            redraw = True

        if flip is not None:
            luna.xzoom = -1 if flip else 1
            redraw = True

        # Handle face update and temporary attributes
        face_attrs = {"mouth": mouth, "eyes": eyes, "eyebrows": eyebrows, "pupils": pupils, "cheeks": cheeks, "tears": tears}
        redraw = redraw or update_face(face_attrs)

        temp_face = renpy.game.context().temporary_attributes
        last_face = handle_temp_attributes(temp_face)
        redraw = redraw or bool(temp_face)

        # Handle display logic
        head_only = luna.pos[1] == sprite_pos.get("y", {}).get("head")
        show_luna(redraw, head_only)

        # Handle dialog and doll positioning
        if what:
            side_doll = None
            if head_only:
                xpos = sprite_pos.get("x", {}).get("right_corner", 0) - 25 if luna.xzoom == 1 else sprite_pos.get("x", {}).get("left_corner", 0) + 25
                ypos = luna.pos[1]
                side_doll = At(luna.image, doll_transform((xpos, ypos), luna.zoom, luna.xzoom))

            character.luna_say(what, who_prefix=name_prefix, who_suffix=name_suffix, show_side_doll=side_doll, **kwargs)

        # Reset face if modified temporarily
        reset_attributes(last_face)