Update app.py

app.py

@@ -25,6 +25,9 @@ health_conditions = [
 # Create a DataFrame from the list dictionary
 df_top_conditions = pd.DataFrame(health_conditions)
 
+# Calculate the total spending
+total_spending = round(df_top_conditions["spending"].sum(), 1)
+
 # Define the reroll function
 def reroll():
     # Randomly select a health condition from the top 10
@@ -39,11 +42,17 @@ def reroll():
     if reroll:
         reroll()
 
-# Create the map graph using Plotly Express
-fig_map = px.choropleth(locations=["CA", "FL", "MN"], locationmode="USA-states", color=[1, 2, 3], scope="usa")
-fig_map.update_layout(geo=dict(bgcolor= "rgba(0, 0, 0, 0)", lakecolor="rgb(255, 255, 255)"))
+# Define the sunburst chart
+fig_sunburst = go.Figure(go.Sunburst(
+    labels=df_top_conditions["emoji"] + " " + df_top_conditions["condition"],
+    parents=[""] * top_n,
+    values=df_top_conditions["spending"],
+    maxdepth=2
+))
+
+# Customize the layout of the sunburst chart
+fig_sunburst.update_layout(title=f"Top {top_n} Health Conditions in {', '.join(states)} by Spending (Total: ${total_spending}B)")
 
-# Create the bar chart using Plotly Graph Objects
-fig_bar = go.Figure(go.Bar(x=df_top_conditions["emoji"] + " " + df_top_conditions["condition"], y=df_top_conditions["spending"], marker_color="#1f77b4"))
-fig_bar.update_layout(title=f"Top {top_n} Health Conditions in {', '.join(states)} by Spending (Total: ${total_spending}B)",
-                      yaxis_title="Spending ($B)", plot_bgcolor='rgba(0,0
+# Display the sunburst chart and reroll button in the Streamlit app
+st.plotly_chart(fig_sunburst)
+st.button("Reroll", on_click=reroll)