from machine import Pin, PWM 
from time import sleep
import random

vermelho = Pin(0, Pin.OUT)
azul = Pin(5, Pin.OUT)
amarelo = Pin(13, Pin.OUT)
laranja = Pin(20, Pin.OUT)
roxo = Pin(16, Pin.OUT)
bot = Pin(22, Pin.IN, Pin.PULL_UP)


while True:
    if bot() == 0:
        num = random.randint(1, 5)
        nume = float(input("Insirá um número de 1-5:"))
        if nume == num:
            if nume == 1:
                vermelho.on(), azul.off(), amarelo.off(), laranja.off(), roxo.off()
                print ("Número = 1")
                sleep(5)
            elif nume == 2:
                vermelho.off(), azul.on(), amarelo.off(), laranja.off(), roxo.off()
                print ("Número = 2")
                sleep(5)
            elif nume == 3:
                vermelho.off(), azul.off(), amarelo.on(), laranja.off(), roxo.off()
                print ("Número = 3")
                sleep(5)
            elif nume == 4:
                vermelho.off(), azul.off(), amarelo.off(), laranja.on(), roxo.off()
                print ("Número = 4")
                sleep(5)
            elif nume == 5:
                vermelho.off(), azul.off(), amarelo.off(), laranja.off(), roxo.on()
                print ("Número = 5")
                sleep(5)
        else:
            print("Número digitado é diferente do número gerado. Número gerado =", num)
    else:
        print("Aperte o botão pra funcionar o sistema")
        vermelho.off(), azul.off(), amarelo.off(), laranja.off(), roxo.off()
        sleep(1)