from machine import Pin
from time import sleep

Gre = Pin(19, Pin.OUT) 
Yel = Pin(18, Pin.OUT)  
Red = Pin(5, Pin.OUT)   
PGre = Pin(12, Pin.OUT) 
PRed = Pin(14, Pin.OUT) 

def mostrar_status(modo_manutencao):
    if modo_manutencao:
        print("Modo de Manutenção Ativo")
        if Gre.value():
            print("Veículo: VERDE")
        elif Yel.value():
            print("Veículo: AMARELO")
        elif Red.value():
            print("Veículo: VERMELHO")
        print("Pedestre: DESLIGADO")
    else:
        if Gre.value():
            print("Veículo: VERDE")
        elif Yel.value():
            print("Veículo: AMARELO")
        elif Red.value():
            print("Veículo: VERMELHO")
        if PGre.value():
            print("Pedestre: VERDE")
        else:
            print("Pedestre: PARE")

def configurar_modo():
    while True:
        escolha = input("Escolha o modo de operação (1 para Normal, 2 para Manutenção): ")
        if escolha == '1':
            return False
        elif escolha == '2':
            return True 
        else:
            print("Entrada inválida. Por favor, escolha 1 para Normal ou 2 para Manutenção.")


modo_manutencao = configurar_modo()

while True:
    if modo_manutencao:
        Gre.value(1)
        Yel.value(0)
        Red.value(0)
        PGre.value(0)
        PRed.value(0)
        mostrar_status(modo_manutencao)
        sleep(1)
        
        Gre.value(0)
        Yel.value(1)
        Red.value(0)
        mostrar_status(modo_manutencao)
        sleep(1)
        
        Yel.value(0)
        Red.value(1)
        mostrar_status(modo_manutencao)
        sleep(1)
        
        Red.value(0)
        mostrar_status(modo_manutencao)
        sleep(1)

    else:
        Gre.value(1)
        Yel.value(0)
        Red.value(0)
        PGre.value(0)
        PRed.value(1)
        mostrar_status(modo_manutencao)
        sleep(2)
        
        Gre.value(0)
        Yel.value(1)
        PGre.value(0)
        PRed.value(1)
        mostrar_status(modo_manutencao)
        sleep(1)
        
        Yel.value(0)
        Red.value(1)
        PGre.value(1)
        PRed.value(0)
        mostrar_status(modo_manutencao)
        sleep(2)

        Red.value(0)
        PGre.value(1)
        PRed.value(0)
        mostrar_status(modo_manutencao)
        sleep(1)