Update app.py

app.py

@@ -2,72 +2,39 @@ 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")
 
-# 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)
+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 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])
-
-# 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 string with the rolls for the selected dice type
-    #orig: rolls = [str(pd.Series([dice_types[dice_type]["sides"]] * num_rolls).apply(pd.Series.sample, replace=True))]
-    #replaced:
-rolls = pd.DataFrame([dice_types[dice_type]["sides"]] * num_rolls)[0].apply(pd.Series.sample, replace=True).tolist()
-
-
-#rolls = [str(pd.Series([dice_types[dice_type]["sides"]] * num_rolls).apply(pd.Series.sample, replace=True))]
+rolls = [str(pd.Series([dice_types[dice_type]["sides"]] * num_rolls).apply(pd.Series.sample, replace=True))]
 rolls = rolls[0].strip('[]').split('\n')
 
-# Create a button to roll the dice
 if st.button("🎲 Roll Dice"):
-    # Display the rolls
     display_roll = [int(i) for i 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))}.")
-
-    # Check if the count is a multiple of the dice sides to see if there's a bonus match
-    bonus_match = None
-    count = sum(display_roll)
+    bonus_match, count = (None, sum(display_roll))
     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']}.")
-
-    # Create a DataFrame with the rolls, count, and dice number of sides
     df = pd.DataFrame({'Roll': rolls, 'Count': count, 'DiceNumberOfSides': dice_types[dice_type]['sides']})
-
-    # 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)
-
-# Check if the file exists and display it as a download link
 if os.path.exists("SharedState.csv"):
-    data
+    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.")
 
 st.write("""