# February Monthly Project
# Winter Olympics logo and medal count
from codex import *

def logo():
    display.fill(LIGHT_GRAY)
    # Draw Olympic rings
    display.draw_circle(50, 150, 35, BLUE)
    display.draw_circle(125, 150, 35, BLACK)
    display.draw_circle(200, 150, 35, RED)
    display.draw_circle(85, 185, 35, YELLOW)
    display.draw_circle(160, 185, 35, DARK_GREEN)
    # Draw text
    display.draw_text("MILANO CORTINA", x=34, y=25, color=BLACK, scale=2)
    display.draw_text("Winter Olympics", x=30, y=50, color=BLACK, scale=2)
    display.draw_text("2026", x=70, y=75, color=BLACK, scale=4)

def get_info():
    display.clear()
    display.print("Answer questions")
    display.print("in the console.")
    country = input("Which country? ")
    gold = int(input("How many gold medals? "))
    silver = int(input("How many silver medals? "))
    bronze = int(input("How many bronze medals? "))
    return country, gold, silver, bronze

def stacks():
    country, gold, silver, bronze = get_info()
    display.clear()
    display.print(country)
    display.print("Olympic medals")
    bar_chart(gold, 20, YELLOW, ORANGE)
    bar_chart(silver, 95, LIGHT_GRAY, GRAY)
    bar_chart(bronze, 170, ORANGE, RED)

def bar_chart(num, x, color1, color2):
    y = 220
    for i in range(num):
        display.fill_rect(x, y, 60, 12, color1)
        display.draw_rect(x, y, 60, 12, color2)
        y = y - 12

def intro():
    display.clear()
    display.print("Welcome to")
    display.print("Winter Olympics")
    display.print()
    display.print("A=Logo")
    display.print("B=Medals")
    display.print("U=Quit")
    display.print("D=Intro")


# === Main Program ===
intro()
while True:
    if buttons.was_pressed(BTN_A):
        logo()
    if buttons.was_pressed(BTN_B):
        stacks()
    if buttons.was_pressed(BTN_D):
        intro()
    if buttons.was_pressed(BTN_U):
        display.clear()
        break