from machine import Pin
import time #full module was imported and not just sleep_ms
#from time import sleep_ms

# Note: On some ESP32 boards, GPIO 2 might have an onboard LED
switch_pin_num = 2
led_pin_Green_num = 17
led_pin_Red1_num = 16
led_pin_Red2_num = 4

# Set up pins
switch_pin = Pin(switch_pin_num, Pin.IN) # Pin as input
led_pin_Green = Pin(led_pin_Green_num, Pin.OUT) # Pin as output
led_pin_Red1 = Pin(led_pin_Red1_num, Pin.OUT)
led_pin_Red2 = Pin(led_pin_Red2_num, Pin.OUT)

def loop():
    # Read the state of the switch pin
    switch_state = switch_pin.value()

    if switch_state == 0: # LOW
        led_pin_Green.value(1) # HIGH
        led_pin_Red1.value(0) # LOW
        led_pin_Red2.value(0)
    else: # HIGH
        led_pin_Green.value(0)
        led_pin_Red1.value(0)
        led_pin_Red2.value(1)

        #sleep_ms(250)
        time.sleep_ms(250) # Delay in milliseconds 

        led_pin_Red1.value(1)
        led_pin_Red2.value(0)

        time.sleep_ms(250)
        #sleep_ms(250)

# Main program loop
while True:
    loop()