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

# Buzzer on GP13
buzzer = PWM(Pin(17))

# LCD pin setup
lcd_rs = Pin(14, Pin.OUT)
lcd_e = Pin(13, Pin.OUT)
lcd_d4 = Pin(12, Pin.OUT)
lcd_d5 = Pin(11, Pin.OUT)
lcd_d6 = Pin(10, Pin.OUT)
lcd_d7 = Pin(9, Pin.OUT)

# Initialize LCD
lcd.init(lcd_rs, lcd_e, lcd_d4, lcd_d5, lcd_d6, lcd_d7)
lcd.clear()
while True:
    # Break Reminder (Short Beep)
    lcd.clear()
    lcd.setCursor(0, 0)
    lcd.print("Take a Break!")
    buzzer.freq(1000)
    buzzer.duty_u16(32768)
    sleep(0.5)
    buzzer.duty_u16(0)
    sleep(3)

    # Back to Study Reminder (Long Beep)
    lcd.clear()
    lcd.setCursor(0, 0)
    lcd.print("Back to Study!")
    buzzer.freq(1000)
    buzzer.duty_u16(32768)
    sleep(1)
    buzzer.duty_u16(0)
    sleep(2)