from machine import Pin
import time

# Definição dos pinos
BTN_AMARELO = 2
BTN_VERDE = 1
BTN_VERMELHO = 0

LED_VERDE = 6
LED_AMARELO = 5
LED_VERMELHO = 4

# Configuração dos LEDs (saída)
led_verde = Pin(LED_VERDE, Pin.OUT)
led_amarelo = Pin(LED_AMARELO, Pin.OUT)
led_vermelho = Pin(LED_VERMELHO, Pin.OUT)

# Configuração dos botões (entrada com pull-up)
btn_amarelo = Pin(BTN_AMARELO, Pin.IN, Pin.PULL_UP)
btn_verde = Pin(BTN_VERDE, Pin.IN, Pin.PULL_UP)
btn_vermelho = Pin(BTN_VERMELHO, Pin.IN, Pin.PULL_UP)

while True:
    amarelo = btn_amarelo.value()
    verde = btn_verde.value()
    vermelho = btn_vermelho.value()

    if amarelo == 1 and verde == 0 and vermelho == 0:
        # Botão amarelo
        led_verde.value(1)
        led_amarelo.value(1)
        led_vermelho.value(1)

    elif amarelo == 0 and verde == 1 and vermelho == 0:
        # Botão verde
        led_verde.value(1)
        led_amarelo.value(1)
        led_vermelho.value(0)

    elif amarelo == 0 and verde == 0 and vermelho == 1:
        # Botão vermelho
        led_verde.value(1)
        led_amarelo.value(0)
        led_vermelho.value(0)

    else:
        # Nenhum botão
        led_verde.value(0)
        led_amarelo.value(0)
        led_vermelho.value(0)

    time.sleep(0.02)