from machine import Pin, PWM
import time

ir = Pin(18, Pin.IN)
green = Pin(11, Pin.OUT)
buzzer = PWM(Pin(15))
servo = PWM(Pin(0))

servo.freq(50)
buzzer.freq(1000)

def servo_center():
  servo.duty_u16(4850)

def servo_left():
  servo.duty_u16(4000)

def servo_right():
  servo.duty_u16(5700)

def wag_tail():
  for i in range(3):
    servo_left()
    time.sleep(0.15)
    servo_right()
    time.sleep(0.15)
  
  servo_center()

def happy_sound():
  buzzer.freq(1200)
  buzzer.duty_u16(20000)
  time.sleep(0.15)

  buzzer.freq(1600)
  time.sleep(0.15)

  buzzer.duty_u16(0)

while True:
  if ir.value() == 0:
    green.value(1)
    wag_tail()
    happy_sound()
    time.sleep(1)
  else:
    green.value(0)
  time.sleep(0.01)