Update app.py

app.py

@@ -15,12 +15,10 @@ def roll_dice(num_rolls, dice_type):
 
 def plot_tokens(health_tokens, coin_tokens):
     fig = go.Figure()
-    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"))
+    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"))
     fig.update_layout(title="Token Accumulation", xaxis_title="Rolls", yaxis_title="Tokens")
     st.plotly_chart(fig)
 
@@ -50,14 +48,7 @@ for dice_type in DICE_TYPES:
 st.write("💰💖 Token Accumulation:")
 plot_tokens(history["health_tokens"], history["coin_tokens"])
 df = pd.concat([pd.DataFrame(history["roll_history"]), pd.DataFrame(history["health_tokens"], columns=["Health Tokens"]), pd.DataFrame(history["coin_tokens"], columns=["Coin Tokens"])], axis=1)
-
-if username:
-    username_str = username
-else:
-    username_str = "Anonymous"
-timestamp_str = datetime.now().strftime("%m-%d-%Y-%I-%M-%S-%p")
-filename = f"{username_str}_{timestamp_str}.csv"
-
+timestamp = datetime.now().strftime("%m-%d-%Y-%H-%M-%S")
+filename = f"{username}_{timestamp}.csv"
 df.to_csv(filename, index=False)
-st.markdown(f'<a href="data:file/csv;base64,{b64encode(open(filename, "rb").read()).decode()}" download="{filename}">Download CSV File</a>', unsafe_allow_html=True)
-
+st.markdown(f'<a href="data:file/csv;base64,{b64encode(open(filename, "rb").read()).decode()}" download="{filename}">Download CSV File</a>', unsafe_allow_html=True)