Update app.py

app.py

@@ -13,9 +13,9 @@ game_mechanics = ["Action Queue", "Action Retrieval", "Campaign / Battle Card Dr
                   "Solo / Solitaire Game", "Storytelling", "Variable Player Powers"]
 
 # Define a function to generate random values for each game mechanic
-def generate_values():
-    health_points = np.random.randint(50, 100, size=len(game_mechanics))
-    coins = np.random.randint(10, 50, size=len(game_mechanics))
+def generate_values(n):
+    health_points = np.random.randint(50, 100, size=n)
+    coins = np.random.randint(10, 50, size=n)
     return {"HealthPoints": health_points, "Coins": coins}
 
 # Define the Streamlit app
@@ -24,14 +24,15 @@ def app():
     st.write("This app displays a Treemap chart of game mechanics.")
 
     # Generate the data for the chart
-    values = generate_values()
+    n = len(game_mechanics)
+    values = generate_values(n)
     data = pd.DataFrame({
-        "category": ["Category 1"] * len(game_mechanics),
+        "category": ["Category 1"] * n,
         "mechanic": game_mechanics,
         "value": list(values.values()),
         "HealthPoints": values["HealthPoints"],
         "Coins": values["Coins"],
-        "description": ["Description"] * len(game_mechanics)
+        "description": ["Description"] * n
     })
     data["value"] = data["value"].apply(lambda x: sum(x))
 
@@ -47,4 +48,4 @@ def app():
     st.markdown(href, unsafe_allow_html=True)
 
 if __name__ == '__main__':
-    app()
+    app()