Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 import plotly.express as px
 import plotly.graph_objects as go
+import os
 
 PLAYERS = {
     "Player 1": {"Sketch": "🎨", "Character": "🧙‍♀️", "Player Board": "📜", "Action Dice": "🎲", "Health Tokens": "❤️", "Coin Tokens": "💰", "Battle Tokens": "⚔️", "Attack": 5, "Defense": 3},
@@ -54,7 +55,7 @@ def battle_player_card():
         st.error("The monster defeated you.")
     save_data()
 
-def load_and_display_player_data():
+def load_and_display_player_data_old():
     player_data = pd.read_csv("player_data.csv")
     st.write("## Player Data")
     st.write(player_data)
@@ -70,6 +71,24 @@ def load_and_display_player_data():
     fig.update_layout(margin=dict(t=50, b=0, l=0, r=0), height=400)
     st.plotly_chart(fig)
     
+def load_and_display_player_data():
+    if os.path.exists("player_data.csv"):
+        player_data = pd.read_csv("player_data.csv")
+        st.write("## Player Data")
+        st.write(player_data)
+
+        fig = px.treemap(player_data,
+                         path=['Player'],
+                         values='Health Tokens',
+                         color='Attack',
+                         color_continuous_scale='Blues',
+                         title='Player Attack Power and Health Tokens')
+        fig.update_traces(textinfo='label+value+percent entry',
+                          hovertemplate='<b>%{label}</b><br>Attack: %{color}<br>Health: %{value}<br>Percent: %{percentEntry:.2f}%<extra></extra>')
+        fig.update_layout(margin=dict(t=50, b=0, l=0, r=0), height=400)
+        st.plotly_chart(fig)
+    else:
+        st.write("Player data file not found.")
 def main():
     st.set_page_config(page_title="Player Cards and Monsters")
     st.title("Player Cards and Monsters")