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

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

const int buttonPin = 9;

void setup() {
  pinMode(buttonPin, INPUT_PULLUP);
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  randomSeed(analogRead(0)); // Seed for random number generation
  
}

void loop() {
  if (digitalRead(buttonPin) == LOW) {
    displayCircle(); // Flash a small circle
    delay(500);
    int diceRoll = random(1, 7); //random number between 1 and 6
    displayDice(diceRoll);
    delay(1000);
    clearDisplay();
  }
}

void displayCircle() {
  display.clearDisplay();
  display.fillCircle(SCREEN_WIDTH / 2, SCREEN_HEIGHT / 2, 10, WHITE); // Draw a white circle
  display.display();
}

void displayDice(int number) {
  display.clearDisplay();
  display.setTextSize(3);
  display.setTextColor(WHITE);
  display.setCursor(40, 20);
  display.print(number);
  display.display();
}

void clearDisplay() {
  display.clearDisplay();
  display.display();
}