from machine import Pin, PWM
from time import sleep

# LEDs
led_rojo_auto = Pin(0, Pin.OUT)
led_amarillo_auto = Pin(1, Pin.OUT)
led_verde_auto = Pin(2, Pin.OUT)
led_rojo_peaton = Pin(27, Pin.OUT)
led_verde_peaton = Pin(26, Pin.OUT)

# Buzzer
buzzer = PWM(Pin(5))

# Botón
boton = Pin(6, Pin.IN, Pin.PULL_DOWN)

def modo_normal():
    """Semáforo vehicular normal"""
    led_verde_auto.on()
    led_amarillo_auto.off()
    led_rojo_auto.off()
    led_rojo_peaton.on()
    led_verde_peaton.off()

def cruce_peatonal():
    """Secuencia para cruce de peatones"""
    
    led_verde_auto.off()
    led_amarillo_auto.on()
    sleep(2)
    led_amarillo_auto.off()
    led_rojo_auto.on()
    
    # Cambio del semáforo peatonal
    led_rojo_peaton.off()
    led_verde_peaton.on()
    
    # Activar buzzer
    for _ in range(10):  # 10 segundos
        buzzer.freq(1000)
        buzzer.duty_u16(30000)
        sleep(0.5)
        buzzer.duty_u16(0)
        sleep(0.5)
    
    
    led_verde_peaton.off()
    led_rojo_peaton.on()
    led_rojo_auto.off()
    led_verde_auto.on()


modo_normal()

print("Sistema de semáforo inteligente iniciado")

while True:
    if boton.value() == 1:
        print("Botón presionado, iniciando cruce peatonal")
        cruce_peatonal()
    sleep(0.1)