Update app.py

app.py

@@ -1,15 +1,3 @@
-# 1. Configuration
-import streamlit as st
-import pandas as pd
-import plotly.express as px
-import random
-import uuid
-import os
-import re
-from datetime import datetime
-from streamlit_flow import streamlit_flow
-from streamlit_flow.elements import StreamlitFlowNode, StreamlitFlowEdge
-from streamlit_flow.layouts import RadialLayout
 
 # Set page configuration to wide mode
 Site_Name = '🦁CatRider🐈'
@@ -32,6 +20,18 @@ if useConfig:
         }
     )
 
+
+import streamlit as st
+import pandas as pd
+import plotly.express as px
+import random
+import uuid
+import os
+from datetime import datetime
+from streamlit_flow import streamlit_flow
+from streamlit_flow.elements import StreamlitFlowNode, StreamlitFlowEdge
+from streamlit_flow.layouts import TreeLayout
+
 # 🐱 Cat Rider and Gear Data
 CAT_RIDERS = [
     {"name": "Whiskers", "type": "Speed", "emoji": "🐾", "strength": 3, "skill": 7},
@@ -55,30 +55,30 @@ RIDING_GEAR = [
 
 # 🌍 Game World Data (Expanded to 10 Situations)
 SITUATIONS = [
-    {"id": "feline_escape", "name": "The Great Feline Escape", "description": "Your cat rider is trapped in an old mansion...", "emoji": "🚪", "type": "escape", "preferred_action": "agility"},
-    {"id": "lost_temple", "name": "The Treasure of the Lost Temple", "description": "On a quest to retrieve an ancient artifact...", "emoji": "🏛️", "type": "exploration", "preferred_action": "resourcefulness"},
-    {"id": "royal_tournament", "name": "The Royal Tournament", "description": "Compete in a grand tournament...", "emoji": "👑", "type": "competition", "preferred_action": "bravery"},
-    {"id": "sky_race", "name": "The Sky Race", "description": "Compete in the annual Sky Race...", "emoji": "☁️", "type": "competition", "preferred_action": "agility"},
-    {"id": "cheese_heist", "name": "The Great Cheese Heist", "description": "Your cat rider must sneak into the royal pantry...", "emoji": "🧀", "type": "heist", "preferred_action": "stealth"},
-    {"id": "pirate_cove", "name": "The Pirate Cove", "description": "Sail the high seas with your trusty crew...", "emoji": "🏴‍☠️", "type": "exploration", "preferred_action": "bravery"},
-    {"id": "feline_moon_mission", "name": "The Feline Moon Mission", "description": "Blast off into space...", "emoji": "🌕", "type": "exploration", "preferred_action": "resourcefulness"},
-    {"id": "purr_summit", "name": "The Purr Summit", "description": "Join a secret gathering of the most intellectual cats...", "emoji": "📜", "type": "debate", "preferred_action": "insight"},
-    {"id": "feline_invasion", "name": "The Feline Invasion", "description": "Aliens have invaded Earth...", "emoji": "👽", "type": "battle", "preferred_action": "strategy"},
-    {"id": "eternal_catnap", "name": "The Eternal Catnap", "description": "You've entered a sacred temple...", "emoji": "💤", "type": "exploration", "preferred_action": "stealth"}
+    {"id": "feline_escape", "name": "The Great Feline Escape", "description": "Your cat rider is trapped in an old mansion, which is about to be demolished. Using agility, wit, and bravery, orchestrate the perfect escape before the walls crumble! 🏚️", "emoji": "🚪", "type": "escape", "preferred_action": "agility"},
+    {"id": "lost_temple", "name": "The Treasure of the Lost Temple", "description": "On a quest to retrieve an ancient artifact, your cat rider must navigate through a labyrinth filled with traps and guardian spirits. Don't let the spooky ghosts get you! 👻", "emoji": "🏛️", "type": "exploration", "preferred_action": "resourcefulness"},
+    {"id": "royal_tournament", "name": "The Royal Tournament", "description": "Compete in a grand tournament where the finest cat riders showcase their skills and bravery to earn the title of the Royal Rider. Prepare for a showdown with noble feline adversaries! 🐱", "emoji": "👑", "type": "competition", "preferred_action": "bravery"},
+    {"id": "sky_race", "name": "The Sky Race", "description": "Compete in the annual Sky Race, where your cat rider flies across the skies on a magical broomstick! Watch out for lightning storms and mischievous crows! 🌩️", "emoji": "☁️", "type": "competition", "preferred_action": "agility"},
+    {"id": "cheese_heist", "name": "The Great Cheese Heist", "description": "Your cat rider must sneak into the royal pantry to steal the legendary Cheese of Destiny. Beware – the palace mice are guarding it! 🧀", "emoji": "🧀", "type": "heist", "preferred_action": "stealth"},
+    {"id": "pirate_cove", "name": "The Pirate Cove", "description": "Sail the high seas with your trusty crew of cats and uncover the secrets of Pirate Cove. Beware of the treacherous Sea Dogs! 🏴‍☠️", "emoji": "🏴‍☠️", "type": "exploration", "preferred_action": "bravery"},
+    {"id": "feline_moon_mission", "name": "The Feline Moon Mission", "description": "Blast off into space! Your mission is to plant the flag of Catopia on the moon. Pilot your rocket through an asteroid field! 🚀", "emoji": "🌕", "type": "exploration", "preferred_action": "resourcefulness"},
+    {"id": "purr_summit", "name": "The Purr Summit", "description": "Join a secret gathering of the most intellectual cats in the world. Engage in a battle of wits and wisdom to become the Grand Purr! 🧠", "emoji": "📜", "type": "debate", "preferred_action": "insight"},
+    {"id": "feline_invasion", "name": "The Feline Invasion", "description": "Aliens have invaded Earth and only your cat rider can save the planet! Use cunning strategies to repel the invaders! 👽", "emoji": "👽", "type": "battle", "preferred_action": "strategy"},
+    {"id": "eternal_catnap", "name": "The Eternal Catnap", "description": "You've entered a sacred temple where cats nap for centuries. Can you navigate the dream world and escape before you too are lulled into eternal slumber? 🛏️", "emoji": "💤", "type": "exploration", "preferred_action": "stealth"}
 ]
 
 # 🧠 Expanded Actions (10 Actions)
 ACTIONS = [
-    {"id": "stealth", "name": "Use Stealth", "description": "Sneak past obstacles...", "emoji": "🤫", "type": "skill"},
-    {"id": "agility", "name": "Showcase Agility", "description": "Perform impressive acrobatic maneuvers...", "emoji": "🏃", "type": "physical"},
-    {"id": "charm", "name": "Charm Others", "description": "Use your cat's natural charisma...", "emoji": "😻", "type": "social"},
-    {"id": "resourcefulness", "name": "Be Resourceful", "description": "Utilize the environment or items...", "emoji": "🧠", "type": "mental"},
-    {"id": "bravery", "name": "Show Bravery", "description": "Face dangers head-on...", "emoji": "🦁", "type": "physical"},
-    {"id": "strategy", "name": "Develop a Strategy", "description": "Use tactical thinking...", "emoji": "🧠", "type": "mental"},
-    {"id": "speed", "name": "Sprint Away", "description": "Run faster than you've ever run...", "emoji": "🏃‍♀️", "type": "physical"},
-    {"id": "insight", "name": "Use Insight", "description": "Tap into ancient feline wisdom...", "emoji": "🔮", "type": "mental"},
-    {"id": "distraction", "name": "Create a Distraction", "description": "Use cunning tricks...", "emoji": "🪄", "type": "mental"},
-    {"id": "negotiation", "name": "Negotiate", "description": "Use diplomacy and clever negotiation...", "emoji": "💼", "type": "social"}
+    {"id": "stealth", "name": "Use Stealth", "description": "Sneak past obstacles or enemies without being detected. You're like a ninja in the shadows! 🐾", "emoji": "🤫", "type": "skill"},
+    {"id": "agility", "name": "Showcase Agility", "description": "Perform impressive acrobatic maneuvers to overcome challenges. Cats always land on their feet, right? 🏃", "emoji": "🏃", "type": "physical"},
+    {"id": "charm", "name": "Charm Others", "description": "Use your cat's natural charisma to win over allies or distract foes. Who could resist those cute eyes? 😻", "emoji": "😻", "type": "social"},
+    {"id": "resourcefulness", "name": "Be Resourceful", "description": "Utilize the environment or items in creative ways to solve problems. Think on your paws! 🧠", "emoji": "🧠", "type": "mental"},
+    {"id": "bravery", "name": "Show Bravery", "description": "Face dangers head-on with your feline courage. Not all heroes wear capes – some wear fur! 🦁", "emoji": "🦁", "type": "physical"},
+    {"id": "strategy", "name": "Develop a Strategy", "description": "Use tactical thinking to outsmart opponents and gain the upper hand in battle. Brains over brawn! 🧠", "emoji": "🧠", "type": "mental"},
+    {"id": "speed", "name": "Sprint Away", "description": "Run faster than you've ever run before to escape danger. Like a cat fleeing a vacuum cleaner! 🏃‍♀️", "emoji": "🏃‍♀️", "type": "physical"},
+    {"id": "insight", "name": "Use Insight", "description": "Tap into ancient feline wisdom to solve puzzles and mysteries. A cat always knows! 🔮", "emoji": "🔮", "type": "mental"},
+    {"id": "distraction", "name": "Create a Distraction", "description": "Use cunning tricks and diversions to draw attention away from your real goal. Look over there! 🪄", "emoji": "🪄", "type": "mental"},
+    {"id": "negotiation", "name": "Negotiate", "description": "Use diplomacy and clever negotiation to get out of a tight spot. Every cat has their price! 💼", "emoji": "💼", "type": "social"}
 ]
 
 # Expanded conclusions for outcomes - 10 items each for success and failure
@@ -112,55 +112,15 @@ FAILURE_CONCLUSIONS = [
 def save_history_to_file(history_df, user_id):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     filename = f"history_{user_id}_{timestamp}.md"
-    markdown, _, _ = process_journey_history(history_df)
+    markdown = create_markdown_preview(history_df)
     with open(filename, 'w', encoding='utf-8') as f:
         f.write(markdown)
     return filename
 
-# Function to load saved histories and reconstruct game state
-def load_saved_histories(user_id):
-    history_files = [f for f in os.listdir('.') if f.startswith(f'history_{user_id}_') and f.endswith('.md')]
-    history_files.sort()  # Ensure files are sorted by timestamp
-    combined_history = ''
-    for history_file in history_files:
-        with open(history_file, 'r', encoding='utf-8') as f:
-            combined_history += f.read() + '\n'
-    return combined_history
-
-# Function to parse markdown history and reconstruct history DataFrame
-def parse_history_markdown(markdown_text):
-    lines = markdown_text.split('\n')
-    data = []
-    current_situation = None
-    for line in lines:
-        situation_match = re.match(r"### (\S+) \*\*(.+)\*\*", line)
-        if situation_match:
-            situation_emoji = situation_match.group(1)
-            situation_name = situation_match.group(2)
-            continue
-        attempt_match = re.match(r"Attempt (\d+): (\S+) \*\*(.+)\*\*: (✅ Success|❌ Failure) (\S*)", line)
-        if attempt_match:
-            attempt = int(attempt_match.group(1))
-            action_emoji = attempt_match.group(2)
-            action_name = attempt_match.group(3)
-            outcome_str = attempt_match.group(4)
-            stars = attempt_match.group(5)
-            outcome = True if outcome_str == '✅ Success' else False
-            score = len(stars)
-            conclusion_line_index = lines.index(line) + 1
-            conclusion = lines[conclusion_line_index] if conclusion_line_index < len(lines) else ''
-            data.append({
-                'situation_name': situation_name,
-                'situation_emoji': situation_emoji,
-                'attempt': attempt,
-                'action_name': action_name,
-                'action_emoji': action_emoji,
-                'outcome': outcome,
-                'score': score,
-                'conclusion': conclusion
-            })
-    history_df = pd.DataFrame(data)
-    return history_df
+# Function to load saved histories
+def load_saved_histories():
+    history_files = [f for f in os.listdir('.') if f.startswith('history_') and f.endswith('.md')]
+    return history_files
 
 # 🧠 Game Mechanics
 def generate_situation():
@@ -209,95 +169,63 @@ def update_character_stats(game_state, outcome):
         game_state['gear_strength'] = max(1, game_state['gear_strength'] - 0.1)
     return game_state
 
-# 🌳 Process Journey History to Create Markdown and Graph Data
-def process_journey_history(history_df):
-    markdown = "## 🌳 Journey Preview\n\n"
+# 🌳 Journey Visualization
+def create_knowledge_graph(history_df):
     nodes = []
     edges = []
     node_ids = {}
-    score = 0  # To keep track of the score for success nodes
-
-    grouped = history_df.groupby(['situation_name'], sort=False)
+    for index, row in history_df.iterrows():
+        situation_node_id = f"situation_{row['situation_id']}"
+        action_node_id = f"action_{index}"
 
-    # Main node to connect all situations
-    main_node_id = "main"
-    main_node = StreamlitFlowNode(
-        main_node_id,
-        pos=(0, 0),
-        data={'content': "# Your Journey"},
-        type='input',
-        target_position='bottom',
-        width=200
-    )
-    nodes.append(main_node)
-
-    for situation_name, group in grouped:
-        situation_emoji = group.iloc[0]['situation_emoji']
-        situation_id = group.iloc[0].get('situation_id', str(uuid.uuid4()))
-        situation_node_id = f"situation_{situation_id}"
-
-        markdown += f"### {situation_emoji} **{situation_name}**\n"
-
-        # Create situation node if not already created
+        # Situation node
         if situation_node_id not in node_ids:
             situation_node = StreamlitFlowNode(
                 situation_node_id,
                 pos=(0, 0),
-                data={'content': f"### {situation_emoji} {situation_name}"},
+                data={'content': f"{row['situation_emoji']} {row['situation_name']}"},
                 type='default',
-                target_position='top',
-                source_position='bottom',
-                width=200
+                shape='ellipse'
             )
             nodes.append(situation_node)
             node_ids[situation_node_id] = situation_node_id
 
-            # Edge from main node to situation node
-            edges.append(StreamlitFlowEdge(
-                id=f"edge_{main_node_id}_{situation_node_id}",
-                source=main_node_id,
-                target=situation_node_id,
-                animated=True
-            ))
-
-        for idx, row in group.iterrows():
-            attempt = row['attempt']
-            action_emoji = row['action_emoji']
-            action_name = row['action_name']
-            outcome = row['outcome']
-            outcome_str = '✅ Success' if outcome else '❌ Failure'
-            stars = '⭐' * int(row.get('score', 0)) if outcome else ''
-            conclusion = row.get('conclusion', '')
-
-            markdown += f"Attempt {attempt}: {action_emoji} **{action_name}**: {outcome_str} {stars}\n"
-            markdown += f"{conclusion}\n"
-
-            # Create attempt node
-            attempt_node_id = f"attempt_{situation_id}_{attempt}_{idx}"
-            attempt_content = f"Attempt {attempt}: {action_emoji} {action_name}\n{outcome_str} {stars}\n{conclusion}"
-
-            attempt_node = StreamlitFlowNode(
-                attempt_node_id,
-                pos=(0, 0),
-                data={'content': attempt_content},
-                type='output',
-                target_position='top',
-                source_position='bottom',
-                width=250
-            )
-            nodes.append(attempt_node)
-
-            # Edge from situation to attempt
-            edges.append(StreamlitFlowEdge(
-                id=f"edge_{situation_node_id}_{attempt_node_id}",
-                source=situation_node_id,
-                target=attempt_node_id,
-                animated=True
-            ))
+        # Attempt node (action and outcome)
+        outcome_content = '✅ Success' if row['outcome'] else '❌ Failure'
+        stars = '⭐' * int(row['score'])
+        attempt_node = StreamlitFlowNode(
+            action_node_id,
+            pos=(0, 0),
+            data={'content': f"{row['action_emoji']} {row['action_name']} ({outcome_content})\n{row['conclusion']}\n{stars}"},
+            type='default',
+            shape='rectangle'
+        )
+        nodes.append(attempt_node)
+
+        # Edge from situation to attempt
+        edges.append(StreamlitFlowEdge(
+            id=f"edge_{situation_node_id}_{action_node_id}",
+            source=situation_node_id,
+            target=action_node_id,
+            animated=True
+        ))
+
+    return nodes, edges
+
+# 📝 Markdown Preview with Subpoints for Each Action
+def create_markdown_preview(history_df):
+    markdown = "## 🌳 Journey Preview\n\n"
+    grouped = history_df.groupby(['situation_name'], sort=False)
 
+    for situation_name, group in grouped:
+        markdown += f"### {group.iloc[0]['situation_emoji']} **{situation_name}**\n"
+        for _, row in group.iterrows():
+            outcome_str = '✅ Success' if row['outcome'] else '❌ Failure'
+            stars = '⭐' * int(row['score'])
+            markdown += f"- Attempt {row['attempt']}: {row['action_emoji']} **{row['action_name']}**: {outcome_str} {stars}\n"
+            markdown += f"  - {row['conclusion']}\n"
         markdown += "\n"
-
-    return markdown, nodes, edges
+    return markdown
 
 # 🔄 Update game state with the result of the action
 def update_game_state(game_state, situation, action, outcome, timestamp):
@@ -310,10 +238,6 @@ def update_game_state(game_state, situation, action, outcome, timestamp):
     # Update attempt count
     attempt = game_state['current_attempt']
 
-    # Update score
-    if outcome:
-        game_state['score'] += 1
-
     # Create a new record for the history
     new_record = pd.DataFrame({
         'user_id': [game_state['user_id']],
@@ -343,9 +267,6 @@ def update_game_state(game_state, situation, action, outcome, timestamp):
     else:
         game_state['history'][action['id']] = 1 if outcome else -1
 
-    # Automatically save history after each action
-    save_history_to_file(game_state['history_df'], game_state['user_id'])
-
     return game_state
 
 # 🏅 Display Scoreboard with Star Emojis and Buckyball Outline
@@ -390,7 +311,7 @@ def main():
     # 🏁 Initialize game state
     if 'game_state' not in st.session_state:
         st.session_state.game_state = {
-            'user_id': None,
+            'user_id': str(uuid.uuid4()),
             'score': 0,
             'history': {},
             'gear_strength': 0,
@@ -404,38 +325,38 @@ def main():
             'succeeded': False
         }
 
-    game_state = st.session_state.game_state
-
-    # 🐱 Cat Rider Selection or Loading Previous State
-    if game_state['cat_rider'] is None:
-        st.markdown("## Choose Your Cat Rider or Load Previous Journey:")
-        # Check for existing histories
-        existing_riders = [f.split('_')[1] for f in os.listdir('.') if f.startswith('history_')]
-        existing_riders = list(set(existing_riders))
+    # Sidebar functionality
+    st.sidebar.header("📂 Saved Histories")
+    history_files = load_saved_histories()
+    if history_files:
+        selected_history = st.sidebar.selectbox("Select a history to load:", history_files)
+        if st.sidebar.button("Load Selected History"):
+            with open(selected_history, 'r', encoding='utf-8') as f:
+                st.session_state.game_state['loaded_history'] = f.read()
+            st.sidebar.success(f"Loaded {selected_history}")
+
+    # 🐱 Cat Rider Selection
+    if st.session_state.game_state['cat_rider'] is None:
+        st.markdown("## Choose Your Cat Rider:")
         cols = st.columns(len(CAT_RIDERS))
         for i, rider in enumerate(CAT_RIDERS):
             if cols[i].button(f"{rider['emoji']} {rider['name']} ({rider['type']})", key=f"rider_{i}"):
-                game_state['cat_rider'] = rider
-                game_state['rider_skill'] = rider['skill']
-                game_state['user_id'] = rider['name']  # Use rider name as user ID
-
-                # Load existing history if available
-                existing_history = load_saved_histories(game_state['user_id'])
-                if existing_history:
-                    history_df = parse_history_markdown(existing_history)
-                    game_state['history_df'] = history_df
-                    game_state['score'] = history_df['score'].max() if not history_df.empty else 0
+                st.session_state.game_state['cat_rider'] = rider
+                st.session_state.game_state['rider_skill'] = rider['skill']
+                st.session_state.game_state['user_id'] = rider['name']  # Use rider name as user ID
 
     # 🏇 Riding Gear Selection
-    if game_state['riding_gear'] is None and game_state['cat_rider'] is not None:
+    if st.session_state.game_state['riding_gear'] is None and st.session_state.game_state['cat_rider'] is not None:
         st.markdown("## Select Your Riding Gear:")
         cols = st.columns(len(RIDING_GEAR))
         for i, gear in enumerate(RIDING_GEAR):
             if cols[i].button(f"{gear['name']} ({gear['type']})", key=f"gear_{i}"):
-                game_state['riding_gear'] = gear
-                game_state['gear_strength'] = gear['strength']
+                st.session_state.game_state['riding_gear'] = gear
+                st.session_state.game_state['gear_strength'] = gear['strength']
 
     # 🎭 Game Loop
+    game_state = st.session_state.game_state
+
     if game_state['cat_rider'] is not None and game_state['riding_gear'] is not None:
         # Check if current_situation is None or if the player succeeded in the previous situation
         if game_state['current_situation'] is None or game_state['succeeded']:
@@ -468,6 +389,7 @@ def main():
                 st.markdown(f"**Success Chance:** {success_chance:.2f}%")
 
                 if outcome:
+                    game_state['score'] += 1
                     game_state['succeeded'] = True
                     # Clear actions for next situation
                     game_state['actions'] = []
@@ -506,21 +428,19 @@ def main():
 
         # Integration point for both functions
         if not game_state['history_df'].empty:
-            # 📝 Process Journey History to get markdown and graph data
-            markdown_preview, nodes, edges = process_journey_history(game_state['history_df'])
-
             # 📝 Display Markdown Preview
-            st.markdown(markdown_preview)
+            st.markdown(create_markdown_preview(game_state['history_df']))
 
             # 🌳 Display Knowledge Journey Graph
             st.markdown("## 🌳 Your Journey (Knowledge Graph)")
+            nodes, edges = create_knowledge_graph(game_state['history_df'])
             try:
                 streamlit_flow('cat_rider_flow',
                                nodes,
                                edges,
-                               layout=RadialLayout(),
+                               layout=TreeLayout(direction='down'),
                                fit_view=True,
-                               height=1000)
+                               height=600)
             except Exception as e:
                 st.error(f"An error occurred while rendering the journey graph: {str(e)}")
                 st.markdown("Please try refreshing the page if the graph doesn't appear.")
@@ -532,12 +452,24 @@ def main():
             fig = px.bar(df, x='Stat', y='Value', title="Cat Rider Stats 📊")
             st.plotly_chart(fig)
 
-    # Automatically load saved history if available
-    if game_state['user_id'] and not game_state['history_df'].empty:
-        existing_history = load_saved_histories(game_state['user_id'])
-        if existing_history:
-            st.markdown("## 📂 Loaded Journey History")
-            st.markdown(existing_history)
+            # Save history to file
+            if st.button("💾 Save Game History"):
+                filename = save_history_to_file(game_state['history_df'], game_state['user_id'])
+                st.success(f"History saved to {filename}")
+                # Update history files list
+                history_files = load_saved_histories()
+
+            # Provide download buttons for saved histories
+            st.sidebar.markdown("### 📥 Download Histories")
+            for history_file in history_files:
+                with open(history_file, 'r', encoding='utf-8') as f:
+                    st.sidebar.download_button(
+                        label=f"Download {history_file}",
+                        data=f.read(),
+                        file_name=history_file,
+                        mime='text/markdown'
+                    )
 
 if __name__ == "__main__":
     main()
+