Update app.py

app.py

@@ -8,31 +8,39 @@ 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('Any 🎲Dice🎲 Way - STEM Math and Data Science Numerical Potluck')
+
 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 'username' not in st.session_state:
     st.session_state.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)
 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']:
@@ -40,6 +48,7 @@ for dice in dice_types:
         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),
@@ -48,27 +57,20 @@ new_roll_data = pd.DataFrame({'Roll': 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
+
 include_name_column = st.checkbox('Include Username Column in Downloaded CSV', value=True)
+
 if st.button('Download Results'):
-    filename = f'dice_roll_history_{st.session_state.username}{dice_type["emoji"]}.csv'
-    if not include_name_column:
+    if include_name_column:
+        filename = f'dice_roll_history_{st.session_state.username}{dice_type["emoji"]}.csv'
+    else:
         dice_roll_history = dice_roll_history.drop(columns=['Username'])
         filename = f'dice_roll_history{dice_type["emoji"]}.csv'
-    st.download_button(label='Download CSV', data=dice_roll_history.to_csv(index=False), file_name=filename, mime='text/csv')
-csv_files = [f for f in os.listdir('.') if os.path.isfile
-              f.endswith('.csv') and 'dice_roll_history' in f]
-if csv_files:
-    st.write('Downloaded files:')
-    for csv_file in csv_files:
-        st.markdown(f'[Download {csv_file}]({csv_file})')
-st.write('---')
-st.subheader('Settings')
-st.session_state.username = st.text_input('Username', st.session_state.username)
-st.write('---')
-st.subheader('Dice Roll History')
-st.write(dice_roll_history)
+    st.download_button(label='Download CSV', data=dice_roll_history.to_csv(index=False), file_name=filename, mime
+
 
 st.write("""
 🍞 Bread