from machine import Pin
from time import sleep

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)


Led1.off()
Led2.off()
Led3.off()
Led4.off()
Led5.off()

def forwardFunction(i: int): 
    Led1.on()
    sleep(i)
    Led1.off()
    Led2.on()
    sleep(i)
    Led2.off()
    Led3.on()
    sleep(i)
    Led3.off()
    Led4.on()
    sleep(i)
    Led4.off()
    Led5.on()
    sleep(i)
    Led5.off()

def backwardFunction(i: int): 
    Led4.on()
    sleep(i)
    Led4.off()
    Led3.on()
    sleep(i)
    Led3.off()
    Led2.on()
    sleep(i)
    Led2.off()


Speed = int(input("Choose Speed (1 = Slow. 2 = Medium, 3 = Fast) :"))

if Speed != 1 and Speed != 2 and Speed != 3 :
    print("Please Select a Given Speed(1, 2, 3)")

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

while Speed == 1 :
    forwardFunction(3)
    backwardFunction(3)


while Speed == 2 :
    forwardFunction(2)
    backwardFunction(2)


while Speed == 3 :
    forwardFunction(1)
    backwardFunction(1)