App kind of works:)

app.py

@@ -76,33 +76,67 @@ def get_buses():
 
 def plot_graph(plot_df):
     #Nu vill vi plotta!
-    categories =  {0 : 'EMPTY',
-    1: 'MANY_SEATS_AVAILABLE',
-    2:'FEW_SEATS_AVAILABLE',
-    3:'STANDING_ROOM_ONLY',
-    4:'CRUSHED_STANDING_ROOM_ONLY',
-    5: 'FULL'}
-    #Steg 2, fixa sa date ersätts av stop genom att mappa location till stop
+    categories =  {0 : 'Empty',
+    1: 'Many seats available',
+    2:'Few seats available',
+    3:'Standing room only',
+    4:'Crushed standing room',
+    5: 'Full'}
     
-    plot_df = plot_df[["datetime", "vehicle_occupancystatus"]]
+    plot_df = plot_df[["datetime", "vehicle_occupancystatus", "stop_name"]]
+    plot_df = plot_df.sort_values("datetime")
     st.write(plot_df.head())
     st.write(plot_df.tail())
     #plot_df = plot_df.set_index("datetime")
-    plot_df["Y_category"] = plot_df["vehicle_occupancystatus"].map(categories)
+    plot_df["Occupancy"] = plot_df["vehicle_occupancystatus"].map(categories)
+    # Explicitly set the order for Y_category
+    category_order = list(categories.values())  # ['Empty', 'Many seats available', ..., 'Full']
+    category_order.reverse()
+
     #st.line_chart(plot_df)
     # Create the Altair chart
-    chart = alt.Chart(plot_df).mark_line(point=True).encode(
-        x=alt.X('datetime:T', title="Datetime"),  # Use column name as string
-        y=alt.Y('Y_category:N', title="Vehicle Occupancy Status (Categories)"),  # Treat Y as categorical
-        tooltip=['datetime', 'Y_category']  # Add tooltips for interactivity
+    chart = alt.Chart(plot_df).mark_line(point=True, interpolate="step-after").encode(
+        x=alt.X('stop_name:N', title="Stop name"),  # Use column name as string
+        y=alt.Y('Occupancy:N', title="Vehicle Occupancy Status (Categories)", sort=category_order, scale=alt.Scale(domain=category_order)),  # Treat Y as categorical
+        tooltip=["datetime", 'stop_name', 'Occupancy']  # Add tooltips for interactivity
     ).properties(
         title="Vehicle Occupancy Status Over Time"
     )
     st.altair_chart(chart, use_container_width=True)
 
+def visualize(filtered_data):
+    import folium
+    from streamlit_folium import st_folium
+
+    categories =  {0 : 'Empty',
+    1: 'Many seats available',
+    2:'Few seats available',
+    3:'Standing room only',
+    4:'Crushed standing room',
+    5: 'Full'}
+
+    # Create a folium map centered around a location
+    m = folium.Map(location=[filtered_data.iloc[0]["stop_lat"], filtered_data.iloc[0]["stop_lon"]], zoom_start=12)
+
+    sw = filtered_data[['stop_lat', 'stop_lon']].min().values.tolist()
+    ne = filtered_data[['stop_lat', 'stop_lon']].max().values.tolist()
+
+    m.fit_bounds([sw, ne])
+
+    # Add bus stop markers based on filtered data
+    for idx, row in filtered_data.iterrows():
+        folium.Marker(
+            [row['stop_lat'], row['stop_lon']],
+            popup=f"Bus stop: {row['stop_name']} Bus occupancy: {categories[row['vehicle_occupancystatus']] }",
+            icon = folium.Icon(icon="bus-simple", prefix="fa")
+        ).add_to(m)
+
+    # Display the map
+    st_folium(m, width=700, height=500)
+
 # Streamlit UI
 def main():
-    st.title("Hopsworks Feature Group Explorer")
+    st.title("Wheely Fun Times - Bus Occupancy Explorer")
     
     # Initialize session state
     if "hopsworks_project" not in st.session_state:
@@ -163,6 +197,19 @@ def main():
             st.write("### Selected Bus")
             st.write(f"{search}: {bus}")
         
+        # Streamlit checkbox to toggle bus direction
+        if "direction" not in st.session_state:
+            st.session_state.direction = False
+        
+        # Streamlit button to toggle bus direction
+        if st.sidebar.button('Change Direction'):
+            # Toggle between 'North' and 'South'
+            st.session_state.direction = not st.session_state.direction
+            print(st.session_state.direction)
+
+            
+        #direction = st.sidebar.checkbox('Direction of bus', value=True) 
+
         today = datetime.now()
         tomorrow = today + timedelta(days=1)
         today = today.date()
@@ -189,6 +236,8 @@ def main():
         trip_ids = list(trips["trip_id"])
         plot_df = st.session_state.data[st.session_state.data["trip_id"]==trip_ids[0]]
 
+        #TODO direction
+
         print(f"start time {type(start_time)}")
         print(f"end time {type(end_time)}")
         print(f"day {type(day_choice)}")
@@ -196,11 +245,14 @@ def main():
         if start_time != None and end_time != None:
             #TODO hur filtrera på tid?
             st.write(f"Displaying buses between {start_time.strftime('%H:%M')} and {end_time.strftime('%H:%M')} the {day_choice}")
-            selected_trips = bus_trips[(bus_trips["datetime"] >= datetime.combine(date_options[day_choice], start_time)) & (bus_trips["datetime"] <= datetime.combine(date_options[day_choice], end_time))]
+            selected_trips = bus_trips[(bus_trips["datetime"] >= datetime.combine(date_options[day_choice], start_time)) 
+                                       & (bus_trips["datetime"] <= datetime.combine(date_options[day_choice], end_time))
+                                       & (bus_trips["direction_id"] == st.session_state.direction )]
             trip_ids = list(pd.unique(selected_trips["trip_id"]))
             st.write(f"Length {len(trip_ids)}")
             for id in trip_ids:
                 plot_graph(st.session_state.data[st.session_state.data["trip_id"]==id])
+                visualize(st.session_state.data[st.session_state.data["trip_id"]==id])