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)
Heatmap.plot_fuel_correction(RACE_SELECTED, YEAR_SELECTED)