from machine import Pin
import time

sensor = Pin(14, Pin.IN, Pin.PULL_DOWN)

rojo1 = Pin(2, Pin.OUT)
amarillo1 = Pin(3, Pin.OUT)
verde1 = Pin(4, Pin.OUT)

rojo2 = Pin(6, Pin.OUT)
amarillo2 = Pin(7, Pin.OUT)
verde2 = Pin(8, Pin.OUT)


def apagar_todos():
    rojo1.off()
    amarillo1.off()
    verde1.off()
    rojo2.off()
    amarillo2.off()
    verde2.off()

def parpadeo_verde(led, tiempo_total):
    ciclos = int(tiempo_total / 0.6)
    for _ in range(ciclos):
        led.off()
        time.sleep(0.3)
        led.on()
        time.sleep(0.3)
    led.off()


def estado_inicial():
    apagar_todos()
    rojo2.on()
    verde1.on()


def secuencia():
    apagar_todos()
    rojo2.on()
    verde1.on()
    time.sleep(5)

    
    parpadeo_verde(verde1, 3)

    apagar_todos()
    rojo2.on()
    amarillo1.on()
    time.sleep(3)

    apagar_todos()
    verde2.on()
    rojo1.on()
    time.sleep(5)

    parpadeo_verde(verde2, 3)

    apagar_todos()
    amarillo2.on()
    rojo1.on()
    time.sleep(3)


estado_inicial()

while True:
    if sensor.value() == 1:
        secuencia()
    else:
        estado_inicial()
    time.sleep(0.1)