Update app.py

app.py

@@ -6,10 +6,9 @@ import textwrap
 from io import BytesIO
 from PIL import Image, ImageDraw, ImageFont
 import numpy as np
-from gtts import gTTS
 from transformers import pipeline
 import base64
-import os
+import re
 
 # Set up the page
 st.set_page_config(
@@ -128,51 +127,61 @@ st.markdown("""
     margin: 0 auto;
 }
 
-.character-option {
-    border: 3px solid transparent;
+.hero-suggestion {
+    background: #ffeb3b;
+    color: #333;
     border-radius: 10px;
-    padding: 5px;
-    margin: 5px;
-    cursor: pointer;
-    transition: all 0.3s;
+    padding: 1rem;
+    margin: 1rem 0;
+    text-align: center;
+    font-weight: bold;
+    border: 2px dashed #ff9800;
 }
 
-.character-option:hover {
-    transform: scale(1.05);
+.world-suggestion {
+    background: #4caf50;
+    color: white;
+    border-radius: 10px;
+    padding: 1rem;
+    margin: 1rem 0;
+    text-align: center;
+    font-weight: bold;
+    border: 2px dashed #2e7d32;
 }
 
-.character-option.selected {
-    border-color: #ff5722;
+.step-container {
+    display: flex;
+    justify-content: space-between;
+    margin-bottom: 2rem;
 }
 
-.tab-content {
-    padding: 1rem;
-    border: 2px solid #4caf50;
+.step {
+    background: rgba(255, 255, 255, 0.9);
     border-radius: 10px;
-    background: rgba(255, 255, 255, 0.8);
-    margin-top: 1rem;
+    padding: 1rem;
+    width: 23%;
+    text-align: center;
+    box-shadow: 0 4px 8px rgba(0,0,0,0.1);
 }
 
-.level-indicator {
-    background: #4caf50;
-    color: white;
-    border-radius: 20px;
-    padding: 0.5rem 1rem;
-    display: inline-block;
+.step-number {
+    font-size: 1.5rem;
     font-weight: bold;
-    margin-bottom: 1rem;
+    background: #ff5722;
+    color: white;
+    border-radius: 50%;
+    width: 30px;
+    height: 30px;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    margin: 0 auto 10px;
 }
 
 /* Progress bar styling */
 .stProgress > div > div > div {
     background-color: #ff5722 !important;
 }
-
-/* Audio player styling */
-.audio-player {
-    width: 100%;
-    margin-top: 10px;
-}
 </style>
 """, unsafe_allow_html=True)
 
@@ -181,22 +190,19 @@ st.markdown("""
 def load_models():
     """Load open-source AI models"""
     try:
-        # Text generation model for code explanations
-        story_to_code = pipeline("text-generation", model="gpt2", max_length=100)
-        
-        # Sentiment analysis for story understanding
-        sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
-        
         # Named entity recognition for identifying objects
         ner_model = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english")
         
-        # Text-to-text generation for better explanations
-        explanation_generator = pipeline("text2text-generation", model="google/flan-t5-base")
+        # Text classification for theme detection
+        classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+        
+        # Text generation for code explanations
+        explanation_generator = pipeline("text2text-generation", model="google/flan-t5-large")
         
-        return story_to_code, sentiment_analyzer, ner_model, explanation_generator
+        return ner_model, classifier, explanation_generator
     except Exception as e:
         st.error(f"Error loading models: {e}")
-        return None, None, None, None
+        return None, None, None
 
 # Image generation functions
 def create_storyboard_image(text, width=400, height=300):
@@ -244,6 +250,9 @@ def generate_sprite_animation(story, character="spaceship", theme="space", num_f
         elif theme == "medieval":
             bg_color = (139, 69, 19)
             star_color = None
+        elif theme == "underwater":
+            bg_color = (0, 105, 148)
+            star_color = None
         else:
             bg_color = (0, 0, 30)
             star_color = (255, 255, 255)
@@ -304,6 +313,23 @@ def generate_sprite_animation(story, character="spaceship", theme="space", num_f
                 # Draw sword swing
                 draw.line([(knight_x+25, knight_y+12), (knight_x+45, knight_y-10)], fill=(255, 255, 0), width=2)
         
+        elif character == "mermaid":
+            mermaid_x = 50 + i * 40
+            mermaid_y = 120
+            # Draw mermaid tail
+            draw.ellipse([mermaid_x, mermaid_y, mermaid_x+30, mermaid_y+20], fill=(255, 105, 180))
+            # Draw mermaid body
+            draw.ellipse([mermaid_x+5, mermaid_y-20, mermaid_x+25, mermaid_y], fill=(255, 218, 185))
+            # Draw hair
+            draw.ellipse([mermaid_x-5, mermaid_y-25, mermaid_x+30, mermaid_y-15], fill=(255, 215, 0))
+            
+            if "swim" in story.lower() and i > 0:
+                # Draw bubbles
+                for j in range(3):
+                    bubble_x = mermaid_x + random.randint(0, 30)
+                    bubble_y = mermaid_y - random.randint(10, 30)
+                    draw.ellipse([bubble_x, bubble_y, bubble_x+5, bubble_y+5], fill=(173, 216, 230))
+        
         # Draw enemies based on theme
         if theme == "space" and "alien" in story.lower():
             alien_x = 200
@@ -325,6 +351,14 @@ def generate_sprite_animation(story, character="spaceship", theme="space", num_f
             draw.ellipse([dragon_x, dragon_y, dragon_x+40, dragon_y+20], fill=(178, 34, 34))
             draw.line([(dragon_x+40, dragon_y+10), (dragon_x+60, dragon_y)], fill=(178, 34, 34), width=3)
         
+        elif theme == "underwater" and "shark" in story.lower():
+            shark_x = 220
+            shark_y = 80 + i*10
+            # Draw shark body
+            draw.ellipse([shark_x, shark_y, shark_x+60, shark_y+30], fill=(169, 169, 169))
+            # Draw shark fin
+            draw.polygon([(shark_x+40, shark_y), (shark_x+50, shark_y-20), (shark_x+60, shark_y)], fill=(169, 169, 169))
+        
         frames.append(img)
     
     return frames
@@ -333,12 +367,12 @@ def generate_code_explanation(story, explanation_generator):
     """Generate code explanation using open-source model"""
     try:
         # Create a prompt for the model
-        prompt = f"Explain to a child how this story would become code: '{story}'"
+        prompt = f"Explain to a child how this story would become code: '{story}'. Use simple analogies and relate to real-world objects."
         
         # Generate text
         result = explanation_generator(
             prompt, 
-            max_length=200,
+            max_length=250,
             num_return_sequences=1,
         )
         
@@ -348,71 +382,85 @@ def generate_code_explanation(story, explanation_generator):
         return f"""See how your story became real code? For example, when you wrote "{story.split()[0]}", 
 we used code like: character.move(). That's how we turn words into actions!"""
 
-def text_to_speech(text, lang='en'):
-    """Convert text to speech and return as base64 encoded audio"""
+def extract_story_elements(story, ner_model, classifier):
+    """Extract hero and world from the story using AI models"""
     try:
-        # Create gTTS object
-        tts = gTTS(text=text, lang=lang, slow=False)
+        # Default values
+        hero = "spaceship"
+        world = "space"
+        
+        # Find hero based on keywords and entities
+        hero_keywords = {
+            "spaceship": ["spaceship", "rocket", "ship", "alien", "planet", "star"],
+            "dragon": ["dragon", "monster", "creature", "beast", "wyvern"],
+            "knight": ["knight", "warrior", "prince", "princess", "king", "queen", "sword"],
+            "mermaid": ["mermaid", "merman", "sea", "ocean", "underwater", "fish"]
+        }
+        
+        # Find world based on keywords and classification
+        world_labels = ["space", "jungle", "medieval", "underwater"]
+        
+        # Find hero by keywords
+        story_lower = story.lower()
+        for candidate, keywords in hero_keywords.items():
+            if any(keyword in story_lower for keyword in keywords):
+                hero = candidate
+                break
+        
+        # Use NER to find entities
+        entities = ner_model(story)
+        person_entities = [e['word'] for e in entities if e['entity'] in ['B-PER', 'I-PER']]
         
-        # Save to in-memory bytes buffer
-        audio_bytes = BytesIO()
-        tts.write_to_fp(audio_bytes)
-        audio_bytes.seek(0)
+        # If we found specific character names, adjust hero
+        if person_entities:
+            if "dragon" in story_lower:
+                hero = "dragon"
+            elif "knight" in story_lower or "king" in story_lower or "queen" in story_lower:
+                hero = "knight"
+            elif "mermaid" in story_lower or "sea" in story_lower:
+                hero = "mermaid"
         
-        # Encode as base64
-        b64 = base64.b64encode(audio_bytes.read()).decode()
+        # Classify world
+        result = classifier(story, world_labels)
+        world = result['labels'][0]
         
-        # Create data URI for HTML5 audio
-        data_url = f"data:audio/mp3;base64,{b64}"
+        # Override based on specific keywords
+        if "underwater" in story_lower or "ocean" in story_lower or "sea" in story_lower:
+            world = "underwater"
+        if "forest" in story_lower or "jungle" in story_lower:
+            world = "jungle"
+        if "castle" in story_lower or "kingdom" in story_lower or "dragon" in story_lower:
+            world = "medieval"
+        if "space" in story_lower or "alien" in story_lower or "planet" in story_lower:
+            world = "space"
         
-        return data_url
+        return hero, world
+    
     except Exception as e:
-        st.error(f"Text-to-speech error: {e}")
-        return None
-
-# Achievement system
-ACHIEVEMENTS = {
-    "first_story": {"name": "Storyteller", "icon": "📖", "description": "Created your first story!"},
-    "code_master": {"name": "Code Master", "icon": "💻", "description": "Used 5 different coding concepts"},
-    "animator": {"name": "Animator", "icon": "🎬", "description": "Created 3 animations"},
-    "voice_artist": {"name": "Voice Artist", "icon": "🎤", "description": "Used text-to-speech feature"},
-    "character_designer": {"name": "Character Designer", "icon": "👾", "description": "Created a custom character"}
-}
-
-def unlock_achievement(achievement_id):
-    """Unlock an achievement for the user"""
-    if achievement_id not in st.session_state.achievements_unlocked:
-        st.session_state.achievements_unlocked.append(achievement_id)
-        st.session_state.new_achievement = achievement_id
-        return True
-    return False
+        st.error(f"Error analyzing story: {str(e)}")
+        return "spaceship", "space"
 
 # Header section
 st.markdown('<div class="header">CodeTales ✨</div>', unsafe_allow_html=True)
 st.markdown('<div class="subheader">Storytime + Coding Magic</div>', unsafe_allow_html=True)
-st.markdown('<div style="text-align:center; color:white; font-size:1.2rem; margin-bottom:2rem;">Turn wild stories into playable games with open-source AI magic!</div>', unsafe_allow_html=True)
+st.markdown('<div style="text-align:center; color:white; font-size:1.2rem; margin-bottom:2rem;">Turn wild stories into playable games with AI magic!</div>', unsafe_allow_html=True)
 
 # How it works section
-with st.expander("✨ How It Works (Like Baking a Cake) 🎂"):
-    st.markdown("""
-    **1. Kids Write a Story (The Recipe)**  
-    Example: *"The spaceship zooms past aliens and shoots lasers!"*
-    
-    **2. AI is the Magic Oven 🔮**  
-    We turn words into code using open-source AI models
-    
-    **3. Animation Pops Out (The Cake!) 🎮**  
-    See your story come to life with custom animations!
-    
-    **4. Robot Teacher Explains 🤖**  
-    Tavus shows: *"See? 'spaceship.move_right()' makes it fly! That's coding!"*
-    
-    **5. Level Up! 🏆**  
-    Earn achievements as you learn and create!
-    """)
+st.markdown("### ✨ How It Works")
+step_container = st.container()
+with step_container:
+    cols = st.columns(4)
+    with cols[0]:
+        st.markdown('<div class="step"><div class="step-number">1</div>Write Your Story</div>', unsafe_allow_html=True)
+    with cols[1]:
+        st.markdown('<div class="step"><div class="step-number">2</div>AI Chooses Hero & World</div>', unsafe_allow_html=True)
+    with cols[2]:
+        st.markdown('<div class="step"><div class="step-number">3</div>Watch Animation</div>', unsafe_allow_html=True)
+    with cols[3]:
+        st.markdown('<div class="step"><div class="step-number">4</div>Learn Coding</div>', unsafe_allow_html=True)
 
 # Load models
-story_to_code, sentiment_analyzer, ner_model, explanation_generator = load_models()
+ner_model, classifier, explanation_generator = load_models()
 
 # Initialize session state
 if 'animation_generated' not in st.session_state:
@@ -427,29 +475,28 @@ if 'selected_character' not in st.session_state:
     st.session_state.selected_character = "spaceship"
 if 'selected_theme' not in st.session_state:
     st.session_state.selected_theme = "space"
-if 'sound_enabled' not in st.session_state:
-    st.session_state.sound_enabled = True
-if 'achievements_unlocked' not in st.session_state:
-    st.session_state.achievements_unlocked = []
-if 'new_achievement' not in st.session_state:
-    st.session_state.new_achievement = None
 if 'story_count' not in st.session_state:
     st.session_state.story_count = 0
 if 'level' not in st.session_state:
     st.session_state.level = 1
 if 'xp' not in st.session_state:
     st.session_state.xp = 0
-if 'voice_enabled' not in st.session_state:
-    st.session_state.voice_enabled = False
-if 'custom_character' not in st.session_state:
-    st.session_state.custom_character = None
+if 'analyzed' not in st.session_state:
+    st.session_state.analyzed = False
+
+# Character and theme mapping
+characters = {
+    "spaceship": "🚀 Spaceship",
+    "dragon": "🐉 Dragon",
+    "knight": "🛡️ Knight",
+    "mermaid": "🧜‍♀️ Mermaid"
+}
 
-# Story templates
-story_templates = {
-    "None": "",
-    "Space Adventure": "A brave spaceship zooms through space, shooting lasers at alien spaceships!",
-    "Dragon Quest": "A knight fights a fire-breathing dragon to save the princess in the castle.",
-    "Jungle Explorer": "An explorer runs through the jungle, jumping over snakes and swinging on vines."
+themes = {
+    "space": "🌌 Space",
+    "jungle": "🌿 Jungle",
+    "medieval": "🏰 Medieval",
+    "underwater": "🌊 Underwater"
 }
 
 # Main content
@@ -460,75 +507,45 @@ col1, col2 = st.columns([1, 1])
 
 with col1:
     st.markdown('<div class="story-box">', unsafe_allow_html=True)
-    st.markdown("### 📖 Write Your Story Here:")
-    
-    # Story template selector
-    selected_template = st.selectbox(
-        "Or choose a story template to start with:",
-        list(story_templates.keys()),
-        index=0
-    )
+    st.markdown("### 📖 Step 1: Write Your Story")
     
     story_text = st.text_area(
         "Tell your adventure story...", 
         height=200,
         placeholder="Once upon a time, a brave spaceship zoomed through space, shooting lasers at alien spaceships...",
         label_visibility="collapsed",
-        value=story_templates[selected_template] if selected_template != "None" else st.session_state.story_text
+        value=st.session_state.story_text
     )
     
-    # Character selection
-    st.markdown("### 🧙 Choose Your Hero")
-    char_cols = st.columns(3)
-    characters = {
-        "spaceship": "🚀 Spaceship",
-        "dragon": "🐉 Dragon",
-        "knight": "🛡️ Knight"
-    }
-    
-    for i, (char_key, char_label) in enumerate(characters.items()):
-        with char_cols[i]:
-            if st.button(
-                char_label,
-                key=f"char_{char_key}",
-                use_container_width=True,
-                type="primary" if st.session_state.selected_character == char_key else "secondary"
-            ):
-                st.session_state.selected_character = char_key
-    
-    # Theme selection
-    st.markdown("### 🎨 Choose Your World")
-    theme_cols = st.columns(3)
-    themes = {
-        "space": "🌌 Space",
-        "jungle": "🌿 Jungle",
-        "medieval": "🏰 Medieval"
-    }
-    
-    for i, (theme_key, theme_label) in enumerate(themes.items()):
-        with theme_cols[i]:
-            if st.button(
-                theme_label,
-                key=f"theme_{theme_key}",
-                use_container_width=True,
-                type="primary" if st.session_state.selected_theme == theme_key else "secondary"
-            ):
-                st.session_state.selected_theme = theme_key
-    
-    # Voice options
-    st.session_state.voice_enabled = st.toggle(
-        "🗣️ Enable Tavus Voice", 
-        value=st.session_state.voice_enabled,
-        help="Hear Tavus explain coding concepts with text-to-speech"
-    )
-    
-    # Generate button with animation
-    if st.button("✨ Generate Animation!", use_container_width=True, key="generate", type="primary"):
+    if st.button("✨ Analyze My Story!", use_container_width=True, key="analyze", type="primary"):
         if story_text.strip():
             st.session_state.story_text = story_text
+            with st.spinner("🔍 AI is reading your story to find the perfect hero and world..."):
+                # Extract story elements using AI
+                hero, world = extract_story_elements(story_text, ner_model, classifier)
+                st.session_state.selected_character = hero
+                st.session_state.selected_theme = world
+                st.session_state.analyzed = True
+                
+                # Show suggestions
+                st.success(f"🎯 AI found a {characters[hero]} hero in a {themes[world]} world!")
+        else:
+            st.warning("Please enter a story first!")
+    
+    if st.session_state.analyzed:
+        st.markdown("### 🧙 Step 2: Your Hero & World")
+        
+        st.markdown(f'<div class="hero-suggestion">Your Hero: {characters[st.session_state.selected_character]}</div>', 
+                    unsafe_allow_html=True)
+        st.markdown(f'<div class="world-suggestion">Your World: {themes[st.session_state.selected_theme]}</div>', 
+                    unsafe_allow_html=True)
+        
+        st.info("💡 The AI chose these based on your story! Click Generate Animation when ready.")
+        
+        # Generate animation button
+        if st.button("🎬 Generate Animation!", use_container_width=True, key="generate", type="primary"):
             st.session_state.animation_generated = True
-            
-            with st.spinner("🧙‍♂️ Cooking your story in the magic oven..."):
+            with st.spinner("🧙‍♂️ Creating your animation..."):
                 # Generate animation frames
                 st.session_state.animation_frames = generate_sprite_animation(
                     story_text, 
@@ -541,38 +558,21 @@ with col1:
                 
                 # Update user progress
                 st.session_state.story_count += 1
-                st.session_state.xp += 15
+                st.session_state.xp += 20
                 
                 # Check for level up
                 if st.session_state.xp >= 100:
                     st.session_state.level += 1
                     st.session_state.xp = st.session_state.xp - 100
                     st.session_state.new_level = True
-                
-                # Unlock achievements
-                if st.session_state.story_count == 1:
-                    unlock_achievement("first_story")
-                
-                # Generate speech if enabled
-                if st.session_state.voice_enabled:
-                    st.session_state.speech_data = text_to_speech(st.session_state.code_explanation)
-                    if st.session_state.speech_data:
-                        unlock_achievement("voice_artist")
-                
-                # Simulate sound effect
-                if st.session_state.sound_enabled:
-                    st.session_state.sound_message = "🔊 Sound effects added: Laser blasts, explosions, and more!"
-                else:
-                    st.session_state.sound_message = ""
-        else:
-            st.warning("Please enter a story first!")
+    
     st.markdown('</div>', unsafe_allow_html=True)
 
 with col2:
     st.markdown('<div class="story-box">', unsafe_allow_html=True)
-    st.markdown("### 🎮 Your Game Animation")
     
     if st.session_state.animation_generated and st.session_state.animation_frames:
+        st.markdown("### 🎮 Step 3: Your Animation")
         # Display animation frames
         st.markdown("**Your Story Comes to Life!**")
         cols = st.columns(len(st.session_state.animation_frames))
@@ -599,195 +599,105 @@ with col2:
             use_container_width=True
         )
         
-        # Display sound message
-        if st.session_state.sound_enabled and 'sound_message' in st.session_state:
-            st.info(st.session_state.sound_message)
-        
         # Display character and theme info
         st.success(f"✨ Your {characters[st.session_state.selected_character]} in the {themes[st.session_state.selected_theme]} world!")
         
-        # Play voice explanation
-        if st.session_state.voice_enabled and 'speech_data' in st.session_state and st.session_state.speech_data:
-            st.markdown("### 🔊 Tavus Explanation")
-            st.markdown(f"""
-            <audio controls autoplay style="width:100%">
-                <source src="{st.session_state.speech_data}" type="audio/mp3">
-                Your browser does not support the audio element.
-            </audio>
-            """, unsafe_allow_html=True)
-        
-    elif st.session_state.animation_generated:
-        st.warning("Couldn't generate animation. Try a different story!")
+    elif st.session_state.analyzed:
+        st.markdown("### 🎮 Step 3: Your Animation")
         st.image("https://img.freepik.com/free-vector/hand-drawn-colorful-space-background_23-2148821798.jpg", 
                  use_container_width=True)
+        st.info("👆 Click 'Generate Animation' to bring your story to life!")
     elif story_text:
-        # Show storyboard preview
+        st.markdown("### 🎮 Your Animation Will Appear Here")
         preview_img = create_storyboard_image(story_text)
         st.image(preview_img, caption="Your Story Preview", use_container_width=True)
-        st.info("👆 Click Generate to bring your story to life!")
+        st.info("👆 Click 'Analyze My Story' to begin the magic!")
     else:
         st.image("https://img.freepik.com/free-vector/hand-drawn-colorful-space-background_23-2148821798.jpg", 
                  use_container_width=True)
-        st.info("👆 Write your story and click Generate to see the magic!")
+        st.info("👆 Write your story and click 'Analyze My Story' to begin!")
     
     st.markdown('</div>', unsafe_allow_html=True)
 
 # Tavus explanation section
 if st.session_state.animation_generated and st.session_state.story_text:
     st.markdown('<div class="robot-speech">', unsafe_allow_html=True)
-    st.markdown("### 🤖 Tavus the Robot Teacher says:")
+    st.markdown("### 🤖 Step 4: Tavus the Robot Teacher Explains Coding")
     
     # Extract action words from story
-    action_words = ["zoom", "shoot", "fly", "move", "jump", "run", "attack", "laser", "alien", "spaceship", 
-                   "dragon", "fire", "hero", "sword", "castle", "run", "escape", "fight", "win"]
-    found_actions = [word for word in action_words if word in st.session_state.story_text.lower()]
+    action_words = ["zoom", "shoot", "fly", "move", "jump", "run", "attack", 
+                   "laser", "alien", "spaceship", "dragon", "fire", "hero", 
+                   "sword", "castle", "escape", "fight", "win", "swim", "dive"]
+    found_actions = [word for word in action_words if re.search(r'\b' + word + r'\b', st.session_state.story_text.lower())]
     
     if found_actions:
         # Create explanation based on found words
-        explanations = []
-        code_snippets = []
-        
-        for action in found_actions[:3]:  # Limit to 3 explanations
-            if action == "zoom":
-                explanations.append("makes your spaceship move super fast!")
-                code_snippets.append("spaceship.accelerate(speed=10)")
-            elif action == "shoot":
-                explanations.append("creates laser beams to defeat enemies!")
-                code_snippets.append("laser = spaceship.fire_weapon()")
-            elif action == "fly":
-                explanations.append("keeps your character moving through the air!")
-                code_snippets.append("character.apply_gravity(False)")
-            elif action == "move":
-                explanations.append("changes position on the screen!")
-                code_snippets.append("player.move(direction='right')")
-            elif action == "jump":
-                explanations.append("makes your character leap upwards!")
-                code_snippets.append("hero.jump(height=100)")
-            elif action == "run":
-                explanations.append("makes your character move faster!")
-                code_snippets.append("player.speed = player.speed * 2")
-            elif action == "attack":
-                explanations.append("lets your hero fight the bad guys!")
-                code_snippets.append("sword.swing(damage=15)")
-            elif action == "laser":
-                explanations.append("creates powerful energy beams!")
-                code_snippets.append("weapon = Laser(color='red', damage=20)")
-            elif action == "alien":
-                explanations.append("adds enemies to your game!")
-                code_snippets.append("enemy = Alien(position=(100, 200))")
-            elif action == "spaceship":
-                explanations.append("creates your main character!")
-                code_snippets.append("player = Spaceship(image='spaceship.png')")
-            elif action == "dragon":
-                explanations.append("adds a fire-breathing challenge!")
-                code_snippets.append("boss = Dragon(health=100, damage=25)")
-            elif action == "fire":
-                explanations.append("creates dangerous fire effects!")
-                code_snippets.append("flame = FireEffect(position, size=30)")
-            elif action == "hero":
-                explanations.append("creates the main player character!")
-                code_snippets.append("player = Hero(name='Your Hero')")
-            elif action == "castle":
-                explanations.append("adds a majestic building to your world!")
-                code_snippets.append("castle = Building(type='castle', position=(300, 150))")
-            elif action == "escape":
-                explanations.append("creates exciting chase sequences!")
-                code_snippets.append("player.start_escape_sequence(speed=15)")
-            elif action == "fight":
-                explanations.append("initiates battle mechanics!")
-                code_snippets.append("initiate_combat(enemy)")
-            elif action == "win":
-                explanations.append("creates victory conditions!")
-                code_snippets.append("if player.score > 100: show_victory_screen()")
+        st.markdown("#### 🧩 Story Actions to Code Concepts")
+        cols = st.columns(3)
+        action_colors = ["#ff6b6b", "#4ecdc4", "#ffd166", "#06d6a0", "#118ab2"]
         
-        st.markdown("See how your story became real code? Here's what happened:")
-        
-        for i, (action, explanation, snippet) in enumerate(zip(found_actions, explanations, code_snippets)):
-            st.markdown(f"**{i+1}. When you said '{action}'**:")
-            st.markdown(f'<div class="code-block">{snippet}</div>', unsafe_allow_html=True)
-            st.markdown(f"This code {explanation}")
+        for i, action in enumerate(found_actions[:6]):
+            with cols[i % 3]:
+                color = action_colors[i % len(action_colors)]
+                st.markdown(f'<div style="background:{color}; color:white; border-radius:10px; padding:10px; text-align:center;">', unsafe_allow_html=True)
+                st.markdown(f"**{action.capitalize()}**")
+                
+                # Add code snippets based on action
+                if action == "zoom":
+                    st.code("hero.move_fast(10)", language="python")
+                elif action == "shoot":
+                    st.code("laser = hero.create_weapon()", language="python")
+                elif action == "fly":
+                    st.code("hero.fly(height=100)", language="python")
+                elif action == "move":
+                    st.code("hero.move(direction='right')", language="python")
+                elif action == "jump":
+                    st.code("hero.jump(power=20)", language="python")
+                elif action == "run":
+                    st.code("hero.speed = hero.speed * 2", language="python")
+                elif action == "attack":
+                    st.code("hero.attack(enemy)", language="python")
+                elif action == "swim":
+                    st.code("hero.swim(speed=5)", language="python")
+                elif action == "dive":
+                    st.code("hero.dive(depth=30)", language="python")
+                else:
+                    st.code(f"hero.{action}()", language="python")
+                
+                st.markdown("</div>", unsafe_allow_html=True)
         
-        st.markdown("That's the magic of coding - turning words into actions!")
-    else:
-        st.markdown("I see you created an awesome story! Every word you write can become game code. Try adding action words like 'jump', 'run', or 'shoot' next time!")
+        st.markdown("---")
     
     # Show AI-generated explanation
     st.markdown("### 🧠 AI-Powered Explanation:")
     st.write(st.session_state.code_explanation)
     
+    st.markdown("#### 💡 Remember:")
+    st.markdown("""
+    - Every action in your story becomes code
+    - Code is just instructions for computers
+    - You're already thinking like a coder!
+    """)
+    
     st.markdown("</div>", unsafe_allow_html=True)
 
-# Achievements section
-if st.session_state.achievements_unlocked:
-    st.markdown("### 🏆 Your Achievements")
-    achievement_cols = st.columns(5)
-    for i, ach_id in enumerate(st.session_state.achievements_unlocked):
-        with achievement_cols[i % 5]:
-            ach = ACHIEVEMENTS[ach_id]
-            st.markdown(f'<div class="achievement-badge">{ach["icon"]}</div>', unsafe_allow_html=True)
-            st.markdown(f"**{ach['name']}**")
-            st.caption(ach['description'])
-
-# New achievement notification
-if 'new_achievement' in st.session_state and st.session_state.new_achievement:
-    ach = ACHIEVEMENTS[st.session_state.new_achievement]
-    st.success(f"🎉 Achievement Unlocked: {ach['name']} - {ach['description']}")
-    st.session_state.new_achievement = None
-
 # New level notification
 if 'new_level' in st.session_state and st.session_state.new_level:
     st.balloons()
     st.success(f"🌟 Level Up! You're now Level {st.session_state.level}!")
     st.session_state.new_level = False
 
-# Coding challenges section
-st.markdown("### 💻 Coding Challenges")
-with st.expander("Complete challenges to earn XP!"):
-    challenge_cols = st.columns(3)
-    
-    with challenge_cols[0]:
-        st.markdown("#### Challenge 1: The Mover")
-        st.code("""
-# Make the character move to the right
-character.move_right(10)
-        """)
-        if st.button("Run Challenge 1", key="challenge1"):
-            st.session_state.xp += 5
-            st.success("+5 XP! Character moved right!")
-    
-    with challenge_cols[1]:
-        st.markdown("#### Challenge 2: The Jumper")
-        st.code("""
-# Make the character jump
-character.jump(20)
-        """)
-        if st.button("Run Challenge 2", key="challenge2"):
-            st.session_state.xp += 5
-            st.success("+5 XP! Character jumped!")
-    
-    with challenge_cols[2]:
-        st.markdown("#### Challenge 3: The Shooter")
-        st.code("""
-# Make the character shoot a laser
-laser = character.shoot()
-        """)
-        if st.button("Run Challenge 3", key="challenge3"):
-            st.session_state.xp += 5
-            st.success("+5 XP! Laser fired!")
-
 # Benefits section
 st.markdown("""
-## ❤ Why Everyone Will Love CodeTales
+## ❤ Why Kids & Parents Love CodeTales
 
 | For Kids 👧👦 | For Parents & Teachers 👪👩‍🏫 |
 |--------------|----------------------------|
-| ✨ Feels like playing, not learning | 🧠 Secretly teaches programming concepts |
-| 🚀 Brings imagination to life | 🔍 Develops logical thinking skills |
-| 🎮 Creates personal video games | ➗ Reinforces math fundamentals |
-| 😄 Makes learning fun and exciting | 🧩 Encourages problem-solving abilities |
-| 🌟 100% private with no API keys | 💰 Completely free and open-source |
-| 🏆 Achievement system motivates learning | 📈 Progress tracking shows growth |
-| 🗣️ Voice explanations for accessibility | 🎨 Creative expression through stories |
+| ✨ Create your own animated stories | 🧠 Teaches computational thinking |
+| 🚀 See imagination come to life | 🔍 Develops problem-solving skills |
+| 🎮 Interactive game creation | ➗ Reinforces STEM fundamentals |
+| 😄 Fun characters and worlds | 🧩 Encourages logical reasoning |
+| 🌟 Unlock new levels | 📈 Tracks learning progress |
 """)
 
 # Footer
@@ -795,8 +705,8 @@ st.markdown("---")
 st.markdown("""
 <center>
     <p style="color:white; font-size:1.1rem;">
-        ✨ Made with open-source magic by CodeTales Team ✨<br>
-        Transforming stories into games since 2023 | No API Keys Required!
+        ✨ Made with magic by CodeTales Team ✨<br>
+        Transforming stories into games since 2023
     </p>
 </center>
 """, unsafe_allow_html=True)