Update app.py

app.py

@@ -8,6 +8,21 @@ 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},
+    {"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}
+]
+
+RIDING_GEAR = [
+    {"name": "Feathered Boots", "type": "Agility", "strength": 2},
+    {"name": "Golden Armor", "type": "Defense", "strength": 4},
+    {"name": "Magic Whisker Wand", "type": "Magic", "strength": 3},
+    {"name": "Sleek Shadow Cape", "type": "Stealth", "strength": 1}
+]
+
 # 🌍 Game World Data - SITUATIONS and ACTIONS expanded with humorous elements
 SITUATIONS = [
     {
@@ -182,40 +197,141 @@ FAILURE_CONCLUSIONS = [
     "It’s okay, every cat has a learning curve. 📚"
 ]
 
-# Function to generate encounter conclusions
+# 🧠 Game Mechanics
+def generate_situation():
+    return random.choice(SITUATIONS)
+
+def generate_actions():
+    return random.sample(ACTIONS, min(3, len(ACTIONS)))
+
+def evaluate_action(action, gear_strength, rider_skill, history):
+    base_success_chance = (gear_strength + rider_skill) / 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 generate_encounter_conclusion(situation, action, outcome):
     if outcome:
         return random.choice(SUCCESS_CONCLUSIONS)
     else:
         return random.choice(FAILURE_CONCLUSIONS)
 
-# Function to generate a random situation
-def generate_situation():
-    return random.choice(SITUATIONS)
+# 🌳 Journey Visualization with Heterogeneous Graph Structure
+def create_heterogeneous_graph(history_df):
+    nodes = []
+    edges = []
 
-# Function to generate random actions
-def generate_actions():
-    return random.sample(ACTIONS, min(3, len(ACTIONS)))
+    # Define node shapes based on situation and action types
+    situation_shapes = {
+        "escape": "diamond",
+        "exploration": "triangle",
+        "competition": "star"
+    }
+    action_shapes = {
+        "skill": "square",
+        "physical": "circle",
+        "social": "hexagon",
+        "mental": "octagon"
+    }
+
+    for index, row in history_df.iterrows():
+        situation_id = f"situation-{index}"
+        action_id = f"action-{index}"
+        conclusion_id = f"conclusion-{index}"
+
+        # Create situation node
+        situation_content = f"{row['situation_emoji']} {row['situation_name']}\n🕒 {row['timestamp']}"
+        situation_node = StreamlitFlowNode(situation_id, (0, 0), {'content': situation_content}, 'output', 'bottom', 'top', shape=situation_shapes.get(row['situation_type'], 'ellipse'))
+        nodes.append(situation_node)
+
+        # Create action node
+        action_content = f"{row['action_emoji']} {row['action_name']}\nOutcome: {'✅ Success' if row['outcome'] else '❌ Failure'}"
+        action_node = StreamlitFlowNode(action_id, (0, 0), {'content': action_content}, 'output', 'bottom', 'top', shape=action_shapes.get(row['action_type'], 'ellipse'))
+        nodes.append(action_node)
+
+        # Create conclusion node
+        conclusion_content = f"📜 {row['conclusion']}\n💪 Gear: {row['gear_strength']:.2f} | 🏋️ Skill: {row['rider_skill']:.2f}"
+        conclusion_node = StreamlitFlowNode(conclusion_id, (0, 0), {'content': conclusion_content}, 'output', 'bottom', 'top', shape='parallelogram')
+        nodes.append(conclusion_node)
+
+        # Create edges
+        edges.append(StreamlitFlowEdge(f"{situation_id}-{action_id}", situation_id, action_id, animated=True, dashed=False))
+        edges.append(StreamlitFlowEdge(f"{action_id}-{conclusion_id}", action_id, conclusion_id, animated=True, dashed=False))
+
+        # Create edge to previous conclusion if not the first node
+        if index > 0:
+            prev_conclusion_id = f"conclusion-{index-1}"
+            edges.append(StreamlitFlowEdge(f"{prev_conclusion_id}-{situation_id}", prev_conclusion_id, situation_id, animated=True, dashed=True))
+
+    return nodes, edges
+
+# 📝 Markdown Preview
+def create_markdown_preview(history_df):
+    markdown = "## 🌳 Journey Preview\n\n"
+    for index, row in history_df.iterrows():
+        indent = "  " * (index * 3)
+        markdown += f"{indent}🌟 **{row['situation_name']}** ({row['situation_type']})\n"
+        markdown += f"{indent}  ↪ {row['action_emoji']} {row['action_name']} ({row['action_type']}): "
+        markdown += "✅ Success\n" if row['outcome'] else "❌ Failure\n"
+        markdown += f"{indent}    📜 {row['conclusion']}\n"
+        markdown += f"{indent}    💪 Gear: {row['gear_strength']:.2f} | 🏋️ Skill: {row['rider_skill']:.2f}\n\n"
+    return markdown
+
+# 🔄 Game State Management
+def update_game_state(game_state, situation, action, outcome, timestamp):
+    conclusion = generate_encounter_conclusion(situation, action, outcome)
+    game_state = update_character_stats(game_state, outcome)
+
+    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']],
+        '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']]
+    })
+    game_state['history_df'] = pd.concat([game_state['history_df'], new_record], ignore_index=True)
+
+    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
+
+# 🔄 Update character stats based on the outcome
+def update_character_stats(game_state, outcome):
+    if outcome:
+        game_state['gear_strength'] = min(10, game_state['gear_strength'] + random.uniform(0.1, 0.5))
+        game_state['rider_skill'] = min(10, game_state['rider_skill'] + random.uniform(0.1, 0.5))
+    return game_state
 
-# The rest of the code remains mostly the same
 # 🎮 Main Game Application
 def main():
     st.title("🐱 Cat Rider 🏇")
-    st.markdown("""
-    ## Welcome to Cat Rider!
-    In this immersive adventure, you will explore the thrilling world of feline riders. This game sets the stage for dramatic situations and guided storytelling with engaging interactive elements.
-    """)
 
-    # 📜 Game Rules (unchanged)
+    # 📜 Game Rules
     st.markdown("""
     ### 📜 Game Rules
-    | 🛤️ Step | 📝 Description |
-    |---------|----------------|
-    | 1️⃣ | Choose your Cat Rider |
-    | 2️⃣ | Select the Riding Gear |
-    | 3️⃣ | Set off on an Adventure |
-    | 4️⃣ | Encounter Challenges and Make Decisions |
-    | 5️⃣ | Complete the Quest and Grow Stronger |
+    | 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
@@ -224,77 +340,100 @@ def main():
             'user_id': str(uuid.uuid4()),
             'score': 0,
             'history': {},
-            'gear_strength': 5,
-            'rider_skill': 5,
+            'gear_strength': 0,
+            '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'])
         }
 
-    # Main gameplay loop and event handling
-    situation = generate_situation()
-    actions = generate_actions()
-    
-    # Display current situation and available actions
-    st.markdown(f"## {situation['emoji']} Current Situation: {situation['name']} ({situation['type']})")
-    st.markdown(situation['description'])
-    st.markdown("### 🎭 Choose your action:")
-    
-    cols = st.columns(3)
-    for i, action in enumerate(actions):
-        if cols[i].button(f"{action['emoji']} {action['name']} ({action['type']})"):
-            outcome, success_chance = evaluate_action(action, st.session_state.game_state['gear_strength'], st.session_state.game_state['rider_skill'], st.session_state.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:
-                st.session_state.game_state['score'] += 1
-            
-            # 🔄 Update game state
-            st.session_state.game_state = update_game_state(
-                st.session_state.game_state,
-                situation,
-                action,
-                outcome,
-                timestamp
-            )
-            
-            # Display conclusion
-            conclusion = st.session_state.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}")
-
-    # 📝 Display Markdown Preview (unchanged)
-    if not st.session_state.game_state['history_df'].empty:
-        st.markdown(create_markdown_preview(st.session_state.game_state['history_df']))
-    
-    # 🌳 Display Heterogeneous Journey Graph (unchanged)
-    if not st.session_state.game_state['history_df'].empty:
-        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)
+    # 🐱 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']})"):
+                st.session_state.game_state['cat_rider'] = rider
+                st.session_state.game_state['rider_skill'] = rider['skill']
+
+    # 🏇 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']})"):
+                st.session_state.game_state['riding_gear'] = gear
+                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()
+
+        st.markdown(f"## {situation['emoji']} Current Situation: {situation['name']} ({situation['type']})")
+        st.markdown(situation['description'])
+        st.markdown("### 🎭 Choose your action:")
+
+        cols = st.columns(3)
+        for i, action in enumerate(actions):
+            if cols[i].button(f"{action['emoji']} {action['name']} ({action['type']})"):
+                outcome, success_chance = evaluate_action(action, st.session_state.game_state['gear_strength'], st.session_state.game_state['rider_skill'], st.session_state.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:
+                    st.session_state.game_state['score'] += 1
+
+                # 🔄 Update game state
+                st.session_state.game_state = update_game_state(
+                    st.session_state.game_state,
+                    situation,
+                    action,
+                    outcome,
+                    timestamp
+                )
+
+                # Display conclusion
+                conclusion = st.session_state.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}")
+
+        # 📝 Display Markdown Preview
+        if not st.session_state.game_state['history_df'].empty:
+            st.markdown(create_markdown_preview(st.session_state.game_state['history_df']))
+
+        # 🌳 Display Heterogeneous Journey Graph
+        if not st.session_state.game_state['history_df'].empty:
+            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)
 
 if __name__ == "__main__":
     main() 
+
+
+