from machine import Pin
from time import sleep

# LEDs on GP1 to GP5
led1 = Pin(1, Pin.OUT)
led2 = Pin(2, Pin.OUT)
led3 = Pin(3, Pin.OUT)
led4 = Pin(4, Pin.OUT)
led5 = Pin(5, Pin.OUT)

while True:
    # Ask user for speed choice
    user = input("Choose speed (1=Slow, 2=Medium, 3=Fast): ")

    # check if input is valid
    if user == "1":
        delay = 1.0
    elif user == "2":
        delay = 0.5
    elif user == "3":
        delay = 0.2
    else:
        print("Invalid input. Please enter 1, 2, or 3.")

    print("Speed selected. Wave is running...")

    # run wave forever
    while True:
        # forward wave
        led1.on()
        sleep(delay)
        led1.off()

        led2.on()
        sleep(delay)
        led2.off()

        led3.on()
        sleep(delay)
        led3.off()

        led4.on()
        sleep(delay)
        led4.off()

        led5.on()
        sleep(delay)
        led5.off()

        # backward wave
        led4.on()
        sleep(delay)
        led4.off()

        led3.on()
        sleep(delay)
        led3.off()

        led2.on()
        sleep(delay)
        led2.off()

        led1.on()
        sleep(delay)
        led1.off()