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

servo = PWM(Pin(0))
servo.freq(50)


# Define the GPIO pins for the red and blue LEDs, and toggle switch
red_led_pin = Pin(18, Pin.OUT)  # GPIO pin for the red LED
blue_led_pin = Pin(19, Pin.OUT)  # GPIO pin for the blue LED
switch_pin = Pin(27, Pin.IN, Pin.PULL_UP)  # GPIO pin for the toggle switch

# Initialize variables
led_state = False
switch_state = switch_pin.value()
counter = 0

# Function to update LED state based on switch state
def update_led_state():
    global led_state
    led_state = switch_pin.value()
    red_led_pin.value(led_state)

# Main loop
while True:
    # Check the state of the toggle switch
    new_switch_state = switch_pin.value()

    # If the switch state has changed, update the LED state accordingly
    if switch_state != new_switch_state:
        update_led_state()
        servo.duty_u16(1500)
        sleep(0.2)

        # If the switch is turned on, increment the counter
        if led_state:
            counter += 1
            print("LED switched on. Count:", counter)
            servo.duty_u16(8100)
            sleep(0.2)
            # If the counter reaches 5, activate the blue LED and disable red LED switching
            if counter == 5:
                blue_led_pin.value(1)
                print("Blue LED activated!")
                red_led_pin.value(0)  # Turn off the red LED

    # Update the switch state
    switch_state = new_switch_state

    # Add a short delay to debounce the switch
    time.sleep(0.1)