from machine import Pin
from time import sleep

# Define each LED pin separately
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:
    num = int(input("How many LEDs do you want to turn ON? (0 to 5): "))
    led1.off()
    led2.off()
    led3.off()
    led4.off()
    led5.off()

    if num == 0:
        led1.off()
        led2.off()
        led3.off()
        led4.off()
        led5.off()

    elif num == 1:
        led1.on()

    elif num == 2:
        led1.on()
        led2.on()

    elif num == 3:
        led1.on()
        led2.on()
        led3.on()

    elif num == 4:
        led1.on()
        led2.on()
        led3.on()
        led4.on()

    elif num == 5:
        led1.on()
        led2.on()
        led3.on()
        led4.on()
        led5.on()

    else:
        print("Please enter a number between 0 and 5.")