|
import streamlit as st |
|
from repo_directory import Heat_Map |
|
from repo_directory import button |
|
|
|
|
|
URL = st.text_input(label="Race Result URL from Formula1.com", value="https://www.formula1.com/en/results.html/2022/races/1136/mexico/race-result.html") |
|
|
|
race_results_df, drivers = Heat_Map.get_results(URL) |
|
|
|
Heat_Map.get_latest_ergast() |
|
|
|
drivers_df, laps_df, race_names_df, results_df = Heat_Map.load_data() |
|
|
|
|
|
|
|
available_years = race_names_df.year.unique().tolist() |
|
available_years.sort(reverse=True) |
|
|
|
YEAR_SELECTED = st.selectbox( |
|
'Select year', |
|
available_years) |
|
|
|
|
|
available_races = race_names_df[race_names_df.year == YEAR_SELECTED].name.tolist() |
|
|
|
RACE_SELECTED = st.selectbox( |
|
'Select Race', |
|
available_races) |
|
|
|
SELECTED_RACEID = race_names_df[ |
|
(race_names_df.year == YEAR_SELECTED) & (race_names_df.name == RACE_SELECTED) |
|
].raceId.values[0] |
|
|
|
Heat_Map.plot(RACE_SELECTED, SELECTED_RACEID, laps_df, drivers_df, drivers) |