Update app.py

app.py

@@ -114,8 +114,8 @@ upper_categories, lower_categories = categories[:6], categories[6:]
 def update_components(*inputs):
     # Parse inputs
     scorecard = {c: inputs[i] for i, c in enumerate(categories)}
-    roll = inputs[len(categories)]
-    dice = np.array(inputs[len(categories) + 1 :])
+    roll, dice = inputs[len(categories):]
+    dice = np.array([int(d) for d in str(dice) if d in "123456"])
 
     # Compute totals
     scorecard_values = [0 if v == EMPTY else v for v in scorecard.values()]
@@ -123,7 +123,7 @@ def update_components(*inputs):
     bonus = 35 if upper > 62 else 0
 
     # Compute optimal action
-    if (EMPTY not in dice) and (EMPTY in scorecard.values()):
+    if (len(dice) == 5) and (EMPTY in scorecard.values()):
         state = np.clip(upper, 0, 63) + sum(2 ** (6 + i) for i, c in enumerate(categories) if scorecard[c] != EMPTY)
         s = get_score(state)[roll]
         r5 = R5_indices[tuple(np.bincount(dice, minlength=7)[1:])]
@@ -172,13 +172,10 @@ This app recommends the optimal strategy for playing solitaire Yahtzee, based on
         with gr.Column():
             gr.Markdown("### Current Roll")
             roll_input = gr.Dropdown([1, 2, 3], label="Roll number", value=1)
-            dice_inputs = [
-                gr.Dropdown([EMPTY, 1, 2, 3, 4, 5, 6], label=f"Die {i+1}")
-                for i in range(5)
-            ]
+            dice_input = gr.Textbox(label="Dice values", lines=1)
             action_box = gr.Textbox(label="Optimal action", lines=2)
 
-    inputs = list(score_inputs.values()) + [roll_input] + dice_inputs
+    inputs = list(score_inputs.values()) + [roll_input, dice_input]
     outputs = [upper_score, bonus_score, lower_score, total_score, action_box]
 
     for inp in inputs: