Update app.py

app.py

@@ -8,12 +8,14 @@ from typing_extensions import TypedDict
 from langgraph.graph import StateGraph, START, END
 
 # Define the game state
-global_score = {"total": 0}  # Track winnings globally
+global_score = {"total": 0}
+game_history = []  # Track game rounds
 
 class State(TypedDict):
     bet: int
     result: str
     score: int
+    roll: int
 
 # Nodes for LangGraph
 def input_node(state):
@@ -21,19 +23,19 @@ def input_node(state):
 
 def success_node(state):
     global_score["total"] += 10
-    state["result"] = f"You win! +$10"
+    state["result"] = f"🎉 You win! +$10 (Roll: {state['roll']})"
     state["score"] = global_score["total"]
     return state
 
 def failure_node(state):
     global_score["total"] -= 5
-    state["result"] = f"You lose! -$5"
+    state["result"] = f"😞 You lose! -$5 (Roll: {state['roll']})"
     state["score"] = global_score["total"]
     return state
 
 def determine_outcome(state) -> Literal["node_2", "node_3"]:
     roll = random.randint(1, 6) + random.randint(1, 6)
-    print(f"[Roll]: {roll} vs Bet: {state['bet']}")
+    state['roll'] = roll
     return "node_2" if roll <= state["bet"] else "node_3"
 
 # Build LangGraph
@@ -52,19 +54,25 @@ graph = builder.compile()
 # Gradio UI function
 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"]})
-    return result["result"], result["score"]
+        return "Please enter a value between 1 and 12.", global_score["total"], "", []
+    result = graph.invoke({"bet": bet, "result": "", "score": global_score["total"], "roll": 0})
+    dice_img = f"https://raw.githubusercontent.com/mnaylor7/dice-roll-animation/main/dice-{result['roll']}.gif" if 2 <= result['roll'] <= 12 else ""
+    game_history.append({"Bet": bet, "Roll": result["roll"], "Result": result["result"], "Score": result["score"]})
+    return result["result"], result["score"], dice_img, game_history[::-1]
 
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## 🎲 Dice Bet Game with LangGraph")
     with gr.Row():
         bet_input = gr.Number(label="Your Bet (1-12)", value=7, precision=0)
+        play_button = gr.Button("🎲 Roll Dice")
+    with gr.Row():
         result_output = gr.Textbox(label="Result")
         score_output = gr.Number(label="Total Winnings", value=0, precision=0, interactive=False)
-    play_button = gr.Button("Roll Dice")
-    play_button.click(play_game, inputs=[bet_input], outputs=[result_output, score_output])
+    dice_display = gr.Image(label="Dice Roll Animation", visible=True)
+    history_table = gr.Dataframe(headers=["Bet", "Roll", "Result", "Score"], label="Game History", interactive=False)
+
+    play_button.click(play_game, inputs=[bet_input], outputs=[result_output, score_output, dice_display, history_table])
 
 if __name__ == "__main__":
     demo.launch()