pages/Penalty_Points.py

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


_, drivers, n = Penalty_Points.get_data()

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

Penalty_Points.plot(DRIVERS_REMOVED)