import streamlit as st import streamlit.components.v1 as components import spacy import random import re from gtts import gTTS from PIL import Image, ImageDraw, ImageFont import io import numpy as np import base64 # Load spaCy model for NLP nlp = spacy.load("en_core_web_sm") # Default word lists for storytelling classes (derived from original app) default_word_lists = { "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 suit_properties = { "Hearts": "emotional or romantic", "Diamonds": "wealthy or luxurious", "Clubs": "conflict or struggle", "Spades": "mysterious or dangerous" } # Sentence templates for story generation sentence_templates = { "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()} for ent in doc.ents: 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["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["Dialogue"].extend(dialogues) return augmented_lists # Create a 52-card deck def create_deck(): suits = ["Hearts", "Diamonds", "Clubs", "Spades"] 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 def assign_card_to_class(card_index): if 0 <= card_index < 10: return "Location" elif 10 <= card_index < 20: return "Actions" elif 20 <= card_index < 30: return "Thoughts" elif 30 <= card_index < 40: return "Emotions" else: return "Dialogue" # 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) 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() # Generate story sentence def generate_story_sentence(story_class, word, property): return sentence_templates[story_class].format(word=word, property=property) # 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"]][: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 app def main(): st.set_page_config(page_title="StoryForge: The Game", page_icon="🎴", layout="wide") st.title("🎴 StoryForge: A Storytelling Adventure 🎴") # 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: 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 if st.button("Start Game"): if user_input: st.session_state.augmented_lists = augment_word_lists(user_input) st.session_state.deck = create_deck() st.session_state.story = [] st.session_state.drawn_cards = 0 st.success("Game started! Draw your first card.") # 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)) # Song generation if st.session_state.drawn_cards == 52: full_story = "\n".join(st.session_state.story) st.markdown("### 🎵 Story Song") lyrics = generate_song_lyrics(full_story) st.write(lyrics) tts = gTTS(text=lyrics, lang="en") audio_file = "story_song.mp3" tts.save(audio_file) st.audio(audio_file, format="audio/mp3") # Enhanced UI with p5.js st.markdown("## 🎮 Game Board") components.html("""
""", height=220) # Retain original star layout st.markdown("## ⭐ Five Pillars of Storytelling ⭐") star_html = """
🏠
Location
🏃
Actions
🧠
Thoughts
😲
Emotions
💬
Dialogue
""" components.html(star_html, height=450) if __name__ == "__main__": main()