app/pages/1_trips_inspector.py

import streamlit as st
from pipeline import load_data, plot_trip_on_map

st.set_page_config(layout="wide")

st.title("Trip inspector")

data = load_data()

routes = data["routes"]
trips = data["trips"]


routes_option = st.selectbox(
    "Select a route",
    routes["route_long_name"].tolist(),
    index=None,
    placeholder="Select a route...",
)

trips_routes = (
    trips[
        trips["route_id"].isin(
            routes[routes["route_long_name"] == routes_option]["route_id"].unique()
        )
    ]
    if routes_option
    else trips.copy()
)

trips_id_option = st.selectbox(
    "Select a Trip",
    trips_routes["trip_id"].unique().tolist(),
    index=None,
    placeholder="Select a Trip id...",
)

if trips_id_option:
    plot_trip_on_map(trips_id_option, data)