Added inout from user

app.py

@@ -74,21 +74,31 @@ def get_buses():
     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):
-    st.write("### Data Preview")
-    st.dataframe(data.head())
-
-    #st.write("### Histogram")
-    #column = st.selectbox("Select column for histogram", data.columns)
-    #fig, ax = plt.subplots()
-    #sns.histplot(data[column], kde=True, ax=ax)
-    #st.pyplot(fig)
-
-    #st.write("### Correlation Matrix")
-    #fig, ax = plt.subplots()
-    #sns.heatmap(data.corr(), annot=True, cmap="coolwarm", ax=ax)
-    #st.pyplot(fig)
+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
+    
+    plot_df = plot_df[["datetime", "vehicle_occupancystatus"]]
+    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)
+    #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
+    ).properties(
+        title="Vehicle Occupancy Status Over Time"
+    )
+    st.altair_chart(chart, use_container_width=True)
 
 # Streamlit UI
 def main():
@@ -152,38 +162,45 @@ def main():
             )
             st.write("### Selected Bus")
             st.write(f"{search}: {bus}")
+        
+        today = datetime.now()
+        tomorrow = today + timedelta(days=1)
+        today = today.date()
+        tomorrow = tomorrow.date()
+        
+        date_options = {
+            today.strftime("%d %B %Y") : today,
+            tomorrow.strftime("%d %B %Y") : tomorrow
+        }
 
-        #Plocka alla akruella trip_ids från buses
-        trips = buses_df[buses_df["route_long_name"]==bus]
+        day_choice = st.sidebar.radio("Select the day:", options=list(date_options.keys()))
 
-        trip_ids = list(trips["trip_id"])
+        # Add time input widgets in the sidebar
+        start_time = st.sidebar.time_input("Select a start time", value=None)
+        end_time = st.sidebar.time_input("Select an end time", value=None)
 
-        st.write(trip_ids[0])
-
-        #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
+        #Plocka alla aktuella trip_ids från buses
+        trips = buses_df[buses_df["route_long_name"]==bus]
+        bus_trips = st.session_state.data[st.session_state.data["route_long_name"]==bus]
+        bus_trips["datetime"] = pd.to_datetime(bus_trips["datetime"])
+        bus_trips["datetime"] = bus_trips["datetime"].dt.tz_convert(None)
+        
+        #TODO remove
+        trip_ids = list(trips["trip_id"])
         plot_df = st.session_state.data[st.session_state.data["trip_id"]==trip_ids[0]]
-        plot_df = plot_df[["datetime", "vehicle_occupancystatus"]]
-        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)
-        #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
-        ).properties(
-            title="Vehicle Occupancy Status Over Time"
-        )
-        st.altair_chart(chart, use_container_width=True)
+
+        print(f"start time {type(start_time)}")
+        print(f"end time {type(end_time)}")
+        print(f"day {type(day_choice)}")
+
+        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))]
+            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])
             
 
 
@@ -194,7 +211,8 @@ def main():
 
     # Display data and graphs
     if st.session_state.data is not None:
-        plot_graphs(st.session_state.data)
+        #plot_graphs(st.session_state.data)
+        st.write("Hi")
 
 main()