Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import streamlit.components.v1 as components
 import spacy
 import random
 import re
@@ -11,13 +12,13 @@ import base64
 # Load spaCy model for NLP
 nlp = spacy.load("en_core_web_sm")
 
-# Default word lists for storytelling classes
+# Default word lists for storytelling classes (derived from original app)
 default_word_lists = {
-    "Characters": ["Hero", "Villain", "Protagonist", "Antagonist", "Wizard", "Knight"],
-    "Settings": ["Forest", "City", "Castle", "Beach", "Mountain", "Space"],
-    "Events": ["Adventure", "Romance", "Mystery", "Conflict", "Discovery", "Escape"],
-    "Objects": ["Sword", "Magic potion", "Treasure chest", "Book", "Ring", "Spaceship"],
-    "Dialogues": ["\"Hello, how are you?\"", "\"Let's go!\"", "\"Beware the danger!\"", "\"I love you.\""]
+    "Location": ["quiet town", "small village", "city", "forest", "mountain"],
+    "Actions": ["walking", "pedaling", "running", "dancing", "exploring"],
+    "Thoughts": ["chasing shadows", "what if", "brilliance of years", "echoes", "secrets"],
+    "Emotions": ["joy", "pain", "trembling smile", "storm", "silent art"],
+    "Dialogue": ["\"Keep moving, dare to feel;\"", "\"Am I chasing shadows?\"", "\"The dawn awaits!\"", "\"I love you.\"", "\"Let’s go!\""]
 }
 
 # Suit properties for narrative flavor
@@ -30,58 +31,58 @@ suit_properties = {
 
 # Sentence templates for story generation
 sentence_templates = {
-    "Characters": "There was a {word} who was {property}.",
-    "Settings": "The {word} was {property}, stretching far into the distance.",
-    "Events": "Then, a {property} {word} unfolded.",
-    "Objects": "A {property} {word} appeared before them.",
-    "Dialogues": "One of them said, {word}, with a {property} tone."
+    "Location": "The story unfolded in a {property} {word}.",
+    "Actions": "Suddenly, a {property} {word} changed everything.",
+    "Thoughts": "A {property} thought, '{word}', crossed their mind.",
+    "Emotions": "A {property} wave of {word} surged through them.",
+    "Dialogue": "Someone spoke with a {property} tone: {word}"
 }
 
 # Function to process user input and augment word lists
 def augment_word_lists(user_input):
     doc = nlp(user_input)
-    augmented_lists = {key: list(set(val)) for key, val in default_word_lists.items()}  # Start with defaults
+    augmented_lists = {key: list(set(val)) for key, val in default_word_lists.items()}
     
-    # Extract elements from user input
     for ent in doc.ents:
-        if ent.label_ == "PERSON":
-            augmented_lists["Characters"].append(ent.text)
-        elif ent.label_ in ["GPE", "LOC"]:
-            augmented_lists["Settings"].append(ent.text)
+        if ent.label_ in ["GPE", "LOC"]:
+            augmented_lists["Location"].append(ent.text)
     
     for token in doc:
-        if token.pos_ == "VERB" and token.text not in augmented_lists["Events"]:
-            augmented_lists["Events"].append(token.text)
-        elif token.pos_ == "NOUN" and token.text not in augmented_lists["Characters"] + augmented_lists["Settings"]:
-            augmented_lists["Objects"].append(token.text)
+        if token.pos_ == "VERB" and token.text not in augmented_lists["Actions"]:
+            augmented_lists["Actions"].append(token.text)
+        elif token.pos_ == "NOUN" and token.text not in augmented_lists["Location"]:
+            if "emotion" in token.text.lower() or token.text in ["joy", "pain", "smile"]:
+                augmented_lists["Emotions"].append(token.text)
+            else:
+                augmented_lists["Thoughts"].append(token.text)
     
     dialogues = re.findall(r'"[^"]*"', user_input)
-    augmented_lists["Dialogues"].extend(dialogues)
+    augmented_lists["Dialogue"].extend(dialogues)
     
     return augmented_lists
 
-# Function to create a 52-card deck
+# Create a 52-card deck
 def create_deck():
     suits = ["Hearts", "Diamonds", "Clubs", "Spades"]
-    ranks = list(range(1, 14))  # 1-13 for each suit
+    ranks = list(range(1, 14))
     deck = [(suit, rank) for suit in suits for rank in ranks]
     random.shuffle(deck)
     return deck
 
-# Assign cards to classes based on position
+# Assign cards to classes
 def assign_card_to_class(card_index):
     if 0 <= card_index < 10:
-        return "Characters"
+        return "Location"
     elif 10 <= card_index < 20:
-        return "Settings"
+        return "Actions"
     elif 20 <= card_index < 30:
-        return "Events"
+        return "Thoughts"
     elif 30 <= card_index < 40:
-        return "Objects"
+        return "Emotions"
     else:
-        return "Dialogues"
+        return "Dialogue"
 
-# Generate a card image with details
+# Generate card image
 def generate_card_image(suit, rank, story_class, word, property):
     img = Image.new("RGB", (200, 300), color="white")
     draw = ImageDraw.Draw(img)
@@ -96,25 +97,27 @@ def generate_card_image(suit, rank, story_class, word, property):
     img.save(buffer, format="PNG")
     return buffer.getvalue()
 
-# Generate story sentence based on card
+# Generate story sentence
 def generate_story_sentence(story_class, word, property):
-    template = sentence_templates[story_class]
-    return template.format(word=word, property=property)
+    return sentence_templates[story_class].format(word=word, property=property)
 
-# Generate song lyrics from story
+# Generate song lyrics
 def generate_song_lyrics(story_text):
     doc = nlp(story_text)
-    key_elements = [token.text for token in doc if token.pos_ in ["NOUN", "VERB", "ADJ"]][:10]  # Top 10 elements
-    lyrics = " ".join(key_elements)  # Simple concatenation for 60-second duration
+    key_elements = [token.text for token in doc if token.pos_ in ["NOUN", "VERB", "ADJ"]][:12]  # ~60 seconds
+    lyrics = "\n".join([f"{key_elements[i]} {key_elements[i+1]}" for i in range(0, len(key_elements)-1, 2)])
     return lyrics
 
-# Main Streamlit app
+# Main app
 def main():
-    st.title("StoryForge: Card-Based Story Generator")
+    st.set_page_config(page_title="StoryForge: The Game", page_icon="🎴", layout="wide")
+    st.title("🎴 StoryForge: A Storytelling Adventure 🎴")
     
-    # User input for story seed
-    user_input = st.text_area("Paste your story seed here (large text accepted):", height=200)
+    # User input
+    st.markdown("## 📝 Your Story Seed")
+    user_input = st.text_area("Paste your story inspiration here:", height=200)
     
+    # Session state initialization
     if "augmented_lists" not in st.session_state:
         st.session_state.augmented_lists = default_word_lists
     if "deck" not in st.session_state:
@@ -124,68 +127,94 @@ def main():
     if "drawn_cards" not in st.session_state:
         st.session_state.drawn_cards = 0
     
-    # Process input and augment word lists
-    if st.button("Process Input"):
+    # Process input
+    if st.button("Start Game"):
         if user_input:
             st.session_state.augmented_lists = augment_word_lists(user_input)
-            st.session_state.deck = create_deck()  # Reset deck
+            st.session_state.deck = create_deck()
             st.session_state.story = []
             st.session_state.drawn_cards = 0
-            st.success("Input processed! Ready to draw cards.")
+            st.success("Game started! Draw your first card.")
     
-    # Draw card button
-    if st.button("Draw a Card") and st.session_state.drawn_cards < 52:
-        card_index = st.session_state.drawn_cards
-        suit, rank = st.session_state.deck[card_index]
-        story_class = assign_card_to_class(card_index)
-        word_list = st.session_state.augmented_lists[story_class]
-        word = random.choice(word_list)
-        property = suit_properties[suit]
-        
-        # Generate and display card image
-        card_image = generate_card_image(suit, rank, story_class, word, property)
-        st.image(card_image, caption=f"Card {card_index + 1}")
-        
-        # Generate and append story sentence
-        sentence = generate_story_sentence(story_class, word, property)
-        st.session_state.story.append(sentence)
-        st.session_state.drawn_cards += 1
-        
-        # Display current story
-        st.write("### Current Story")
-        st.write("\n".join(st.session_state.story))
+    # Draw card
+    col1, col2 = st.columns([1, 3])
+    with col1:
+        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]
+            story_class = assign_card_to_class(card_index)
+            word = random.choice(st.session_state.augmented_lists[story_class])
+            property = suit_properties[suit]
+            
+            card_image = generate_card_image(suit, rank, story_class, word, property)
+            st.image(card_image, caption=f"Card {card_index + 1}")
+            
+            sentence = generate_story_sentence(story_class, word, property)
+            st.session_state.story.append(sentence)
+            st.session_state.drawn_cards += 1
+    
+    with col2:
+        st.markdown("### 📜 Your Story Unfolds")
+        if st.session_state.story:
+            st.write("\n".join(st.session_state.story))
     
-    # Generate song when deck is fully drawn
+    # Song generation
     if st.session_state.drawn_cards == 52:
-        st.write("### Full Story Generated!")
         full_story = "\n".join(st.session_state.story)
-        st.write(full_story)
-        
-        # Generate and play song
+        st.markdown("### 🎵 Story Song")
         lyrics = generate_song_lyrics(full_story)
-        st.write("### Song Lyrics")
         st.write(lyrics)
         
         tts = gTTS(text=lyrics, lang="en")
-        audio_file = "song.mp3"
+        audio_file = "story_song.mp3"
         tts.save(audio_file)
-        audio_bytes = open(audio_file, "rb").read()
-        st.audio(audio_bytes, format="audio/mp3")
+        st.audio(audio_file, format="audio/mp3")
     
-    # Basic p5.js integration for interactivity (placeholder)
-    st.components.v1.html("""
+    # Enhanced UI with p5.js
+    st.markdown("## 🎮 Game Board")
+    components.html("""
         <script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/1.4.2/p5.min.js"></script>
         <div id="sketch-holder"></div>
         <script>
             function setup() {
-                let canvas = createCanvas(400, 100);
+                let canvas = createCanvas(600, 200);
                 canvas.parent('sketch-holder');
-                background(220);
-                textSize(16);
-                text('Draw cards to build your story!', 10, 50);
+                background(240);
+                textSize(20);
+                textAlign(CENTER);
+                text('Draw cards to weave your epic tale!', width/2, height/2);
+            }
+            function draw() {
+                if (mouseIsPressed) {
+                    fill(255, 0, 0);
+                    ellipse(mouseX, mouseY, 20, 20);
+                }
             }
         </script>
-    """, height=120)
+    """, height=220)
+
+    # Retain original star layout
+    st.markdown("## ⭐ Five Pillars of Storytelling ⭐")
+    star_html = """
+    <div style="position: relative; width: 400px; height: 400px; margin: auto; border: 1px dashed #aaa; border-radius: 50%;">
+        <div style="position: absolute; top: 50px; left: 200px; transform: translate(-50%, -50%); text-align: center;">
+            <div style="font-size: 2em;">🏠</div><div><b>Location</b></div>
+        </div>
+        <div style="position: absolute; top: 154px; left: 57px; transform: translate(-50%, -50%); text-align: center;">
+            <div style="font-size: 2em;">🏃</div><div><b>Actions</b></div>
+        </div>
+        <div style="position: absolute; top: 321px; left: 112px; transform: translate(-50%, -50%); text-align: center;">
+            <div style="font-size: 2em;">🧠</div><div><b>Thoughts</b></div>
+        </div>
+        <div style="position: absolute; top: 321px; left: 288px; transform: translate(-50%, -50%); text-align: center;">
+            <div style="font-size: 2em;">😲</div><div><b>Emotions</b></div>
+        </div>
+        <div style="position: absolute; top: 154px; left: 343px; transform: translate(-50%, -50%); text-align: center;">
+            <div style="font-size: 2em;">💬</div><div><b>Dialogue</b></div>
+        </div>
+    </div>
+    """
+    components.html(star_html, height=450)
 
 if __name__ == "__main__":
     main()