from machine import Pin
from time import sleep
import random

# Define LED and button pins
led_pins = [26, 27, 14, 13, 2, 4]
button_pin = Pin(23, Pin.IN, Pin.PULL_DOWN)

# Initialize LEDs
leds = [Pin(pin, Pin.OUT) for pin in led_pins]

def mode1():
    for led in leds:
        led.off()
        sleep(0.01)
        led.on()  # Turn on the LED
        sleep(0.5)
        led.off()  # Turn off the LED
        sleep(0.1)  # Slight delay before turning on the next LED

def mode2():
    for i in range(100):
        selected_led = random.choice(leds)
        selected_led.on()
        sleep(0.1)
        selected_led.off()
        sleep(0.01)

while True:
    button_val = button_pin.value()  # Read button value

    if button_val:
        userinp = int(input("Enter mode number 1 or 2: "))
        
        if userinp == 1:
            mode1()
        elif userinp == 2:
            mode2()
        else:
            print("Invalid mode number. Please enter 1 or 2.")
    sleep(0.1)  # Small delay to prevent bouncing issues