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()