#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

#define BUTTON_PIN PA0
#define LED_PIN PA5

int count = 0;

void setup() {
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  pinMode(LED_PIN, OUTPUT);

  // OLED init
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    for (;;); // Don't proceed, loop forever
  }
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("Okeme Ready!");
  display.display();
  delay(1000);
}

void loop() {
  if (digitalRead(BUTTON_PIN) == LOW) {  // Button pressed
    count++;
    digitalWrite(LED_PIN, HIGH);

    display.clearDisplay();
    display.setCursor(0, 0);
    display.print("Count: ");
    display.println(count);
    display.display();

    delay(300); // Debounce
  } else {
    digitalWrite(LED_PIN, LOW);
  }
}