import network
import urequests
import ujson
import time
from machine import Pin

wifi = network.WLAN(network.STA_IF)
wifi.active(True)
wifi.connect("Wokwi-GUEST", "")

print("Connecting WiFi...", end="")
while not wifi.isconnected():
    print(".", end="")
    time.sleep(0.1)
print("\nConnected!")

DATABASE_URL = "https://iottia-default-rtdb.firebaseio.com/task1.json"

button_hijau = Pin(17, Pin.IN, Pin.PULL_UP)
button_merah = Pin(5, Pin.IN, Pin.PULL_UP)

led_hijau = Pin(19, Pin.OUT)
led_merah = Pin(18, Pin.OUT)

hijau = False
merah = False
lh = 1
lm = 1

def write_fb():
    data = {"hijau": "ON" if hijau else "OFF", "merah": "ON" if merah else "OFF"}
    try:
        r = urequests.put(DATABASE_URL, data=ujson.dumps(data))
        print("Firebase Connected:", data)
        r.close()
    except:
        print("Firebase gagal")

while True:
    nh = button_hijau.value()
    if lh == 1 and nh == 0:
        hijau = not hijau
        led_hijau.value(hijau)
        write_fb()
    lh = nh

    nm = button_merah.value()
    if lm == 1 and nm == 0:
        merah = not merah
        led_merah.value(merah)
        write_fb()
    lm = nm

    time.sleep(0.1)