pages/Penalty_Points.py

import streamlit as st
from repo_directory import Penalty_Points
from repo_directory import button

_, drivers, n = Penalty_Points.get_data()

#Remove drivers
DRIVERS_REMOVED = st.multiselect(
    'Select Drivers to remove',
    drivers)

Penalty_Points.plot(DRIVERS_REMOVED)