from machine import Pin
import time

leds = [
    Pin(2, Pin.OUT),
    Pin(3, Pin.OUT),
    Pin(4, Pin.OUT),
    Pin(5, Pin.OUT)
]

button = Pin(16, Pin.IN, Pin.PULL_UP)
mode = 0

def all_off():
    for led in leds:
        led.off()

while True:
    if button.value() == 0:
        mode = (mode + 1) % 4
        time.sleep(0.3)

    if mode == 0:          # Blink all
        for led in leds: led.on()
        time.sleep(0.3)
        all_off()
        time.sleep(0.3)

    elif mode == 1:        # Running light
        for led in leds:
            led.on()
            time.sleep(0.2)
            led.off()

    elif mode == 2:        # Alternate
        leds[0].on(); leds[2].on()
        leds[1].off(); leds[3].off()
        time.sleep(0.3)
        leds[0].off(); leds[2].off()
        leds[1].on(); leds[3].on()
        time.sleep(0.3)

    elif mode == 3:        # All ON
        for led in leds: led.on()