from machine import Pin
from time import sleep

count = int(input("Enter how many LEDs you want to control (starting from GP1): "))
repeat = int(input("Enter how many times to repeat the pattern: "))
delay = float(input("Enter delay time in seconds: "))

for _ in range(repeat):
  # Forward (GP1 -> GPn)
  for i in range(1, count + 1):
    led = Pin(i, Pin.OUT)
    led.value(1)
    sleep(delay)
    led.value(0)
    sleep(delay)

  # Reverse bounce (GP(n-1) -> GP2)
  if count > 2:
    for i in range(count - 1, 1, -1):
      led = Pin(i, Pin.OUT)
      led.value(1)
      sleep(delay)
      led.value(0)
      sleep(delay)