Update app.py

app.py

@@ -18,6 +18,10 @@ class State(TypedDict):
     roll: int
 
 # Nodes for LangGraph
+def roll_node(state):
+    state["roll"] = random.randint(1, 6) + random.randint(1, 6)
+    return state
+
 def input_node(state):
     return state
 
@@ -34,17 +38,17 @@ def failure_node(state):
     return state
 
 def determine_outcome(state) -> Literal["node_2", "node_3"]:
-    roll = random.randint(1, 6) + random.randint(1, 6)
-    state['roll'] = roll
-    return "node_2" if roll <= state["bet"] else "node_3"
+    return "node_2" if state["roll"] <= state["bet"] else "node_3"
 
 # Build LangGraph
 builder = StateGraph(State)
+builder.add_node("roll_node", roll_node)
 builder.add_node("node_1", input_node)
 builder.add_node("node_2", success_node)
 builder.add_node("node_3", failure_node)
 
-builder.add_edge(START, "node_1")
+builder.add_edge(START, "roll_node")
+builder.add_edge("roll_node", "node_1")
 builder.add_conditional_edges("node_1", determine_outcome)
 builder.add_edge("node_2", END)
 builder.add_edge("node_3", END)
@@ -55,7 +59,7 @@ graph = builder.compile()
 def play_game(bet: int):
     if bet < 1 or bet > 12:
         return "Please enter a value between 1 and 12.", global_score["total"], []
-    result = graph.invoke({"bet": bet, "result": "", "score": global_score["total"], "roll": 0})
+    result = graph.invoke({"bet": bet, "result": "", "score": global_score["total"]})
     game_history.append({"Bet": bet, "Roll": result["roll"], "Result": result["result"], "Score": result["score"]})
     history_table_data = [[g["Bet"], g["Roll"], g["Result"], g["Score"]] for g in game_history[::-1]]
     return result["result"], result["score"], history_table_data