Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 import streamlit as st
 import pandas as pd
 import plotly.express as px
@@ -32,7 +30,6 @@ if useConfig:
         }
     )
 
-
 # 🐱 Cat Rider and Gear Data
 CAT_RIDERS = [
     {"name": "Whiskers", "type": "Speed", "emoji": "🐾", "strength": 3, "skill": 7},
@@ -82,9 +79,32 @@ ACTIONS = [
     {"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
+SUCCESS_CONCLUSIONS = [
+    "Your swift paws led you to victory! 🎉",
+    "You pounced at the perfect moment! 🏆",
+    "The stars aligned for your cat rider! 🌟",
+    "You navigated the challenge like a true feline champion! 🐱",
+    "Victory is sweet, just like a bowl of fresh milk! 🥛",
+    "Your opponents are left in awe of your skills! 😺",
+    "You’ve earned the title of Cat Commander! 🏅",
+    "All the other cats are jealous of your agility! 🏃‍♂️",
+    "Your strategy was flawless, and the victory is yours! 🎖️",
+    "Your cat rider is now a legend in the feline world! 👑"
+]
 
-SUCCESS_CONCLUSIONS = ["Your swift paws led you to victory! 🎉"]
-FAILURE_CONCLUSIONS = ["You tried your best, but it just wasn’t enough. 😿"]
+FAILURE_CONCLUSIONS = [
+    "You tried your best, but it just wasn’t enough. 😿",
+    "Maybe next time, kitty. Keep your tail up! 🐾",
+    "That didn’t go as planned. Time for a catnap to recover! 💤",
+    "Even the best cats have their off days. 😔",
+    "The challenge was too great this time. Better luck next time! 🍀",
+    "You might need more than nine lives to get through this. 🐈",
+    "The enemy was too clever for your plan. 🧠",
+    "You tripped over your own paws! 🐾",
+    "The cat gods were not in your favor today. 🙀",
+    "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):
@@ -100,50 +120,54 @@ def load_saved_histories():
     history_files = [f for f in os.listdir('.') if f.startswith('history_') and f.endswith('.md')]
     return history_files
 
-# Generate a random situation
+# 🧠 Game Mechanics
 def generate_situation():
     return random.choice(SITUATIONS)
 
-# Generate random actions
 def generate_actions():
     return random.sample(ACTIONS, min(3, len(ACTIONS)))
 
-# Evaluate the chosen action
 def evaluate_action(situation, action, gear_strength, rider_skill, history):
+    # Adjusted base success chance to around 50%
     base_success_chance = 50
-    stat_modifier = (gear_strength + rider_skill - 10) * 2
+
+    # 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 += 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
 
-# Update character stats based on outcome
+def generate_encounter_conclusion(situation, action, outcome):
+    if outcome:
+        return random.choice(SUCCESS_CONCLUSIONS)
+    else:
+        return random.choice(FAILURE_CONCLUSIONS)
+
+# 🔄 Update character stats based on outcome
 def update_character_stats(game_state, outcome):
     if outcome:
+        # Increase stats on success
         game_state['rider_skill'] += 0.5
         game_state['gear_strength'] += 0.2
     else:
+        # Decrease stats on failure, but not below 1
         game_state['rider_skill'] = max(1, game_state['rider_skill'] - 0.3)
         game_state['gear_strength'] = max(1, game_state['gear_strength'] - 0.1)
     return game_state
 
-# Create markdown preview of history
-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
-
-# Create knowledge graph with markdown support in nodes
+# 🌳 Journey Visualization
 def create_knowledge_graph(history_df):
     nodes = []
     edges = []
@@ -152,38 +176,27 @@ def create_knowledge_graph(history_df):
         situation_node_id = f"situation_{row['situation_id']}"
         action_node_id = f"action_{index}"
 
-        # Situation node with markdown content
+        # Situation node
         if situation_node_id not in node_ids:
             situation_node = StreamlitFlowNode(
-                id=situation_node_id,
+                situation_node_id,
                 pos=(0, 0),
-                data={'content': f"### {row['situation_emoji']} {row['situation_name']}\n\n{row['description']}"},
+                data={'content': f"{row['situation_emoji']} {row['situation_name']}"},
                 type='default',
-                shape='ellipse',
-                width=300
+                shape='ellipse'
             )
             nodes.append(situation_node)
             node_ids[situation_node_id] = situation_node_id
 
-        # Attempt node (action and outcome) with markdown content
-        outcome_content = '✅ **Success**' if row['outcome'] else '❌ **Failure**'
+        # Attempt node (action and outcome)
+        outcome_content = '✅ Success' if row['outcome'] else '❌ Failure'
         stars = '⭐' * int(row['score'])
-        attempt_node_content = f"""
-#### {row['action_emoji']} {row['action_name']} ({outcome_content})
-
-**Conclusion:** {row['conclusion']}
-
-**Score:** {stars}
-
-**Gear Strength:** {row['gear_strength']:.2f} | **Rider Skill:** {row['rider_skill']:.2f}
-"""
         attempt_node = StreamlitFlowNode(
-            id=action_node_id,
+            action_node_id,
             pos=(0, 0),
-            data={'content': attempt_node_content},
+            data={'content': f"{row['action_emoji']} {row['action_name']} ({outcome_content})\n{row['conclusion']}\n{stars}"},
             type='default',
-            shape='rectangle',
-            width=300
+            shape='rectangle'
         )
         nodes.append(attempt_node)
 
@@ -197,11 +210,103 @@ def create_knowledge_graph(history_df):
 
     return nodes, edges
 
-# Main game application
+# 📝 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
+
+# 🔄 Update game state with the result of the action
+def update_game_state(game_state, situation, action, outcome, timestamp):
+    # Generate the encounter conclusion (success or failure)
+    conclusion = generate_encounter_conclusion(situation, action, outcome)
+
+    # 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']],
+        'timestamp': [timestamp],
+        'situation_id': [situation['id']],
+        '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']],
+        'action_type': [action['type']],
+        'outcome': [outcome],
+        'conclusion': [conclusion],
+        'gear_strength': [game_state['gear_strength']],
+        'rider_skill': [game_state['rider_skill']],
+        'score': [game_state['score']]
+    })
+
+    # Add the new record to the game history DataFrame
+    game_state['history_df'] = pd.concat([game_state['history_df'], new_record], ignore_index=True)
+
+    # Update the history of actions (tracking how many times each action was used)
+    if action['id'] in game_state['history']:
+        game_state['history'][action['id']] += 1 if outcome else -1
+    else:
+        game_state['history'][action['id']] = 1 if outcome else -1
+
+    return game_state
+
+# 🏅 Display Scoreboard with Star Emojis and Buckyball Outline
+def display_scoreboard(game_state):
+    # Calculate number of star emojis based on score
+    score = game_state['score']
+    stars = '⭐' * int(score)
+
+    # Create buckyball style outline (simplified)
+    outline = ''
+    if score > 0:
+        outline = '''
+        ⬡ ⬡ ⬡ ⬡ ⬡
+         ⬡ ⬡ ⬡ ⬡
+        ⬡ ⬡ ⬡ ⬡ ⬡
+         ⬡ ⬡ ⬡ ⬡
+        ⬡ ⬡ ⬡ ⬡ ⬡
+        '''
+    else:
+        outline = 'No successes yet.'
+
+    st.markdown("## 🏅 Scoreboard")
+    st.markdown(f"**Score:** {stars} ({score})")
+    st.markdown(outline)
+
+# 🎮 Main Game Application
 def main():
     st.title("🐱 Cat Rider 🏇")
 
-    # Initialize game state
+    # 📜 Game Rules
+    st.markdown("""
+    ### 📜 Game Rules
+    | Step | Description |
+    |------|-------------|
+    | 1️⃣  | Choose your Cat Rider |
+    | 2️⃣  | Select your Riding Gear |
+    | 3️⃣  | Set off on an Adventure |
+    | 4️⃣  | Encounter Challenges and Make Decisions |
+    | 5️⃣  | Complete the Quest and Grow Stronger |
+    """)
+
+    # 🏁 Initialize game state
     if 'game_state' not in st.session_state:
         st.session_state.game_state = {
             'user_id': str(uuid.uuid4()),
@@ -218,20 +323,150 @@ def main():
             'succeeded': False
         }
 
-    # Game logic...
-
+    # 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}"):
+                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:
+        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}"):
+                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
 
-    # 🌳 Display Knowledge Journey Graph
-    if not game_state['history_df'].empty:
-        nodes, edges = create_knowledge_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.")
-
-    # Rest of the game logic...
+    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
+            # Clear actions for new situation
+            game_state['actions'] = []
+
+        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)
+        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']})")
+                st.markdown(action['description'])
+                st.markdown(f"**Outcome:** {'✅ Success!' if outcome else '❌ Failure.'}")
+                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'] = []
+                else:
+                    game_state['current_attempt'] += 1
+                    # Generate new actions for retry
+                    game_state['actions'] = generate_actions()
+
+                # 🔄 Update game state
+                game_state = update_game_state(
+                    game_state,
+                    situation,
+                    action,
+                    outcome,
+                    timestamp
+                )
+
+                # Display 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: {game_state['gear_strength']:.2f}")
+                st.markdown(f"🏋️ Rider Skill: {game_state['rider_skill']:.2f}")
+
+                # 🏅 Display Scoreboard
+                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 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=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'])
+                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()