Update app.py

app.py

@@ -26,4 +26,25 @@ def app():
     # Generate the data for the chart
     values = generate_values()
     data = pd.DataFrame({
-        "category": ["Category 
+        "category": ["Category 1"] * len(game_mechanics),
+        "mechanic": game_mechanics,
+        "value": list(values.values()),
+        "HealthPoints": values["HealthPoints"],
+        "Coins": values["Coins"],
+        "description": ["Description"] * len(game_mechanics)
+    })
+    data["value"] = data["value"].apply(lambda x: sum(x))
+
+    # Draw the chart
+    fig = px.treemap(data, path=["category", "mechanic"], values="value",
+                     color="HealthPoints", hover_data=["Coins"])
+    st.plotly_chart(fig)
+
+    # Display a download link for the data
+    csv = data.to_csv(index=False)
+    b64 = base64.b64encode(csv.encode()).decode()
+    href = f'<a href="data:file/csv;base64,{b64}" download="game_mechanics.csv">Download CSV</a>'
+    st.markdown(href, unsafe_allow_html=True)
+
+if __name__ == '__main__':
+    app()