import neopixel_pin
import ws2812
import machine
import time

neopixel_pin = 18

led_pins = [34, 26, 21, 35, 25, 19, 33, 14, 23, 32, 27, 22]
num_leds = len(led_pins) // 3

np = ws2812.WS2812(machine.Pin(neopixel_pin), num_leds)

def set_led_color(led_num, color):
    start_index = led_num * 3
    for i in range(3):
        np.set_pixel(start_index + i, color)
    np.show()

def turn_off_all():
    for i in range(num_leds):
        set_led_color(i, (0, 0, 0))

def mode_on():
    for i in range(num_leds):
        set_led_color(i, (255, 255, 255))
    np.show()

def mode_off():
    turn_off_all()
    np.show()

def mode_on1():
    turn_off_all()
    set_led_color(0, (255, 255, 255))
    np.show()

def mode_on2():
    turn_off_all()
    set_led_color(1, (255, 255, 255))
    np.show()

def mode_on3():
    turn_off_all()
    set_led_color(2, (255, 255, 255))
    np.show()

def mode_on4():
    turn_off_all()
    set_led_color(3, (255, 255, 255))
    np.show()

modes = {
    "ligar": mode_on,
    "desligar": mode_off,
    'um': mode_on1,
    'dois': mode_on2,
    'tres': mode_on3,
    'quatro': mode_on4,
}

while True:
    command = input("Digite um comando (ligar, desligar, um, dois, tres, quatro): ")

    if command in modes:
        modes[command]()  # Executar o modo correspondente
    else:
        print("Comando inválido. Tente novamente.")
    
    time.sleep(0.1)  # Pequeno atraso para evitar entradas muito rápidas