Update app.py

app.py

@@ -1,18 +1,12 @@
 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 for storytelling classes
 default_word_lists = {
     "Location": ["quiet town", "small village", "city", "forest", "mountain"],
     "Actions": ["walking", "pedaling", "running", "dancing", "exploring"],
@@ -38,27 +32,37 @@ sentence_templates = {
     "Dialogue": "Someone spoke with a {property} tone: {word}"
 }
 
-# Function to process user input and augment word lists
+# Pure Python function to augment word lists from user input
 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)
+    # Split input into words
+    words = user_input.lower().split()
     
-    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)
+    # Simple heuristic lists for categorization
+    location_keywords = ["town", "village", "city", "forest", "mountain", "place", "land"]
+    action_keywords = ["walk", "run", "dance", "pedal", "explore", "move", "jump"]
+    emotion_keywords = ["joy", "pain", "smile", "storm", "fear", "love", "anger"]
     
+    # Extract dialogues with regex
     dialogues = re.findall(r'"[^"]*"', user_input)
     augmented_lists["Dialogue"].extend(dialogues)
     
+    # Categorize words
+    for word in words:
+        if any(keyword in word for keyword in location_keywords):
+            augmented_lists["Location"].append(word)
+        elif any(keyword in word for keyword in action_keywords):
+            augmented_lists["Actions"].append(word)
+        elif any(keyword in word for keyword in emotion_keywords):
+            augmented_lists["Emotions"].append(word)
+        elif "?" in word or "what" in word or "why" in word:  # Simple thought detection
+            augmented_lists["Thoughts"].append(word)
+    
+    # Remove duplicates
+    for key in augmented_lists:
+        augmented_lists[key] = list(set(augmented_lists[key]))
+    
     return augmented_lists
 
 # Create a 52-card deck
@@ -103,8 +107,8 @@ def generate_story_sentence(story_class, word, 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
+    words = story_text.split()
+    key_elements = [word for word in words if len(word) > 3][:12]  # Simple filter for ~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
 
@@ -136,7 +140,7 @@ def main():
             st.session_state.drawn_cards = 0
             st.success("Game started! Draw your first card.")
     
-    # Draw card
+    # Draw card and story display
     col1, col2 = st.columns([1, 3])
     with col1:
         if st.button("Draw Card") and st.session_state.drawn_cards < 52:
@@ -170,30 +174,20 @@ def main():
         tts.save(audio_file)
         st.audio(audio_file, format="audio/mp3")
     
-    # 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(600, 200);
-                canvas.parent('sketch-holder');
-                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=220)
+    # Enhanced UI with HTML/CSS
+    st.markdown("""
+        ## 🎮 Game Board
+        <div style="background-color: #f0f0f0; padding: 20px; border-radius: 10px; text-align: center;">
+            <h3 style="color: #333;">Draw cards to weave your epic tale!</h3>
+            <button style="background-color: #ff9900; color: white; border: none; padding: 10px 20px; border-radius: 5px; cursor: pointer;"
+                    onmouseover="this.style.backgroundColor='#cc7700'" 
+                    onmouseout="this.style.backgroundColor='#ff9900'">
+                Hover for a surprise!
+            </button>
+        </div>
+    """, unsafe_allow_html=True)
 
-    # Retain original star layout
+    # 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%;">
@@ -214,7 +208,7 @@ def main():
         </div>
     </div>
     """
-    components.html(star_html, height=450)
+    st.markdown(star_html, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()