Update app.py

app.py

@@ -12,9 +12,8 @@ def roll_dice(n):
     return np.random.randint(1, DICE_SIDES + 1, n)
 
 def display_dice(dice):
-    #return " ".join([emojize(":game_die:") for _ in range(dice)])
-    #return " ".join([emojize(":game_die:") for _ in range(dice.item())])
-    return " ".join([emojize(":game_die:") for _ in range(dice.flatten())])
+    # Use the length of the dice array to determine how many emojis to display
+    return " ".join([emojize(":game_die:") for _ in dice])
     
 def update_scores(scores, dice):
     for i, score in enumerate(scores):
@@ -43,10 +42,11 @@ scores = [0 for _ in range(players)]
 rolled_dice = roll_dice(players * dice_count)
 scores = update_scores(scores, rolled_dice)
 
-# Display game board
-st.write(f"Player 1 rolled: {display_dice(rolled_dice[:dice_count])}")
-if players > 1:
-    st.write(f"Player 2 rolled: {display_dice(rolled_dice[dice_count:])}")
+# Display game board for all players
+for player in range(players):
+    start_index = player * dice_count
+    end_index = start_index + dice_count
+    st.write(f"Player {player + 1} rolled: {display_dice(rolled_dice[start_index:end_index])}")
 
 st.write(f"Scores: {scores}")
 
@@ -55,7 +55,8 @@ if st.button("Roll again"):
     scores = [0 for _ in range(players)]
     rolled_dice = roll_dice(players * dice_count)
     scores = update_scores(scores, rolled_dice)
-    st.write(f"Player 1 rolled: {display_dice(rolled_dice[:dice_count])}")
-    if players > 1:
-        st.write(f"Player 2 rolled: {display_dice(rolled_dice[dice_count:])}")
-    st.write(f"Scores: {scores}")
+    for player in range(players):
+        start_index = player * dice_count
+        end_index = start_index + dice_count
+        st.write(f"Player {player + 1} rolled: {display_dice(rolled_dice[start_index:end_index])}")
+    st.write(f"Scores: {scores}")