Update app.py

app.py

@@ -1,84 +1,124 @@
-import os
-import pandas as pd
 import streamlit as st
+import numpy as np
+import pandas as pd
+import plotly.express as px
 
-# 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 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, axis=1)
-    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])
-
-# 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)
+    rolls = np.random.randint(1, dice_type['sides'] + 1, num_rolls, dtype=np.int64)
+    roll_counts = pd.Series(rolls).value_counts().sort_index()
+    return rolls, roll_counts
+
+def update_dice_roll_history(dice_roll_history, name, rolls, num_rolls, dice_type, bonus_dice_type='', bonus_dice_emoji=''):
+    new_roll_data = pd.DataFrame({
+        'Roll': rolls,
+        'Count': np.ones(num_rolls, dtype=np.int64),
+        'DiceNumberOfSides': [dice_type['sides']] * num_rolls,
+        'DiceRollerName': [name] * num_rolls
+    })
+    if bonus_dice_type:
+        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)
+    return dice_roll_history
+
+def download_results(dataframe, name):
+    filename = f'dice_roll_history_{name}.csv'
+    st.download_button(
+        label='Download CSV',
+        data=dataframe.to_csv(index=False),
+        file_name=filename,
+        mime='text/csv'
+    )
+
+def main():
+    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 = 'Username'
+    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,
+        value=1000,
+        step=1
+    )
+
+    rolls, roll_counts = build_rolls(dice_type, num_rolls)
+
+    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')
+    st.plotly_chart(fig)
+
+    bonus_match = False
+    bonus_dice_type = ''
+    bonus_dice_emoji = ''
+    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
+        dice_roll_history = update_dice_roll_history(
+            dice_roll_history,
+            st.session_state.name,
+            rolls,
+            num_rolls,
+            dice_type,
+            bonus_dice_type,
+            bonus_dice_emoji
+        )
+        st.session_state.dice_roll_history = dice_roll_history
+
+if st.button('Download Results'):
+    download_results(dice_roll_history, st.session_state.name)
+
+if dice_roll_history.shape[0] > 0:
+    st.markdown('### Dice Roll History')
+    st.dataframe(dice_roll_history)
 
-# Create a button to roll the dice
-if st.button("🎲 Roll Dice"):
-    # 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))}.")
-
-    # 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']}.")
-
-    # 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")
-
-    # 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)
+st.write("""
+🍞 Bread, 🥐 Croissant, 🥖 Baguette Bread, 🫓 Flatbread, 🥨 Pretzel, 🥯 Bagel, 🥞 Pancakes, 🧇 Waffle, 🧀 Cheese Wedge, 🍖 Meat on Bone, 🍗 Poultry Leg, 🥩 Cut of Meat, 🥓 Bacon, 🍔 Hamburger, 🍟 French Fries, 🍕 Pizza, 🌭 Hot Dog, 🥪 Sandwich, 🌮 Taco, 🌯 Burrito, 🫔 Tamale, 🥙 Stuffed Flatbread, 🧆 Falafel, 🥚 Egg, 🍳 Cooking, 🥘 Shallow Pan of Food, 🍲 Pot of Food, 🫕 Fondue, 🥣 Bowl with Spoon, 🥗 Green Salad, 🍿 Popcorn, 🧈 Butter, 🧂 Salt, 🥫 Canned Food, 🍱 Bento Box, 🍘 Rice Cracker, 🍙 Rice Ball, 🍚 Cooked Rice, 🍛 Curry Rice, 🍜 Steaming Bowl, 🍝 Spaghetti, 🍠 Roasted Sweet Potato, 🍢 Oden, 🍣 Sushi, 🍤 Fried Shrimp, 🍥 Fish Cake with Swirl, 🥮 Moon Cake, 🍡 Dango, 🥟 Dumpling, 🥠 Fortune Cookie, 🥡 Takeout Box, 🦪 Oyster, 🍦 Soft Ice Cream, 🍧 Shaved Ice, 🍨 Ice Cream, 🍩 Doughnut, 🍪 Cookie, 🎂 Birthday Cake, 🍰 Shortcake, 🧁 Cupcake, 🥧 Pie, 🍫 Chocolate Bar, 🍬 Candy, 🍭 Lollipop, 🍮 Custard, 🍯 Honey Pot, 🍼 Baby Bottle, 🥛 Glass of Milk, ☕ Hot Beverage, 🫖 Teapot, 🍵 Teacup Without Handle, 🍶 Sake, 🍾 Bottle with Popping Cork, 🍷 Wine Glass, 🍸 Cocktail Glass, 🍹 Tropical Drink, 🍺 Beer Mug, 🍻 Clinking Beer Mugs, 🥂 Clinking Glasses, 🥃 Tumbler Glass, 🫗 Pouring Liquid, 🥤 Cup with Straw, 🧋 Bubble Tea, 🧃 Beverage Box, 🧉 Mate, 🧊 Ice, 🥢 Chopsticks, 🍽️ Fork and Knife with Plate, 🍴 Fork and Knife, 🥄 Spoon, 🫙 Jar"
+""")
 
-# 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("""
-
 🍞 Bread, 🥐 Croissant, 🥖 Baguette Bread, 🫓 Flatbread, 🥨 Pretzel, 🥯 Bagel, 🥞 Pancakes, 🧇 Waffle, 🧀 Cheese Wedge, 🍖 Meat on Bone, 🍗 Poultry Leg, 🥩 Cut of Meat, 🥓 Bacon, 🍔 Hamburger, 🍟 French Fries, 🍕 Pizza, 🌭 Hot Dog, 🥪 Sandwich, 🌮 Taco, 🌯 Burrito, 🫔 Tamale, 🥙 Stuffed Flatbread, 🧆 Falafel, 🥚 Egg, 🍳 Cooking, 🥘 Shallow Pan of Food, 🍲 Pot of Food, 🫕 Fondue, 🥣 Bowl with Spoon, 🥗 Green Salad, 🍿 Popcorn, 🧈 Butter, 🧂 Salt, 🥫 Canned Food, 🍱 Bento Box, 🍘 Rice Cracker, 🍙 Rice Ball, 🍚 Cooked Rice, 🍛 Curry Rice, 🍜 Steaming Bowl, 🍝 Spaghetti, 🍠 Roasted Sweet Potato, 🍢 Oden, 🍣 Sushi, 🍤 Fried Shrimp, 🍥 Fish Cake with Swirl, 🥮 Moon Cake, 🍡 Dango, 🥟 Dumpling, 🥠 Fortune Cookie, 🥡 Takeout Box, 🦪 Oyster, 🍦 Soft Ice Cream, 🍧 Shaved Ice, 🍨 Ice Cream, 🍩 Doughnut, 🍪 Cookie, 🎂 Birthday Cake, 🍰 Shortcake, 🧁 Cupcake, 🥧 Pie, 🍫 Chocolate Bar, 🍬 Candy, 🍭 Lollipop, 🍮 Custard, 🍯 Honey Pot, 🍼 Baby Bottle, 🥛 Glass of Milk, ☕ Hot Beverage, 🫖 Teapot, 🍵 Teacup Without Handle, 🍶 Sake, 🍾 Bottle with Popping Cork, 🍷 Wine Glass, 🍸 Cocktail Glass, 🍹 Tropical Drink, 🍺 Beer Mug, 🍻 Clinking Beer Mugs, 🥂 Clinking Glasses, 🥃 Tumbler Glass, 🫗 Pouring Liquid, 🥤 Cup with Straw, 🧋 Bubble Tea, 🧃 Beverage Box, 🧉 Mate, 🧊 Ice, 🥢 Chopsticks, 🍽️ Fork and Knife with Plate, 🍴 Fork and Knife, 🥄 Spoon, 🫙 Jar")
 """)
 
+if name == 'main':
+main()