from machine import Pin
import neopixel
from time import sleep
import random

# Nastavitve
NUM_LED = 20
leds = neopixel.NeoPixel(Pin(4), NUM_LED)
gumb = Pin(2, Pin.IN, Pin.PULL_UP)
stop = False  # Spremenljivka za sledenje, ali je animacija ustavljena

# Funkcija za preverjanje stanja gumba
def check_button():
    global stop
    if gumb.value() == 0:  # Gumb je pritisnjen
        stop = not stop
        sleep(0.2)  # Debounce delay

# Glavna zanka
while True:
    check_button()  # Preveri stanje gumba
    if not stop:
        # Izvedi animacijo
        for i in range(NUM_LED):
            rgb = [random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)]
            leds[i] = rgb
            leds.write()
            sleep(0.1)
            leds[i] = (0, 0, 0)
        leds.fill((0, 0, 0))
        leds.write()
    else:
        sleep(0.1)  # Če je animacija ustavljena, počakaj malo