label end_susan_event:
    call sus_chibi("hide")
    hide susan_main
    with d3
    pause.5

    call update_susan

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

    call music_block
    jump main_room_menu

label update_susan:
    # Chibi Update
    $ susan_chibi.update()
    $ susan_chibi.position(flip=False)
    $ susan.xzoom = 1
    hide screen susan_cloth_pile

    return

define character.susan_say = Character("name_susan_genie", show_icon="susan", dynamic=True)

init python:
    def sus(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 Susan's face with provided attributes."""
            if any(face_attrs.values()):
                susan.set_face(**face_attrs)
                return True
            return False

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

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

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

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

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

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

        # If no visual changes, just make Susan speak
        if not any((mouth, eyes, eyebrows, pupils, cheeks, tears, emote, xpos, ypos, flip, trans, animation)):
            character.susan_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:
            susan.set_emote(emote)
            redraw = True

        if animation:
            susan.animation = animation
            redraw = True

        if flip is not None:
            susan.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 = susan.pos[1] == sprite_pos.get("y", {}).get("head")
        show_susan(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 susan.xzoom == 1 else sprite_pos.get("x", {}).get("left_corner", 0) + 25
                ypos = susan.pos[1]
                side_doll = At(susan.image, doll_transform((xpos, ypos), susan.zoom, susan.xzoom))

            character.susan_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)