Update src/streamlit_app.py

src/streamlit_app.py

@@ -269,23 +269,25 @@ fig.update_layout(
 # 4. Render
 st.plotly_chart(fig, use_container_width=True)
 
-# Aggregate counts by region & crime
-sun = (
-    df
-      .groupby(["RegionName","crm_cd_desc"])
-      .size()
-      .reset_index(name="Count")
-)
-
-fig = px.sunburst(
-    sun,
-    path=["RegionName","crm_cd_desc"],
-    values="Count",
-    color="Count",
-    color_continuous_scale="Agsunset",
-    title="Crime Distribution: Region → Crime Type"
+# Prepare source/target/value lists
+grouped = tm.copy()
+regions = list(grouped["RegionName"].unique())
+crimes  = list(grouped["crm_cd_desc"].unique())
+
+labels = regions + crimes
+source = grouped["RegionName"].apply(lambda x: labels.index(x))
+target = grouped["crm_cd_desc"].apply(lambda x: labels.index(x))
+value  = grouped["Count"]
+
+fig = go.Figure(data=[go.Sankey(
+    node=dict(label=labels, pad=15, thickness=20),
+    link=dict(source=source, target=target, value=value)
+)])
+fig.update_layout(
+    title_text="Sankey: Region → Crime Type Flows",
+    font_size=10,
+    margin=dict(t=50, l=25, r=25, b=25)
 )
-fig.update_layout(margin=dict(t=50, l=25, r=25, b=25))
 st.plotly_chart(fig, use_container_width=True)