Update app.py

app.py

@@ -3,6 +3,7 @@ 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
@@ -13,7 +14,13 @@ CAT_RIDERS = [
     {"name": "Whiskers", "type": "Speed", "emoji": "🐾", "strength": 3, "skill": 7},
     {"name": "Fluffy", "type": "Bravery", "emoji": "🦁", "strength": 5, "skill": 5},
     {"name": "Midnight", "type": "Stealth", "emoji": "🌑", "strength": 4, "skill": 6},
-    {"name": "Bella", "type": "Charm", "emoji": "😺", "strength": 2, "skill": 8}
+    {"name": "Bella", "type": "Charm", "emoji": "😺", "strength": 2, "skill": 8},
+    {"name": "Shadow", "type": "Mystery", "emoji": "👤", "strength": 4, "skill": 6},
+    {"name": "Simba", "type": "Royalty", "emoji": "🦁", "strength": 5, "skill": 7},
+    {"name": "Luna", "type": "Magic", "emoji": "🌙", "strength": 3, "skill": 8},
+    {"name": "Leo", "type": "Courage", "emoji": "🐯", "strength": 6, "skill": 5},
+    {"name": "Milo", "type": "Playful", "emoji": "😼", "strength": 4, "skill": 7},
+    {"name": "Nala", "type": "Grace", "emoji": "🐈", "strength": 5, "skill": 6}
 ]
 
 RIDING_GEAR = [
@@ -78,6 +85,20 @@ FAILURE_CONCLUSIONS = [
     "It’s okay, every cat has a learning curve. 📚"
 ]
 
+# Function to save history to a markdown file
+def save_history_to_file(history_df, user_id, scenario_name):
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"history_{user_id}_{scenario_name}_{timestamp}.md"
+    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
+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():
     return random.choice(SITUATIONS)
@@ -86,16 +107,24 @@ def generate_actions():
     return random.sample(ACTIONS, min(3, len(ACTIONS)))
 
 def evaluate_action(situation, action, gear_strength, rider_skill, history):
-    # Determine if the action is the preferred action
-    base_success_chance = (gear_strength + rider_skill) / 2
+    # Adjusted base success chance to around 50%
+    base_success_chance = 50
+
+    # Adjust success chance based on gear strength and rider skill
+    stat_modifier = (gear_strength + rider_skill - 10) * 2  # Assuming average stats sum to 10
+    success_chance = base_success_chance + stat_modifier
+
+    # Boost success chance for preferred action
     if action['id'] == situation['preferred_action']:
-        success_chance = base_success_chance + 30  # Boost success chance for preferred action
-    else:
-        success_chance = base_success_chance
+        success_chance += 10
 
+    # Adjust based on action history
     if action['id'] in history:
         success_chance += history[action['id']] * 2
 
+    # Clamp success chance between 5% and 95%
+    success_chance = max(5, min(95, success_chance))
+
     outcome = random.randint(1, 100) <= success_chance
     return outcome, success_chance
 
@@ -123,7 +152,7 @@ def create_heterogeneous_graph(history_df):
     edges = []
     node_ids = {}
     for index, row in history_df.iterrows():
-        situation_node_id = f"situation_{row['situation_id']}_{index}"
+        situation_node_id = f"situation_{row['situation_id']}_{row['attempt']}"
         action_node_id = f"action_{row['action_id']}_{index}"
         outcome_node_id = f"outcome_{index}"
 
@@ -186,10 +215,10 @@ def create_heterogeneous_graph(history_df):
 # 📝 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)
+    grouped = history_df.groupby(['situation_name', 'attempt'], sort=False)
 
-    for situation_name, group in grouped:
-        markdown += f"### {group.iloc[0]['situation_emoji']} **{situation_name}**\n"
+    for (situation_name, attempt), group in grouped:
+        markdown += f"### {group.iloc[0]['situation_emoji']} **{situation_name}** (Attempt {attempt})\n"
         for _, row in group.iterrows():
             outcome_str = '✅ Success' if row['outcome'] else '❌ Failure'
             stars = '⭐' * int(row['score'])
@@ -206,6 +235,9 @@ def update_game_state(game_state, situation, action, outcome, timestamp):
     # Update stats based on the outcome
     game_state = update_character_stats(game_state, outcome)
 
+    # Update attempt count
+    attempt = game_state['current_attempt']
+
     # Create a new record for the history
     new_record = pd.DataFrame({
         'user_id': [game_state['user_id']],
@@ -214,6 +246,7 @@ def update_game_state(game_state, situation, action, outcome, timestamp):
         'situation_name': [situation['name']],
         'situation_emoji': [situation['emoji']],
         'situation_type': [situation['type']],
+        'attempt': [attempt],
         'action_id': [action['id']],
         'action_name': [action['name']],
         'action_emoji': [action['emoji']],
@@ -285,9 +318,22 @@ def main():
             'rider_skill': 0,
             'cat_rider': None,
             'riding_gear': None,
-            'history_df': pd.DataFrame(columns=['user_id', 'timestamp', 'situation_id', 'situation_name', 'situation_emoji', 'situation_type', 'action_id', 'action_name', 'action_emoji', 'action_type', 'outcome', 'conclusion', 'gear_strength', 'rider_skill', 'score'])
+            'history_df': pd.DataFrame(columns=['user_id', 'timestamp', 'situation_id', 'situation_name', 'situation_emoji', 'situation_type', 'attempt', 'action_id', 'action_name', 'action_emoji', 'action_type', 'outcome', 'conclusion', 'gear_strength', 'rider_skill', 'score']),
+            'current_situation': None,
+            'current_attempt': 1,
+            'actions': []
         }
 
+    # 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:")
@@ -296,6 +342,7 @@ def main():
             if cols[i].button(f"{rider['emoji']} {rider['name']} ({rider['type']})", key=f"rider_{i}"):
                 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 st.session_state.game_state['riding_gear'] is None and st.session_state.game_state['cat_rider'] is not None:
@@ -307,18 +354,30 @@ def main():
                 st.session_state.game_state['gear_strength'] = gear['strength']
 
     # 🎭 Game Loop
-    if st.session_state.game_state['cat_rider'] is not None and st.session_state.game_state['riding_gear'] is not None:
-        situation = generate_situation()
-        actions = generate_actions()
+    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']:
+            # Generate a new situation
+            game_state['current_situation'] = generate_situation()
+            game_state['current_attempt'] = 1
+            game_state['succeeded'] = False
+
+        situation = game_state['current_situation']
         st.markdown(f"## {situation['emoji']} Current Situation: {situation['name']} ({situation['type']})")
         st.markdown(situation['description'])
         st.markdown("### 🎭 Choose your action:")
 
+        # Generate actions if not already generated
+        if not game_state['actions']:
+            game_state['actions'] = generate_actions()
+
         cols = st.columns(3)
-        for i, action in enumerate(actions):
-            if cols[i].button(f"{action['emoji']} {action['name']} ({action['type']})", key=f"action_{i}"):
-                outcome, success_chance = evaluate_action(situation, action, st.session_state.game_state['gear_strength'], st.session_state.game_state['rider_skill'], st.session_state.game_state['history'])
+        action_chosen = False
+        for i, action in enumerate(game_state['actions']):
+            if cols[i].button(f"{action['emoji']} {action['name']} ({action['type']})", key=f"action_{i}_{game_state['current_attempt']}"):
+                outcome, success_chance = evaluate_action(situation, action, game_state['gear_strength'], game_state['rider_skill'], game_state['history'])
                 timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
                 st.markdown(f"You decided to: **{action['name']}** ({action['type']})")
@@ -327,11 +386,18 @@ def main():
                 st.markdown(f"**Success Chance:** {success_chance:.2f}%")
 
                 if outcome:
-                    st.session_state.game_state['score'] += 1
+                    game_state['score'] += 1
+                    game_state['succeeded'] = True
+                    # Clear actions for next situation
+                    game_state['actions'] = []
+                else:
+                    game_state['current_attempt'] += 1
+                    # Generate new actions for retry
+                    game_state['actions'] = generate_actions()
 
                 # 🔄 Update game state
-                st.session_state.game_state = update_game_state(
-                    st.session_state.game_state,
+                game_state = update_game_state(
+                    game_state,
                     situation,
                     action,
                     outcome,
@@ -339,42 +405,65 @@ def main():
                 )
 
                 # Display conclusion
-                conclusion = st.session_state.game_state['history_df'].iloc[-1]['conclusion']
+                conclusion = game_state['history_df'].iloc[-1]['conclusion']
                 st.markdown(f"**Encounter Conclusion:** {conclusion}")
 
                 # Display updated stats
                 st.markdown(f"**Updated Stats:**")
-                st.markdown(f"💪 Gear Strength: {st.session_state.game_state['gear_strength']:.2f}")
-                st.markdown(f"🏋️ Rider Skill: {st.session_state.game_state['rider_skill']:.2f}")
+                st.markdown(f"💪 Gear Strength: {game_state['gear_strength']:.2f}")
+                st.markdown(f"🏋️ Rider Skill: {game_state['rider_skill']:.2f}")
 
                 # 🏅 Display Scoreboard
-                display_scoreboard(st.session_state.game_state)
-
-                # Integration point for both functions
-                if not st.session_state.game_state['history_df'].empty:
-                    # 📝 Display Markdown Preview
-                    st.markdown(create_markdown_preview(st.session_state.game_state['history_df']))
-
-                    # 🌳 Display Heterogeneous Journey Graph
-                    st.markdown("## 🌳 Your Journey (Heterogeneous Graph)")
-                    nodes, edges = create_heterogeneous_graph(st.session_state.game_state['history_df'])
-                    try:
-                        streamlit_flow('cat_rider_flow',
-                                       nodes,
-                                       edges,
-                                       layout=TreeLayout(direction='down'),
-                                       fit_view=True,
-                                       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.")
-
-                # 📊 Character Stats Visualization
-                data = {"Stat": ["Gear Strength 🛡️", "Rider Skill 🏇"],
-                        "Value": [st.session_state.game_state['gear_strength'], st.session_state.game_state['rider_skill']]}
-                df = pd.DataFrame(data)
-                fig = px.bar(df, x='Stat', y='Value', title="Cat Rider Stats 📊")
-                st.plotly_chart(fig)
+                display_scoreboard(game_state)
+
+                action_chosen = True
+                break  # Exit the loop after action is taken
+
+        # If no action was chosen, show a message
+        if not action_chosen:
+            st.markdown("Please choose an action to proceed.")
+
+        # Integration point for both functions
+        if not game_state['history_df'].empty:
+            # 📝 Display Markdown Preview
+            st.markdown(create_markdown_preview(game_state['history_df']))
+
+            # 🌳 Display Heterogeneous Journey Graph
+            st.markdown("## 🌳 Your Journey (Heterogeneous Graph)")
+            nodes, edges = create_heterogeneous_graph(game_state['history_df'])
+            try:
+                streamlit_flow('cat_rider_flow',
+                               nodes,
+                               edges,
+                               layout=TreeLayout(direction='down'),
+                               fit_view=True,
+                               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.")
+
+            # 📊 Character Stats Visualization
+            data = {"Stat": ["Gear Strength 🛡️", "Rider Skill 🏇"],
+                    "Value": [game_state['gear_strength'], game_state['rider_skill']]}
+            df = pd.DataFrame(data)
+            fig = px.bar(df, x='Stat', y='Value', title="Cat Rider Stats 📊")
+            st.plotly_chart(fig)
+
+            # Save history to file
+            if st.button("💾 Save Game History"):
+                filename = save_history_to_file(game_state['history_df'], game_state['user_id'], situation['name'])
+                st.success(f"History saved to {filename}")
+
+            # Provide download button for saved history
+            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()