Added the possibility to search for a bus and started ploting

api.py

@@ -0,0 +1,3 @@
+import os
+
+os.environ["HOPSWORKS_API_KEY"] = "vjR4XXGkTzrmVZOS.5faua2qELLblTMrj9OXO6QpvXsQP1nu0EZfqWsQwUu8YgOz3NFpCAJTYTp6QNKQG"

app.py

@@ -7,6 +7,8 @@ import matplotlib.pyplot as plt
 import seaborn as sns
 from datetime import datetime, timedelta
 
+import api
+
 # Constants
 DATA_DIR = "data"
 TIMESTAMP_FILE = "last_download_time.txt"
@@ -62,6 +64,14 @@ def is_local_data_valid():
     except Exception as e:
         st.warning(f"Error reading timestamp: {e}")
         return False
+    
+def get_buses():
+    bus_df = st.session_state.data[["trip_id", "route_long_name", "route_short_name"]]
+    bus_df = bus_df.drop_duplicates()
+    bus_list = bus_df[["route_long_name", "route_short_name"]]
+    bus_list = bus_list.drop_duplicates()
+    short_bus_list = list(pd.unique(bus_df["route_short_name"]))
+    return bus_df, bus_list, short_bus_list
 
 # Plot graphs
 def plot_graphs(data):
@@ -108,11 +118,61 @@ def main():
         
         project = st.session_state.hopsworks_project
         data = fetch_data_from_feature_group(project, "predictions", 1)
-        print(data.head())
+        #print(data.head())
         filepath = save_data_locally(data, "data.csv")
         st.session_state.data = data
         st.success(f"Data fetched and saved locally at {filepath}")
 
+    buses_df, bus_list, short_bus = get_buses()
+
+    # Sidebar section for searching buses
+    st.sidebar.title("Search for your desired bus")
+
+    # Create a multiselect dropdown in the sidebar
+    search = st.sidebar.selectbox(
+        "Search for your bus number:",
+        options=short_bus,
+        help="Select one bus to view details."
+    )
+
+    # Display the results
+    if search:
+        route = bus_list[bus_list["route_short_name"]==search]
+        long_names = list(pd.unique(route["route_long_name"]))
+        if len(long_names)==1:
+            bus = long_names[0]
+            st.write("### Selected Bus")
+            st.write(f"{search}: {bus}")
+        else:
+            bus = st.sidebar.selectbox(
+            "Pick bus route:",
+            options=long_names,
+            help="Select one bus to view details."
+            )
+            st.write("### Selected Bus")
+            st.write(f"{search}: {bus}")
+
+        #Plocka alla akruella trip_ids från buses
+        trips = buses_df[buses_df["route_long_name"]==bus]
+
+        trip_ids = list(trips["trip_id"])
+
+        st.write(trip_ids[0])
+
+        #Nu vill vi plotta!
+        #Steg 2, fixa sa date ersätts av stop genom att mappa location till stop
+        plot_df = st.session_state.data[st.session_state.data["trip_id"]==trip_ids[0]]
+        plot_df = plot_df[["datetime", "vehicle_occupancystatus"]]
+        plot_df = plot_df.set_index("datetime")
+        st.line_chart(plot_df)
+        
+
+
+    else:
+        st.write("No buses selected. Please search in the sidebar.")
+    
+    
+
     # Display data and graphs
     if st.session_state.data is not None:
         plot_graphs(st.session_state.data)