from machine import Pin

# Setup LED
led_merah = Pin(15, Pin.OUT)
led_kuning = Pin(2, Pin.OUT)
led_ijo = Pin(4, Pin.OUT)

# Fungsi matikan semua LED
def matikan_semua():
    led_kuning.off()
    led_ijo.off()
    led_merah.off()

while True:
    cmd = input("Ketik perintah (on/off kuning/ijo/merah): ").strip().lower()

    if cmd == "on kuning":
        if led_ijo.value() == 0 and led_merah.value() == 0:
            led_kuning.on()
        else:
            print("Hanya satu lampu boleh menyala!")
    
    elif cmd == "off kuning":
        led_kuning.off()

    elif cmd == "on ijo":
        if led_kuning.value() == 0 and led_merah.value() == 0:
            led_ijo.on()
        else:
            print("Hanya satu lampu boleh menyala!")

    elif cmd == "off ijo":
        led_ijo.off()

    elif cmd == "on merah":
        if led_kuning.value() == 0 and led_ijo.value() == 0:
            led_merah.on()
        else:
            print("Hanya satu lampu boleh menyala!")

    elif cmd == "off merah":
        led_merah.off()

    elif cmd == "off all":
        matikan_semua()

    else:
        print("Perintah tidak dikenali. Contoh: on kuning / off ijo / off all")