Update app.py

app.py

@@ -2,41 +2,79 @@ import os
 import pandas as pd
 import streamlit as st
 
+# Set page title and icon
 st.set_page_config(page_title="Dice Roller", page_icon="🎲")
+
+# Set page title
 st.title("🎲 Dice Roller")
 
-if 'username' not in st.session_state: st.session_state.username = st.text_input("👤 Username", max_chars=50)
+# Check if username is already set, if not, create a text input to set it
+if 'username' not in st.session_state:
+    st.session_state.username = st.text_input("👤 Username", max_chars=50)
+
+# Define the dice types
+dice_types = {
+    "🎲 Six-sided dice": {"sides": 6, "name": "d6", "emoji": "🎲"},
+    "🎲 Thirty-sided dice": {"sides": 30, "name": "d30", "emoji": "🎲"},
+    "🎲 Twenty-sided dice": {"sides": 20, "name": "d20", "emoji": "🎲"},
+    "🎲 One hundred-sided dice": {"sides": 100, "name": "d100", "emoji": "💯"},
+}
 
-dice_types = {"🎲 Six-sided dice": {"sides": 6, "name": "d6", "emoji": "🎲"},
-              "🎲 Thirty-sided dice": {"sides": 30, "name": "d30", "emoji": "🎲"},
-              "🎲 Twenty-sided dice": {"sides": 20, "name": "d20", "emoji": "🎲"},
-              "🎲 One hundred-sided dice": {"sides": 100, "name": "d100", "emoji": "💯"}}
+# Create a function to build the rolls pandas series to be a list of dictionaries
+def build_rolls(dice_type, num_rolls):
+    sides = dice_types[dice_type]["sides"]
+    rolls = pd.Series([sides] * num_rolls).apply(pd.Series.sample, replace=True)
+    rolls = rolls.tolist()
+    count = sum(rolls)
+    return [{"Roll": roll, "Count": count, "DiceNumberOfSides": sides} for roll in rolls]
 
+# Create a selectbox to choose the dice type
 dice_type = st.selectbox("🎲 Select a dice type", options=list(dice_types.keys()), format_func=lambda x: x.split(' ')[-2])
-num_rolls = st.slider("🎲 How many times do you want to roll the dice?", 1, 1000000, 100)
-#rolls = [str(pd.Series([dice_types[dice_type]["sides"]] * num_rolls).apply(pd.Series.sample, replace=True))]
-rolls = pd.Series([dice_types[dice_type]["sides"]] * num_rolls).apply(pd.Series.sample, replace=True).astype(str).tolist()
 
-rolls = rolls[0].strip('[]').split('\n')
+# Create a slider to choose the number of rolls
+num_rolls = st.slider("🎲 How many times do you want to roll the dice?", 1, 1000000, 100)
 
+# Create a button to roll the dice
 if st.button("🎲 Roll Dice"):
-    display_roll = [int(i) for i in rolls]
+    # Build the rolls list of dictionaries
+    rolls = build_rolls(dice_type, num_rolls)
+
+    # Display the rolls
+    display_roll = [roll["Roll"] for roll in rolls]
     st.write(f"{dice_types[dice_type]['emoji']} You rolled {len(display_roll)} {dice_types[dice_type]['name']} dice: {', '.join(map(str, display_roll))}.")
-    bonus_match, count = (None, sum(display_roll))
+
+    # Check if the count is a multiple of the dice sides to see if there's a bonus match
+    bonus_match = None
+    count = rolls[0]["Count"]
     if count % dice_types[dice_type]['sides'] == 0:
         bonus_match = dice_types[dice_type]['name']
+        bonus_emoji = dice_types[dice_type]['emoji']
         st.write(f"🎉 Bonus match! You rolled a multiple of {dice_types[dice_type]['sides']}.")
-    df = pd.DataFrame({'Roll': rolls, 'Count': count, 'DiceNumberOfSides': dice_types[dice_type]['sides']})
+
+    # Create a DataFrame with the rolls, count, and dice number of sides
+    df = pd.DataFrame(rolls)
+
+    # Check if the file exists
     file_exists = os.path.exists("SharedState.csv")
-    if file_exists: df.to_csv("SharedState.csv", mode='a', header=False, index=False)
-    else: df.to_csv("SharedState.csv", mode='w', header=True, index=False)
 
+    # Append the new rolls to the file
+    if file_exists:
+        df.to_csv("SharedState.csv", mode='a', header=False, index=False)
+    # If the file doesn't exist, create it and write the headers
+    else:
+        df.to_csv("SharedState.csv", mode='w', header=True, index=False)
+
+# Create a checkbox to download the roll history
 download_roll_history = st.checkbox("📥 Download Roll History", value=True)
-if os.path.exists("SharedState.csv"):
-    if download_roll_history:
-        with open("SharedState.csv", "rb") as f:
-            st.download_button(label="📥 Download Roll History", data=f, file_name="SharedState.csv", mime="text/csv")
-else: st.warning("No roll history found.")
+
+# Check if the file exists and display it as a download link
+if os.path.exists("SharedState.csv") and download_roll_history:
+    with open("SharedState.csv", "r") as f:
+        file_contents = f.read()
+    st.markdown(f"<a href='data:file/csv;base64,{file_contents.encode().b64encode()}'>📥 Download Roll History</a>", unsafe_allow_html=True)
+
+
+
 
 st.write("""