from machine import Pin
import time


led_pins = [Pin(18, Pin.OUT), Pin(19, Pin.OUT), Pin(21, Pin.OUT), Pin(22, Pin.OUT), Pin(23, Pin.OUT)]


def clear_leds():
    for led in led_pins:
        led.off()




def sequence_1(loops):
    
    for _ in range(loops):
        for led in led_pins:
            led.on()
        time.sleep(0.5) 
        clear_leds()
        time.sleep(0.5)  

    
def sequence_2(loops):
    
    for _ in range(loops):
        clear_leds()
        for i in range(5):
            led_pins[i].on()
            time.sleep(0.2)
            led_pins[i].off()

def sequence_3(loops):
    for _ in range(loops):
        clear_leds()
        for i in range(4, -1, -1):
            led_pins[i].on()
            time.sleep(0.2)
            led_pins[i].off()

def sequence_4(loops):
    for _ in range(loops):
        clear_leds()
        for i in range(0, 5, 2):
            led_pins[i].on()
            time.sleep(0.5)
            led_pins[i].off()
        time.sleep(0.5)
        clear_leds()
def sequence_5(loops):
    for _ in range(loops):
        for i in range(5):
            clear_leds()
            led_pins[i].on()
            time.sleep(0.2)
        for i in range(4, -1, -1):
            clear_leds()
            led_pins[i].on()
            time.sleep(0.2)
def sequence_6(loops):
    
    for _ in range(loops):
        
        led_pins[0].on()
        led_pins[1].on()
        time.sleep(0.3)
        led_pins[0].off()
        led_pins[1].off()
        time.sleep(0.3)

        
        led_pins[1].on()
        led_pins[2].on()
        time.sleep(0.3)
        led_pins[1].off()
        led_pins[2].off()
        time.sleep(0.3)

        
        led_pins[2].on()
        led_pins[3].on()
        time.sleep(0.3)
        led_pins[2].off()
        led_pins[3].off()
        time.sleep(0.3)

       
        led_pins[3].on()
        led_pins[4].on()
        time.sleep(0.3)
        led_pins[3].off()
        led_pins[4].off()
        time.sleep(0.3)

def sequence_7(loops):
    
    import random
    for _ in range(loops):
        clear_leds()
        for _ in range(5):
            led_index = random.randint(0, 4)
            led_pins[led_index].on()
            time.sleep(0.1)
            led_pins[led_index].off()
        time.sleep(0.5)  


# Main loop
while True:
    try:
       
        sequence_choice = int(input("Enter the sequence number(1-7): "))
        if sequence_choice not in [1, 2, 3, 4, 5,6,7]:
            print("Invalid choice, please select a number between 1 and 7")
            continue
        
        loops = int(input("Enter the number of loops: "))
        if loops <= 0:
            print("Number of loops must be a positive integer")
            continue

        if sequence_choice == 1:
            sequence_1(loops)
        elif sequence_choice == 2:
            sequence_2(loops)
        elif sequence_choice == 3:
            sequence_3(loops)
        elif sequence_choice == 4:
            sequence_4(loops)
        elif sequence_choice == 5:
            sequence_5(loops)
        elif sequence_choice == 6:
            sequence_6(loops)
        elif sequence_choice == 7:
            sequence_7(loops)
    except ValueError:
        print("Invalid input. Please enter valid numbers.")
    
    time.sleep(1)