from microbit import *
import random

def dice(n):
    display.clear()
    if n == 1:
        display.set_pixel(2,2,9)
    elif n == 2:
        display.set_pixel(0,4,9)
        display.set_pixel(4,0,9)
    elif n == 3:
        display.set_pixel(2,2,9)
        display.set_pixel(0,4,9)
        display.set_pixel(4,0,9)
    elif n == 4:
        display.set_pixel(0,4,9)
        display.set_pixel(4,0,9)
        display.set_pixel(0,0,9)
        display.set_pixel(4,4,9)
    elif n == 5:
        display.set_pixel(2,2,9)
        display.set_pixel(0,4,9)
        display.set_pixel(4,0,9)
        display.set_pixel(0,0,9)
        display.set_pixel(4,4,9)
    elif n == 6:
        display.set_pixel(0,4,9)
        display.set_pixel(4,0,9)
        display.set_pixel(0,0,9)
        display.set_pixel(4,4,9)
        display.set_pixel(0,2,9)
        display.set_pixel(4,2,9)

while True:
    if button_a.get_presses() or button_b.get_presses():
        dice(random.randint(1,6))
    sleep(100)