Update app.py

app.py

@@ -32,6 +32,15 @@ sentence_templates = {
     "Dialogue": "Someone spoke with a {property} tone: {word}"
 }
 
+# Choice templates for branching narrative
+choice_templates = {
+    "Location": ["Explore the {word} further.", "Leave the {word} behind."],
+    "Actions": ["Continue {word} despite the risk.", "Stop {word} and reconsider."],
+    "Thoughts": ["Pursue the '{word}' idea.", "Ignore the '{word}' thought."],
+    "Emotions": ["Embrace the {word}.", "Suppress the {word}."],
+    "Dialogue": ["Respond to {word}.", "Ignore {word} and move on."]
+}
+
 # Pure Python function to augment word lists from user input
 def augment_word_lists(user_input):
     augmented_lists = {key: list(set(val)) for key, val in default_word_lists.items()}
@@ -79,7 +88,6 @@ def assign_card_to_class(card_index):
 
 # Generate card visualization with p5.js
 def generate_card_visualization(suit, rank, story_class, word, property):
-    # Parameterize animation based on card properties
     num_balls = rank * 5  # More balls for higher rank
     jelly_size = {"Hearts": 40, "Diamonds": 50, "Clubs": 30, "Spades": 60}[suit]  # Suit affects jellyfish size
     rotation_speed = {"Location": 0.005, "Actions": 0.01, "Thoughts": 0.003, "Emotions": 0.007, "Dialogue": 0.009}[story_class]
@@ -239,6 +247,10 @@ def generate_card_visualization(suit, rank, story_class, word, property):
 def generate_story_sentence(story_class, word, property):
     return sentence_templates[story_class].format(word=word, property=property)
 
+# Generate choice options
+def generate_choices(story_class, word):
+    return [template.format(word=word) for template in choice_templates[story_class]]
+
 # Generate song lyrics
 def generate_song_lyrics(story_text):
     words = story_text.split()
@@ -248,8 +260,8 @@ def generate_song_lyrics(story_text):
 
 # Main app
 def main():
-    st.set_page_config(page_title="StoryForge: The Animated Game", page_icon="🎴", layout="wide")
-    st.title("🎴 StoryForge: An Animated Storytelling Adventure 🎴")
+    st.set_page_config(page_title="StoryForge: The Animated Adventure", page_icon="🎴", layout="wide")
+    st.title("🎴 StoryForge: A Choose Your Own Adventure Game 🎴")
     
     # User input
     st.markdown("## 📝 Your Story Seed")
@@ -264,6 +276,12 @@ def main():
         st.session_state.story = []
     if "drawn_cards" not in st.session_state:
         st.session_state.drawn_cards = 0
+    if "history" not in st.session_state:
+        st.session_state.history = []
+    if "current_choices" not in st.session_state:
+        st.session_state.current_choices = []
+    if "last_card" not in st.session_state:
+        st.session_state.last_card = None
     
     # Process input
     if st.button("Start Game"):
@@ -271,12 +289,17 @@ def main():
             st.session_state.augmented_lists = augment_word_lists(user_input)
             st.session_state.deck = create_deck()
             st.session_state.story = []
+            st.session_state.history = []
             st.session_state.drawn_cards = 0
+            st.session_state.current_choices = []
+            st.session_state.last_card = None
             st.success("Game started! Draw your first card.")
     
-    # Draw card and visualization
+    # Layout
     col1, col2 = st.columns([2, 3])
+    
     with col1:
+        # Draw card
         if st.button("Draw Card") and st.session_state.drawn_cards < 52:
             card_index = st.session_state.drawn_cards
             suit, rank = st.session_state.deck[card_index]
@@ -288,15 +311,31 @@ def main():
             viz_html = generate_card_visualization(suit, rank, story_class, word, property)
             st.components.v1.html(viz_html, height=420, scrolling=False)
             
-            # Generate story sentence
+            # Generate story sentence and choices
             sentence = generate_story_sentence(story_class, word, property)
             st.session_state.story.append(sentence)
+            st.session_state.current_choices = generate_choices(story_class, word)
+            st.session_state.last_card = (suit, rank, story_class, word, property)
             st.session_state.drawn_cards += 1
+        
+        # Display choices
+        if st.session_state.current_choices:
+            st.markdown("#### Make Your Choice:")
+            choice = st.radio("What happens next?", st.session_state.current_choices)
+            if st.button("Confirm Choice"):
+                st.session_state.history.append((st.session_state.story[-1], choice))
+                st.session_state.current_choices = []
+                st.success(f"Choice made: {choice}")
     
     with col2:
         st.markdown("### 📜 Your Story Unfolds")
         if st.session_state.story:
             st.write("\n".join(st.session_state.story))
+        
+        st.markdown("### 🕰️ Adventure History")
+        if st.session_state.history:
+            for i, (event, choice) in enumerate(st.session_state.history):
+                st.write(f"**Step {i+1}**: {event} → *Choice: {choice}*")
     
     # Song generation
     if st.session_state.drawn_cards == 52: