Update app.py

app.py

@@ -3,61 +3,113 @@ import pandas as pd
 import plotly.express as px
 import random
 import json
+from datetime import datetime
 from streamlit_flow import streamlit_flow
 from streamlit_flow.elements import StreamlitFlowNode, StreamlitFlowEdge
 from streamlit_flow.layouts import TreeLayout
 
+# Rich data structures for game content
+SITUATIONS = [
+    {
+        "id": "village_peril",
+        "name": "The Village in Peril",
+        "description": "A once-peaceful village is shrouded in dark miasma. Villagers cower in fear as shadowy figures lurk in the mist.",
+        "emoji": "🏘️"
+    },
+    {
+        "id": "cursed_artifact",
+        "name": "The Cursed Artifact",
+        "description": "Deep in an ancient tomb, a glowing artifact pulses with malevolent energy, its whispers echoing in your mind.",
+        "emoji": "🏺"
+    },
+    {
+        "id": "sacred_pact",
+        "name": "The Sacred Pact",
+        "description": "At a moonlit shrine, spirits of the land gather, awaiting a mediator to forge a new covenant between realms.",
+        "emoji": "🌙"
+    },
+    {
+        "id": "shapeshifter_challenge",
+        "name": "The Shapeshifter's Challenge",
+        "description": "A rival kitsune issues a challenge, their forms flickering between human and fox. The air crackles with competitive energy.",
+        "emoji": "🦊"
+    },
+    {
+        "id": "fox_fire_trial",
+        "name": "The Fox Fire Trial",
+        "description": "Sacred blue flames dance before you, a test of your mastery over kitsune magic. The heat is both inviting and intimidating.",
+        "emoji": "🔥"
+    }
+]
+
+ACTIONS = [
+    {
+        "id": "fox_fire",
+        "name": "Use Fox Fire",
+        "description": "Summon mystical flames to illuminate the darkness or ward off evil spirits.",
+        "emoji": "🔥"
+    },
+    {
+        "id": "shapeshift",
+        "name": "Shapeshift",
+        "description": "Transform your appearance to blend in or deceive others.",
+        "emoji": "🦊"
+    },
+    {
+        "id": "possess_object",
+        "name": "Possess an Object",
+        "description": "Infuse your spirit into an inanimate object to manipulate or gather information.",
+        "emoji": "👻"
+    },
+    {
+        "id": "call_spirits",
+        "name": "Call Upon Ancient Spirits",
+        "description": "Invoke the wisdom and power of your ancestors to guide you.",
+        "emoji": "🌟"
+    },
+    {
+        "id": "use_artifact",
+        "name": "Use Mystical Artifact",
+        "description": "Activate a powerful magical item to unleash its effects.",
+        "emoji": "🗡️"
+    }
+]
+
 def generate_situation():
-    situations = [
-        "The Village in Peril",
-        "The Cursed Artifact",
-        "The Sacred Pact",
-        "The Shapeshifter's Challenge",
-        "The Fox Fire Trial"
-    ]
-    return random.choice(situations)
+    return random.choice(SITUATIONS)
 
 def generate_actions():
-    actions = [
-        ("Use Fox Fire", "🔥"),
-        ("Shapeshift", "🦊"),
-        ("Possess an Object", "👻"),
-        ("Call Upon Ancient Spirits", "🌟"),
-        ("Use Mystical Artifact", "🗡️")
-    ]
-    return random.sample(actions, 3)
+    return random.sample(ACTIONS, 3)
 
 def evaluate_action(action, power_level, mystical_energy, history):
     base_success_chance = (power_level + mystical_energy) / 2
-    
-    # Adjust success chance based on history
-    if action in history:
-        success_chance = base_success_chance + (history[action] * 2)
+    if action['id'] in history:
+        success_chance = base_success_chance + (history[action['id']] * 2)
     else:
         success_chance = base_success_chance
-    
     outcome = random.randint(1, 100) <= success_chance
     return outcome, success_chance
 
-def update_graph(nodes, edges, situation, action, outcome):
-    new_node_id = f"{len(nodes)}-{situation}-{action}"
-    new_node = StreamlitFlowNode(new_node_id, (0, 0), {'content': f"{situation}\n{action}\nOutcome: {'Success' if outcome else 'Failure'}"}, 'output', 'bottom', 'top')
+def update_graph(nodes, edges, situation, action, outcome, timestamp):
+    new_node_id = f"{len(nodes)}-{situation['id']}-{action['id']}"
+    content = f"{situation['emoji']} {situation['name']}\n{action['emoji']} {action['name']}\nOutcome: {'✅ Success' if outcome else '❌ Failure'}\n🕒 {timestamp}"
+    new_node = StreamlitFlowNode(new_node_id, (0, 0), {'content': content}, 'output', 'bottom', 'top')
     nodes.append(new_node)
     
     if len(nodes) > 1:
-        new_edge = StreamlitFlowEdge(f"{nodes[-2].id}-{new_node.id}", nodes[-2].id, new_node.id, animated=True)
+        new_edge = StreamlitFlowEdge(f"{nodes[-2].id}-{new_node.id}", nodes[-2].id, new_node.id, animated=True, dashed=True)
         edges.append(new_edge)
     
     return nodes, edges
 
 def save_game_state(state):
-    return json.dumps(state)
+    return json.dumps(state, default=str)  # Use default=str to handle datetime objects
 
 def load_game_state(state_json):
     return json.loads(state_json)
 
 def app():
-    st.markdown("# Kitsune - The Mystical Shape-Shifter Game 🦊")
+    st.markdown("# 🦊 Kitsune - The Mystical Shape-Shifter Game 🦊")
     
     if 'game_state' not in st.session_state:
         st.session_state.game_state = {
@@ -70,8 +122,8 @@ def app():
         }
     
     # Game Stats
-    st.sidebar.markdown("## Game Stats")
-    st.sidebar.markdown(f"Score: {st.session_state.game_state['score']}")
+    st.sidebar.markdown("## 📊 Game Stats")
+    st.sidebar.markdown(f"**Score:** {st.session_state.game_state['score']}")
     
     # Character Stats
     power_level = st.sidebar.slider('Power Level ⚡', 0, 100, st.session_state.game_state['power_level'])
@@ -81,37 +133,42 @@ def app():
     situation = generate_situation()
     actions = generate_actions()
     
-    st.markdown(f"## Current Situation: {situation}")
-    st.markdown("Choose your action:")
+    st.markdown(f"## {situation['emoji']} Current Situation: {situation['name']}")
+    st.markdown(situation['description'])
+    st.markdown("### 🎭 Choose your action:")
     
     cols = st.columns(3)
-    for i, (action, emoji) in enumerate(actions):
-        if cols[i].button(f"{emoji} {action}"):
+    for i, action in enumerate(actions):
+        if cols[i].button(f"{action['emoji']} {action['name']}"):
             outcome, success_chance = evaluate_action(action, power_level, mystical_energy, st.session_state.game_state['history'])
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
             
-            st.markdown(f"You decided to: {action}")
-            st.markdown(f"Outcome: {'Success!' if outcome else 'Failure.'}")
-            st.markdown(f"Success Chance: {success_chance:.2f}%")
+            st.markdown(f"You decided to: **{action['name']}**")
+            st.markdown(action['description'])
+            st.markdown(f"**Outcome:** {'✅ Success!' if outcome else '❌ Failure.'}")
+            st.markdown(f"**Success Chance:** {success_chance:.2f}%")
             
             if outcome:
                 st.session_state.game_state['score'] += 1
             
             # Update history
-            if action in st.session_state.game_state['history']:
-                st.session_state.game_state['history'][action] += 1 if outcome else -1
+            if action['id'] in st.session_state.game_state['history']:
+                st.session_state.game_state['history'][action['id']] += 1 if outcome else -1
             else:
-                st.session_state.game_state['history'][action] = 1 if outcome else -1
+                st.session_state.game_state['history'][action['id']] = 1 if outcome else -1
             
             st.session_state.game_state['nodes'], st.session_state.game_state['edges'] = update_graph(
                 st.session_state.game_state['nodes'], 
                 st.session_state.game_state['edges'], 
                 situation, 
                 action, 
-                outcome
+                outcome,
+                timestamp
             )
     
     # Display Graph
     if st.session_state.game_state['nodes']:
+        st.markdown("## 🌳 Your Journey")
         streamlit_flow('kitsune_game_flow', 
                        st.session_state.game_state['nodes'], 
                        st.session_state.game_state['edges'], 
@@ -120,14 +177,14 @@ def app():
                        height=600)
     
     # Character Stats Visualization
-    data = {"Stat": ["Power Level", "Mystical Energy"],
+    data = {"Stat": ["Power Level ⚡", "Mystical Energy ✨"],
             "Value": [power_level, mystical_energy]}
     df = pd.DataFrame(data)
     fig = px.bar(df, x='Stat', y='Value', title="Kitsune Stats 📊")
     st.plotly_chart(fig)
     
     # Save and Load Game State
-    st.markdown("## Save/Load Game")
+    st.markdown("## 💾 Save/Load Game")
     if st.button("Save Game"):
         game_state_json = save_game_state(st.session_state.game_state)
         st.download_button(