Spaces:
Running
Running
| import streamlit as st | |
| from repo_directory import Heat_Map | |
| from repo_directory import button | |
| # select box for url | |
| 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() | |
| # select year | |
| available_years = race_names_df.year.unique().tolist() | |
| available_years.sort(reverse=True) | |
| YEAR_SELECTED = st.selectbox( | |
| 'Select year', | |
| available_years) | |
| # select race | |
| 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) |