Update app.py

app.py

@@ -1,93 +1,65 @@
-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)
-
-# 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.")
+import numpy as np
+import pandas as pd
+import plotly.express as px
+
+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': '🎲'}]
+
+if 'name' not in st.session_state:
+    st.session_state.name = ''
+if 'dice_roll_history' not in st.session_state:
+    st.session_state.dice_roll_history = pd.DataFrame()
+
+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()
+
+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(num_rolls, dtype=np.uint64),
+                              'DiceNumberOfSides': [dice_type['sides']] * num_rolls,
+                              'DiceRollerName': [st.session_state.name] * num_rolls})
+if bonus_match:
+    new_roll_data['BonusMatchToDiceName'] = [bonus_dice_type] * num_rolls
+    new_roll_data['BonusMatchToDiceEmoji'] = [bonus_dice_emoji] * num_rolls
+dice_roll_history = dice_roll_history.append(new_roll_data, ignore_index=True)
+st.session_state.dice_roll_history = dice_roll_history
+
+if st.button('Download Results'):
+    filename = f'dice_roll_history_{st.session_state.name}.csv'
+    st.download_button(label='Download CSV', data=dice_roll_history.to_csv(index=False), file_name=filename, mime='text/csv')
+
+if st.session_state.dice_roll_history.shape[0] > 0:
+    st.markdown('### Dice Roll History')
+    st.dataframe(st.session_state.dice_roll_history)
 
 
 st.write("""