import penndraw as pd

button_released = False
button_held = False
counter = 0
pd.clear(pd.RED)
while not button_released:
    if pd.has_next_key_typed():
        button_held = True

    if button_held:
        counter = counter + 1
        pd.clear(pd.GREEN)
        pd.text(0.5, 0.5, f"{counter}...")
        if not pd.has_next_key_typed():
            button_released = True

    pd.advance()

pd.clear(pd.HSS_YELLOW)
pd.text(0.5, 0.5, f"Button held for {counter} frames.")
pd.run()
