Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
+import pandas as pd
 import folium
 from folium.plugins import MarkerCluster, HeatMap
-import pandas as pd
+from streamlit_plotly_events import plotly_events
 import plotly.express as px
 
 # Function to create the map
@@ -76,16 +77,18 @@ def app():
     # Bar Chart visualization
     st.subheader("Accidents by Severity")
     bar_chart = create_bar_chart(df, selected_year)
-    selected_points = st.plotly_chart(bar_chart)
     
-    # Update severity based on bar chart selection
+    # Capture user interaction
+    selected_points = plotly_events(bar_chart, click_event=True, hover_event=False)
     if selected_points:
-        selected_data = selected_points['points'][0]['x']  # Get the clicked bar's x value
-        st.session_state["selected_severity"] = selected_data
+        # Get the selected severity from the clicked bar
+        selected_severity = selected_points[0]['x']
+        st.session_state["selected_severity"] = selected_severity
+    else:
+        selected_severity = st.session_state["selected_severity"]
     
     # Map visualization
     st.subheader("Accidents Map")
-    selected_severity = st.session_state["selected_severity"]
     map_obj = create_map(df, selected_year, selected_severity)
     st.components.v1.html(map_obj._repr_html_(), height=600)