Update app.py

app.py

@@ -1,73 +1,93 @@
 import os
-import streamlit as st
-import numpy as np
 import pandas as pd
-import plotly.express as px
-
-st.set_page_config(layout='wide')
-
-st.title('🎲 Random Dice Game')
-
-dice_types = [{'name': 'Six-sided Dice', 'sides': 6, 'emoji': '🎲'},
-              {'name': 'Twenty-sided Dice', 'sides': 20, 'emoji': '🎯'},
-              {'name': 'Thirty-sided Dice', 'sides': 30, 'emoji': '🎯'},
-              {'name': 'One Hundred-sided Dice', 'sides': 100, 'emoji': '💯'}]
-
-def username_input():
-    if 'username' not in st.session_state:
-        st.session_state.username = ''
-    st.text_input('👤 Username', key='username')
-
-def display_dice_roll_distribution():
-    dice_type = st.selectbox('🎲 Choose a type of dice', dice_types, format_func=lambda d: f"{d['name']} {d['emoji']}")
-    num_rolls = st.slider('🔄 How many times do you want to roll the dice?', 1, 1000000, 1000)
-
-    rolls = np.random.randint(1, dice_type['sides'] + 1, num_rolls, dtype=np.uint64)
-    roll_counts = pd.Series(rolls).value_counts().sort_index()
-
-    fig = px.sunburst(names=[f'Roll {i}' for i in roll_counts.index],
-                      parents=['Dice Rolls'] * dice_type['sides'],
-                      values=roll_counts.values,
-                      color=[f'Roll {i}' for i in roll_counts.index],
-                      color_discrete_sequence=px.colors.qualitative.Dark24,
-                      maxdepth=2)
-    fig.update_layout(title='📈 Dice Roll Distribution', margin=dict(l=20, r=20, t=40, b=20), width=800, height=600)
-
-    show_labels = st.checkbox('🏷️ Show Labels', value=True)
-    if not show_labels:
-        fig.update_traces(textinfo='none')
-    fig.show()
-
-    return dice_type, rolls
-
-def display_bonus_match(bonus_match, bonus_dice_type, bonus_dice_emoji):
-    if bonus_match:
-        st.success(f'🎉 You rolled a {bonus_dice_type} {bonus_dice_emoji} on the first roll!')
-
-def update_dice_roll_history(dice_type, rolls):
-    bonus_match = False
-    for dice in dice_types:
-        if rolls[0] == dice['sides']:
-            bonus_match = True
-            bonus_dice_type = dice['name']
-            bonus_dice_emoji = dice['emoji']
-            break
-
-    dice_roll_history = st.session_state.dice_roll_history
-    new_roll_data = pd.DataFrame({'Roll': rolls,
-                                  'Count': np.ones(len(rolls), dtype=np.uint64),
-                                  'DiceNumberOfSides': [dice_type['sides']] * len(rolls),
-                                  'Username': [st.session_state.username] * len(rolls)})
-    if bonus_match:
-        new_roll_data['BonusMatchToDiceName'] = [bonus_dice_type] * len(rolls)
-        new_roll_data['BonusMatchToDiceEmoji'] = [bonus_dice_emoji] * len(rolls)
-    dice_roll_history = dice_roll_history.append(new_roll_data, ignore_index=True)
-    st.session_state.dice_roll_history = dice_roll_history
-
-    return bonus_match, bonus_dice_type, bonus_dice_emoji
-
-def download_dice_roll_history(include_name_column):
-    dice_roll
+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)
+
+# 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": "💯"},
+}
+
+# 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
+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)
+    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")
+
+    # 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"):
+    # Read the file into a DataFrame
+    data = pd.read_csv("SharedState.csv")
+
+    # Check if the user wants to include their name in the output
+    if st.checkbox("💻 Include Username in Output", value=True):
+        # Add the username to the DataFrame
+        data["DiceRollerName"] = st.session_state.username
+        # Check if there is a bonus match and add the bonus match column to the DataFrame
+        if bonus_match:
+            data["BonusMatchToDiceName"] = bonus_match
+            data["BonusMatchToDiceEmoji"] = bonus_emoji
+        # Check if the user wants to download the file
+        if download_roll_history:
+            st.download_button("📥 Download Roll History", data.to_csv(index=False), f"shared_state_{st.session_state.username}.csv", "text/csv")
+    else:
+        # Remove the username from the DataFrame
+        data = data.drop(columns=["DiceRollerName"])
+        # Check if there is a bonus match and add the bonus match column to the DataFrame
+        if bonus_match:
+            data["BonusMatchToDiceName"] = bonus_match
+            data["BonusMatchToDiceEmoji"] = bonus_emoji
+        # Check if the user wants to download the file
+        if download_roll_history:
+            st.download_button("📥 Download Roll History", data.to_csv(index=False), "shared_state.csv", "text/csv")
+else:
+    st.write("📜 No roll history available.")
 
 
 st.write("""