#include <stdio.h>
#include <pico/stdlib.h>

// Define GPIO pins 
#define LED_PIN_VD 13
#define LED_PIN_VM 12
#define LED_PIN_AZ 11
#define BUTTON_PIN_A 5
#define BUTTON_PIN_B 6
int main() {
  // Initialize LED as output
  gpio_init(LED_PIN_VM);
  gpio_set_dir(LED_PIN_VM, GPIO_OUT);
  
  gpio_init(LED_PIN_VD);
  gpio_set_dir(LED_PIN_VD, GPIO_OUT);
  
  gpio_init(LED_PIN_AZ);
  gpio_set_dir(LED_PIN_AZ, GPIO_OUT);

  // Initialize button as input with pull-down resistor
  gpio_init(BUTTON_PIN_A);
  gpio_set_dir(BUTTON_PIN_A, GPIO_IN);
  gpio_pull_down(BUTTON_PIN_A); // Activate pull-down resistor

  gpio_init(BUTTON_PIN_B);
  gpio_set_dir(BUTTON_PIN_B, GPIO_IN);
  gpio_pull_down(BUTTON_PIN_B); // Activate pull-down resistor

  while (true) {
    // Check button state (active low with pull-down)
    if (gpio_get(BUTTON_PIN_A)) { 
      // Button pressed, turn on LED
      gpio_put(LED_PIN_AZ, 1);
      gpio_put(LED_PIN_VM, 0);
      gpio_put(LED_PIN_VD, 0);
      sleep_ms(100); // Delay of 100 milliseconds
    } else    if (gpio_get(BUTTON_PIN_B)) { 
      // Button pressed, turn on LED
      gpio_put(LED_PIN_AZ, 0);
      gpio_put(LED_PIN_VM, 1);
      gpio_put(LED_PIN_VD, 0);
      sleep_ms(100); // Delay of 100 milliseconds
      }
      else{
      gpio_put(LED_PIN_AZ, 0);
      gpio_put(LED_PIN_VM, 0);
      gpio_put(LED_PIN_VD, 0);
      }
  }
}