Start på försök att ändra loggan

app.py

@@ -196,7 +196,7 @@ def plot_graph_title(plot_df, stop, time):
     st.altair_chart(chart, use_container_width=True)
 
 
-def visualize(filtered_data):
+def visualize(filtered_data, stop_name):
     import folium
     from streamlit_folium import st_folium
 
@@ -217,11 +217,18 @@ def visualize(filtered_data):
 
     # 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)
+        if row["stop_name"] == stop_name:
+            folium.Marker(
+                [row['stop_lat'], row['stop_lon']],
+                popup=f"Bus stop: {row['stop_name']} Bus occupancy: {categories[row['vehicle_occupancystatus']] }",
+                icon = folium.Icon(colour="yellow", icon="bus-simple", prefix="fa")
+            ).add_to(m)
+        else:
+            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)
@@ -361,6 +368,7 @@ def main():
 
             chioce = selected_trips[selected_trips["stop_name"]==stop_choice]
             chioce.head()
+            #TODO ta bort stop_name:)
             chioce = chioce[["trip_id", "stop_name", "datetime"]]
             #Ev lägga stop_chioce i session_state
 
@@ -374,7 +382,7 @@ def main():
             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])
+                visualize(st.session_state.data[st.session_state.data["trip_id"]==id], stop_choice)