Update app.py

app.py

@@ -15,10 +15,12 @@ def roll_dice(num_rolls, dice_type):
 
 def plot_tokens(health_tokens, coin_tokens):
     fig = go.Figure()
-    health_foods = [FOOD_LIST[i] for i in health_tokens]
-    coin_foods = [FOOD_LIST[i] for i in coin_tokens]
-    fig.add_trace(go.Scatter(x=list(range(1, len(health_foods) + 1)), y=health_foods, name="💖 Health"))
-    fig.add_trace(go.Scatter(x=list(range(1, len(coin_foods) + 1)), y=coin_foods, name="💰 Coins"))
+    if len(health_tokens) > 0:
+        health_foods = [FOOD_LIST[i] for i in health_tokens]
+        fig.add_trace(go.Scatter(x=list(range(1, len(health_foods) + 1)), y=health_foods, name="💖 Health"))
+    if len(coin_tokens) > 0:
+        coin_foods = [FOOD_LIST[i] for i in coin_tokens]
+        fig.add_trace(go.Scatter(x=list(range(1, len(coin_foods) + 1)), y=coin_foods, name="💰 Coins"))
     fig.update_layout(title="Token Accumulation", xaxis_title="Rolls", yaxis_title="Tokens")
     st.plotly_chart(fig)