app.py

import streamlit as st
import pandas as pd
import plotly.express as px
import random
import uuid
from datetime import datetime
from streamlit_flow import streamlit_flow
from streamlit_flow.elements import StreamlitFlowNode, StreamlitFlowEdge
from streamlit_flow.layouts import TreeLayout

# 🌍 Game World Data - SITUATIONS and ACTIONS expanded with humorous elements
SITUATIONS = [
    {
        "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"
    },
    {
        "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"
    },
    {
        "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. Be prepared to face noble feline adversaries! 🐱",
        "emoji": "👑",
        "type": "competition"
    },
    {
        "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. But beware – the palace mice are guarding it! 🧀",
        "emoji": "🧀",
        "type": "heist"
    },
    {
        "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"
    },
    {
        "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"
    },
    {
        "id": "cat_nap",
        "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"
    },
    {
        "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. But first, you must pilot your rocket through an asteroid field. 🚀",
        "emoji": "🌕",
        "type": "exploration"
    },
    {
        "id": "pirate_cove",
        "name": "The Pirate Cove",
        "description": "Sail the high seas with your trusty crew of cats and uncover the secrets of the Pirate Cove. But beware of the treacherous Sea Dogs! 🏴‍☠️",
        "emoji": "🏴‍☠️",
        "type": "exploration"
    },
    {
        "id": "cat_casino",
        "name": "The Cat Casino",
        "description": "Test your luck in the glamorous Cat Casino! Bet your whiskers on the tables and try not to lose it all. Meow is the time! 🎲",
        "emoji": "🎰",
        "type": "competition"
    }
]

ACTIONS = [
    {
        "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": "negotiation",
        "name": "Negotiate",
        "description": "Use diplomacy and clever negotiation to get out of a tight spot. Every cat has their price! 💼",
        "emoji": "💼",
        "type": "social"
    },
    {
        "id": "precision",
        "name": "Precision Attack",
        "description": "Execute a perfectly timed attack to disable traps or defeat enemies. Purrfection in motion! 🎯",
        "emoji": "🎯",
        "type": "skill"
    },
    {
        "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": "speed",
        "name": "Sprint Away",
        "description": "Run faster than you've ever run before to escape danger. Just 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"
    }
]

# 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! 👑"
]

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 generate encounter conclusions
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)

# Function to generate random actions
def generate_actions():
    return random.sample(ACTIONS, min(3, len(ACTIONS)))

# 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)
    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 |
    """)

    # 🏁 Initialize game state
    if 'game_state' not in st.session_state:
        st.session_state.game_state = {
            'user_id': str(uuid.uuid4()),
            'score': 0,
            'history': {},
            'gear_strength': 5,
            'rider_skill': 5,
            '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)

if __name__ == "__main__":
    main()