Update app.py

app.py

@@ -1,39 +1,44 @@
 import streamlit as st
 import numpy as np
 import pandas as pd
-import pygraphviz as pgv
+import plotly.graph_objects as go
 from datetime import datetime
 from base64 import b64encode
 
+# Define general functions
 FOOD_LIST = {4: "🍔", 6: "🍟", 8: "🌮", 10: "🍕", 12: "🍩", 20: "🥗", 50: "🍣", 100: "🍾"}
-DICE_TYPES = [4, 6, 8, 10, 12, 20, 50, 100]
-DEFAULT_ROLLS = 3
 
 def roll_dice(num_rolls, dice_type):
     rolls = np.random.randint(1, dice_type + 1, size=num_rolls)
     return rolls
 
 def plot_tokens(health_tokens, coin_tokens):
-    dot = pgv.AGraph(strict=False, directed=True)
-    dot.node_attr['shape'] = 'plaintext'
-    dot.edge_attr['arrowhead'] = 'none'
-
-    health_foods = [FOOD_LIST[i] for i in health_tokens]
-    coin_foods = [FOOD_LIST[i] for i in coin_tokens]
-
-    health_tokens_str = " | ".join(health_foods)
-    coin_tokens_str = " | ".join(coin_foods)
-
-    dot.add_edge('Health', health_tokens_str)
-    dot.add_edge('Coins', coin_tokens_str)
-
-    st.graphviz_chart(dot.string())
+    fig = go.Figure()
+    fig.add_trace(go.Sankey(
+        node = {
+            "label": ["Health", "Coins"] + [FOOD_LIST[i] for i in DICE_TYPES],
+            "pad": 15
+        },
+        link = {
+            "source": [0, 1] + list(range(2, len(DICE_TYPES) + 2)),
+            "target": [2] * len(DICE_TYPES) + [3 + i for i in range(len(DICE_TYPES))],
+            "value": health_tokens + coin_tokens
+        },
+    ))
+    st.plotly_chart(fig)
 
+# Define Streamlit app
+st.set_page_config(page_title="Slot Machine Game", page_icon=":game_die:")
 st.title("🎲 Slot Machine Game")
-username = st.text_input("👤 Enter your username:")
-num_rolls = st.slider("🔢 Choose the number of rolls:", 1, 100, DEFAULT_ROLLS)
 
+# Sidebar
+username = st.sidebar.text_input("👤 Enter your username:")
+num_rolls = st.sidebar.slider("🔢 Choose the number of rolls:", 1, 100, 3)
+
+# Main content
+DICE_TYPES = [4, 6, 8, 10, 12, 20, 50, 100]
 history = {"health_tokens": [0], "coin_tokens": [0]}
+
 for dice_type in DICE_TYPES:
     rolls = roll_dice(num_rolls, dice_type)
     highest_rolls = sum(roll == dice_type for roll in rolls)
@@ -55,10 +60,10 @@ for dice_type in DICE_TYPES:
     history["roll_history"] = {**history.get("roll_history", {}), dice_type: rolls}
     history["coin_tokens"].append(history["coin_tokens"][-1] + coin_tokens_added)
 
-st.write("💰💖 Token Accumulation:")
 plot_tokens(history["health_tokens"], history["coin_tokens"])
 
 df = pd.concat([pd.DataFrame(history["roll_history"]), pd.DataFrame(history["health_tokens"], columns=["Health Tokens"]), pd.DataFrame(history["coin_tokens"], columns=["Coin Tokens"])], axis=1)
+
 timestamp = datetime.now().strftime("%m-%d-%Y-%H-%M-%S")
 filename = f"{username}_{timestamp}.csv"
 df.to_csv(filename, index=False)