import streamlit as st
from repo_directory.v2 import HeatMap
from repo_directory import button
from repo_directory.v2 import utils

YEAR_SELECTED = st.selectbox(
    'Select Year',
    utils.get_years())

RACE_SELECTED = st.selectbox(
    'Select Race',
    utils.get_events(YEAR_SELECTED))

HeatMap.plot(RACE_SELECTED, YEAR_SELECTED)