Update app.py

app.py

@@ -47,26 +47,34 @@ def display_bonus_match(bonus_match, bonus_dice_type, bonus_dice_emoji):
         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
+    history = st.session_state.dice_roll_history
+    count = st.session_state.count
+    dice_name = dice_type['name']
+    dice_emoji = dice_type['emoji']
+    bonus_match = None
+
+    # Check if the count matches the number of sides of the dice
+    if count > 0 and count % dice_type['sides'] == 0:
+        bonus_match = dice_type['name']
+        bonus_emoji = dice_type['emoji']
+
+    # Update the dice roll history
+    if history is None:
+        history = pd.DataFrame({'Roll': rolls, 'Count': count, 'DiceNumberOfSides': dice_type['sides']})
+    else:
+        new_history = pd.DataFrame({'Roll': rolls, 'Count': count, 'DiceNumberOfSides': dice_type['sides']})
+        history = history.append(new_history, ignore_index=True)
+
+    if bonus_match is not None:
+        history['BonusMatchToDiceName'] = bonus_match
+        history['BonusMatchToDiceEmoji'] = bonus_emoji
+
+    # Add username to history
+    history['Username'] = st.session_state.username
+
+    st.session_state.dice_roll_history = history
+    return history
 
-    return bonus_match
 
 def display_dice_roll_results():
     download_history = st.checkbox('📥 Download Roll History', value=False)