from machine import Pin, PWM
from time import sleep_ms, ticks_ms

lcd_rs=Pin(12, Pin.OUT)
lcd_e=Pin(11, Pin.OUT)
lcd_d4=Pin(7, Pin.OUT)
lcd_d5=Pin(8, Pin.OUT)
lcd_d6=Pin(9, Pin.OUT)
lcd_d7=Pin(10, Pin.OUT)

lcd.init(lcd_rs,lcd_e,lcd_d4,lcd_d5,lcd_d6,lcd_d7)

button_1=Pin(16,Pin.IN)
snooze_button=Pin(17,Pin.IN)

count=0
buzz= PWM(Pin(18))

elasped_time=start = ticks_ms()
while True:
    if button_1.value()==1:
        count=count+1
        while count<=8:
            lcd.clear()
            lcd.SetCursor(0,0)
            lcd.print(count,"/8 Glasses")
            lcd.SetCursor(1,0)
            lcd.print("Drink Water!!!")
            buzz.freq(1000)
            buzz.duty_u16(37268)
            sleep_ms(10000)
            lcd.SetCursor(2,0)
            lcd.print("Time Passed:")
            lcd.SetCursor(1,12)
            lcd.print(ticks_ms)
        
    elif count==8:
        lcd.clear()
        lcd.SetCursor(0,0)
        lcd.print("Great Job!")
        lcd.SetCursor(0,1)
        lcd.print("Keep Going!")

    elif snooze_button.value()==1:
        lcd.SetCursor(0,0)
        lcd.print("Snoozed For 10 Sec")
        buzz.freq(0)
        sleep_ms(10000)
        while count<=8:
            lcd.SetCursor(0,0)
            lcd.print(count)
            buzz.freq(1000)
            buzz.duty_u16(37268)
            sleep_ms(10000)
            lcd.SetCursor(1,0)
            lcd.print("Time Passed:")
            lcd.SetCursor(1,12)
            lcd.print(elasped_time)