from driver_led import APA102

# Initialisation des LEDs
num_leds = 12
led = APA102(num_led=num_leds)

def set_couleur(couleur):
    couleur = couleur.upper()
    couleurs_rgb = {
        'R': (255, 0, 0),         # Rouge
        'V': (0, 255, 0),         # Vert
        'B': (0, 0, 255),         # Bleu
        'J': (255, 255, 0),       # Jaune
        'VI': (128, 0, 128),      # Violet
        'RO': (255, 105, 180),    # Rose
        'O': (255, 165, 0),       # Orange
    }

    rgb = couleurs_rgb.get(couleur, (0, 0, 0))  # Éteindre si couleur inconnue

    for i in range(num_leds):
        led.set_pixel(i, *rgb)
    led.show()

def eteindre():
    set_couleur("")

def cleanup():
    led.cleanup()