from machine import Pin, PWM
from utime import sleep

# Define LEDs
red = Pin(5, Pin.OUT)
yellow = Pin(4, Pin.OUT)
green = Pin(3, Pin.OUT)

# Define Servo Motor on GPIO 2
servo = PWM(Pin(2))
servo.freq(50)  # Set PWM frequency to 50Hz



# Function to set servo angle
def set_servo_angle(angle):
    duty = int((angle / 180) * 5000 + 1000)  # Convert angle to duty cycle (1000-8000)
    servo.duty_u16(duty)

# Function to make the buzzer beep
def beep(duration):
    sleep(duration)
   
    sleep(0.2)

while True:
    # Red Light ON (Servo Closed) - Long Buzzer Beep
    red.value(1)
    yellow.value(0)
    green.value(0)
    set_servo_angle(0)  # Close the servo
    print("RED Light ON - Servo Closed")
   
    sleep(1)

    # Yellow Light ON - Short Buzzer Beep
    red.value(0)
    yellow.value(1)
    green.value(0)
    print("YELLOW Light ON")
   
    sleep(1)

    # Green Light ON (Servo Open) - No Buzzer
    red.value(0)
    yellow.value(0)
    green.value(1)
    set_servo_angle(90)  # Open the servo
    print("GREEN Light ON - Servo Open")
    sleep(3)