Update app.py

app.py

@@ -1,225 +1,191 @@
 import streamlit as st
-import streamlit.components.v1 as components
-import math
+import spacy
+import random
+import re
+from gtts import gTTS
+from PIL import Image, ImageDraw, ImageFont
+import io
+import numpy as np
+import base64
 
-def main():
-    st.set_page_config(page_title="Storytelling Mastery", page_icon="📚", layout="wide")
-    
-    st.title("🎭 The Art of Storytelling: Master Guide 🎭")
-    
-    # -----------------------------
-    # 1. The Storyteller's Song
-    # -----------------------------
-    st.markdown("## 🎵 The Storyteller's Journey Song 🎵")
-    st.markdown("""
-**Verse 1 – Location**  
-*I'm in a quiet town, where twilight meets the dawn,*  
-*A simple street and weathered walls, where whispered dreams are drawn.*
+# Load spaCy model for NLP
+nlp = spacy.load("en_core_web_sm")
 
-**Verse 2 – Actions**  
-*I lace my worn-out sneakers, stepping into a new day’s light,*  
-*I pedal through cobbled memories, heart racing with each flight.*
+# Default word lists for storytelling classes
+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.\""]
+}
 
-**Verse 3 – Thoughts**  
-*In every echo of my footsteps, a raw, unfiltered thought appears,*  
-*I muse, “Am I chasing shadows, or the brilliance of my years?”*
+# Suit properties for narrative flavor
+suit_properties = {
+    "Hearts": "emotional or romantic",
+    "Diamonds": "wealthy or luxurious",
+    "Clubs": "conflict or struggle",
+    "Spades": "mysterious or dangerous"
+}
 
-**Verse 4 – Emotions**  
-*My eyes reflect a storm of joy and pain—a silent, candid art,*  
-*A trembling smile, a furrowed brow, emotions painted on my heart.*
+# 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."
+}
 
-**Verse 5 – Dialogue**  
-*Then comes a gentle, piercing voice: “Keep moving, dare to feel;”*  
-*Those words ignite the darkened skies—my destiny is real.*
+# 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
+    
+    # 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)
+    
+    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)
+    
+    dialogues = re.findall(r'"[^"]*"', user_input)
+    augmented_lists["Dialogues"].extend(dialogues)
+    
+    return augmented_lists
 
-**Chorus**  
-*Let each moment unfold its magic, in places, moves, and minds so bold,*  
-*For every whispered thought and burst of feeling tells a story yet untold.*
-    """)
-    
-    st.markdown("---")
-    
-    # -----------------------------
-    # 2. The Storytelling Points Outline
-    # -----------------------------
-    storytelling_points = [
-        {"number": 1, "emoji": "🏠", "keyword": "location", "description": "Set a scene that sparks the imagination: a place that's simply stated but vividly felt."},
-        {"number": 2, "emoji": "🏃", "keyword": "actions", "description": "Show the kinetic energy—whether it’s walking, biking, or a spontaneous burst of movement."},
-        {"number": 3, "emoji": "🧠", "keyword": "thoughts", "description": "Reveal those raw, neurotic inner monologues that make you real and relatable."},
-        {"number": 4, "emoji": "😲", "keyword": "emotions", "description": "Let your body speak the language of your feelings—every twitch and tear is a story."},
-        {"number": 5, "emoji": "💬", "keyword": "dialogue", "description": "Capture crisp, memorable dialogue that slices through the moment like lightning."},
-        # Additional points can be used for further storytelling techniques…
-    ]
-    
-    # -----------------------------
-    # 3. Random Literary Wit – Ten Poetic Lines
-    # -----------------------------
-    random_poetry = [
-        "🌅 Location: Where the horizon kisses the edge of dreams. 🌌",
-        "🚴 Action: Pedaling through the city's heartbeat, every turn a burst of rhythm. 🎶",
-        "💭 Thoughts: Whispering secrets of a chaotic mind under a tranquil sky. 🌒",
-        "😊 Emotions: A smile that trembles like morning dew on a rose. 🌈",
-        "🗣 Dialogue: Words crackling like fire in the midnight hush. 🔥",
-        "🌲 Nature: The forest murmurs ancient tales to those who dare to listen. 🍃",
-        "🌊 Waves: Dancing to the eternal hymn of the deep blue. 🐚",
-        "🌟 Dreams: Starlight scattered over the canvas of restless nights. ✨",
-        "🔥 Passion: A flame burning fierce in the silence of despair. ❤️",
-        "🍂 Reflection: Leaves falling like whispered memories in autumn’s embrace. 🕊️"
-    ]
-    
-    st.markdown("## ✨ Random Literary Wit")
-    st.code("\n".join(random_poetry), language="python")
-    
-    # -----------------------------
-    # 4. Interactive Story Element Explorer (Vertical Flow and Tabs)
-    # -----------------------------
-    st.markdown("## 📚 The Art of Storytelling: 5 Essential Elements")
-    
-    st.markdown("> \"The universe is made of stories, not of atoms.\" — Muriel Rukeyser")
-    
-    for point in storytelling_points:
-        st.markdown(f"{point['number']}. {point['emoji']} **{point['keyword'].capitalize()}** - {point['description']}")
-    
-    # -----------------------------
-    # 5. Star Layout for the Five Pillars
-    # -----------------------------
-    st.markdown("## ⭐ Five Pillars in a Star Layout ⭐")
-    
-    # Using absolute positioning in a relative container to arrange five points on a circle
-    # Calculated using basic trigonometry (center = 200, radius = 150)
-    star_html = """
-    <div style="position: relative; width: 400px; height: 400px; margin: auto; border: 1px dashed #aaa; border-radius: 50%;">
-        <!-- Point 1: Location at angle 90° -->
-        <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>
-        <!-- Point 2: Actions at angle ~162° -->
-        <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>
-        <!-- Point 3: Thoughts at angle ~234° -->
-        <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>
-        <!-- Point 4: Emotions at angle ~306° -->
-        <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>
-        <!-- Point 5: Dialogue at angle ~18° -->
-        <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)
-    
-    # -----------------------------
-    # 6. Additional Interactive Elements (Tabs & Dropdown)
-    # -----------------------------
-    tab1, tab2, tab3 = st.tabs(["Vertical Flow", "Circular Layout", "Categorical Groups"])
-    
-    with tab1:
-        st.markdown("### Vertical Flow Diagram")
-        mermaid_vertical = generate_vertical_diagram(storytelling_points)
-        st.markdown(mermaid_vertical, unsafe_allow_html=True)
-        
-    with tab2:
-        st.markdown("### Circular Storytelling Process")
-        mermaid_circular = generate_circular_diagram(storytelling_points)
-        st.markdown(mermaid_circular, unsafe_allow_html=True)
-        
-    with tab3:
-        st.markdown("### Categorical Groups")
-        mermaid_categories = generate_categorical_diagram(storytelling_points)
-        st.markdown(mermaid_categories, unsafe_allow_html=True)
-    
-    st.markdown("## 🔍 Explore Individual Storytelling Elements")
-    col1, col2 = st.columns([1, 3])
-    
-    with col1:
-        options = [f"{point['number']}. {point['emoji']} {point['keyword'].capitalize()}" for point in storytelling_points]
-        selected_option = st.selectbox("Select an element to explore:", options)
-        selected_number = int(selected_option.split('.')[0])
-        selected_point = next(point for point in storytelling_points if point['number'] == selected_number)
-    
-    with col2:
-        st.markdown(f"### {selected_option}")
-        st.markdown(f"**Description:** {selected_point['description']}")
-        examples = {
-            1: "'In the stillness of a small town, I take in the crisp air...'",
-            2: "'I lace up my sneakers and step boldly into the unknown, each stride a heartbeat.'",
-            3: "'I muse: What if these fleeting moments are the stitches of my very soul?'",
-            4: "'My eyes shimmer with unspoken stories, my face a canvas of turbulent joy.'",
-            5: "'Then, a voice cuts through the silence: ‘Keep pushing, the dawn awaits!’'"
-        }
-        st.markdown(f"**Example:** {examples.get(selected_number, 'No example available.')}")
-    
-    st.markdown("> \"There is no greater agony than bearing an untold story inside you.\" — Maya Angelou")
+# Function to create a 52-card deck
+def create_deck():
+    suits = ["Hearts", "Diamonds", "Clubs", "Spades"]
+    ranks = list(range(1, 14))  # 1-13 for each suit
+    deck = [(suit, rank) for suit in suits for rank in ranks]
+    random.shuffle(deck)
+    return deck
 
-def generate_vertical_diagram(points):
-    """Generate a vertical flow Mermaid diagram"""
-    nodes = []
-    connections = []
-    styles = []
-    
-    for i, point in enumerate(points):
-        node_id = point['number']
-        node_label = f"{node_id}. {point['emoji']} {point['keyword'].capitalize()}"
-        nodes.append(f"    {node_id}(\"{node_label}\")")
-        if i < len(points) - 1:
-            connections.append(f"    {node_id} --> {points[i+1]['number']}")
-        else:
-            connections.append(f"    {node_id} --> {points[0]['number']}")
-    
-    styles.extend([
-        "    %% Style definitions",
-        "    classDef foundation fill:#ff9900,stroke:#333,stroke-width:2px,color:#000;",
-        "    classDef technique fill:#66cc99,stroke:#333,stroke-width:2px,color:#000;",
-        "    classDef learning fill:#6699cc,stroke:#333,stroke-width:2px,color:#000;",
-        "    classDef mastery fill:#cc6699,stroke:#333,stroke-width:2px,color:#000;",
-        "    class 1,2,3,4,5 foundation;"
-    ])
-    
-    diagram = ["```mermaid", "graph TD"]
-    diagram.extend(nodes)
-    diagram.extend(connections)
-    diagram.extend(styles)
-    diagram.append("```")
-    return "\n".join(diagram)
+# Assign cards to classes based on position
+def assign_card_to_class(card_index):
+    if 0 <= card_index < 10:
+        return "Characters"
+    elif 10 <= card_index < 20:
+        return "Settings"
+    elif 20 <= card_index < 30:
+        return "Events"
+    elif 30 <= card_index < 40:
+        return "Objects"
+    else:
+        return "Dialogues"
+
+# Generate a card image with details
+def generate_card_image(suit, rank, story_class, word, property):
+    img = Image.new("RGB", (200, 300), color="white")
+    draw = ImageDraw.Draw(img)
+    font = ImageFont.load_default()
+    
+    draw.text((10, 10), f"{rank} of {suit}", fill="black", font=font)
+    draw.text((10, 50), f"Class: {story_class}", fill="black", font=font)
+    draw.text((10, 90), f"Word: {word}", fill="black", font=font)
+    draw.text((10, 130), f"Property: {property}", fill="black", font=font)
+    
+    buffer = io.BytesIO()
+    img.save(buffer, format="PNG")
+    return buffer.getvalue()
 
-def generate_circular_diagram(points):
-    """Generate a circular flow Mermaid diagram"""
-    diagram = ["```mermaid", "graph TD"]
-    for point in points:
-        node_id = point['number']
-        node_label = f"{node_id}. {point['emoji']} {point['keyword'].capitalize()}"
-        diagram.append(f"    {node_id}[\"{node_label}\"]")
-    diagram.extend([
-        "    1 --> 2 --> 3 --> 4 --> 5 --> 1",
-        "    %% Style definitions",
-        "    classDef q1 fill:#ff9900,stroke:#333,stroke-width:2px,color:#000;",
-        "    class 1,2,3,4,5 q1;"
-    ])
-    diagram.append("```")
-    return "\n".join(diagram)
+# Generate story sentence based on card
+def generate_story_sentence(story_class, word, property):
+    template = sentence_templates[story_class]
+    return template.format(word=word, property=property)
 
-def generate_categorical_diagram(points):
-    """Generate a categorical Mermaid diagram grouping the storytelling elements"""
-    diagram = ["```mermaid", "graph TB",
-               "    subgraph Foundation[\"Foundation Elements\"]",
-               "        1(\"`1. 🏠 Location`\")",
-               "        2(\"`2. 🏃 Actions`\")",
-               "        3(\"`3. 🧠 Thoughts`\")",
-               "        4(\"`4. 😲 Emotions`\")",
-               "        5(\"`5. 💬 Dialogue`\")",
-               "    end",
-               "```"]
-    return "\n".join(diagram)
+# Generate song lyrics from story
+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
+    return lyrics
+
+# Main Streamlit app
+def main():
+    st.title("StoryForge: Card-Based Story Generator")
+    
+    # User input for story seed
+    user_input = st.text_area("Paste your story seed here (large text accepted):", height=200)
+    
+    if "augmented_lists" not in st.session_state:
+        st.session_state.augmented_lists = default_word_lists
+    if "deck" not in st.session_state:
+        st.session_state.deck = create_deck()
+    if "story" not in st.session_state:
+        st.session_state.story = []
+    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"):
+        if user_input:
+            st.session_state.augmented_lists = augment_word_lists(user_input)
+            st.session_state.deck = create_deck()  # Reset deck
+            st.session_state.story = []
+            st.session_state.drawn_cards = 0
+            st.success("Input processed! Ready to draw cards.")
+    
+    # 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))
+    
+    # Generate song when deck is fully drawn
+    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
+        lyrics = generate_song_lyrics(full_story)
+        st.write("### Song Lyrics")
+        st.write(lyrics)
+        
+        tts = gTTS(text=lyrics, lang="en")
+        audio_file = "song.mp3"
+        tts.save(audio_file)
+        audio_bytes = open(audio_file, "rb").read()
+        st.audio(audio_bytes, format="audio/mp3")
+    
+    # Basic p5.js integration for interactivity (placeholder)
+    st.components.v1.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);
+                canvas.parent('sketch-holder');
+                background(220);
+                textSize(16);
+                text('Draw cards to build your story!', 10, 50);
+            }
+        </script>
+    """, height=120)
 
 if __name__ == "__main__":
-    main()
+    main()