Update app.py

app.py

@@ -1,18 +1,15 @@
-# app.py - Final Deployment-Ready Version
+# app.py - Final Working Version
 import streamlit as st
 import os
 import time
-import base64
 import random
-import tempfile
-import subprocess
 import numpy as np
-import pygame
-import sys
-import imageio
+import matplotlib.pyplot as plt
+from matplotlib.animation import FuncAnimation
+from matplotlib import rc
+import base64
 from PIL import Image
 import io
-import matplotlib.pyplot as plt
 
 # Configure Streamlit page
 st.set_page_config(
@@ -131,6 +128,13 @@ st.markdown("""
         overflow: hidden;
         margin: 0 auto;
         display: block;
+        max-width: 100%;
+    }
+    
+    .character {
+        font-size: 48px;
+        text-align: center;
+        margin: 10px 0;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -175,16 +179,16 @@ CONCEPTS = {
 }
 
 # Character database
-CHARACTERS = {
-    "rabbit": {"color": (255, 150, 150), "speed": 5, "size": 30},
-    "dragon": {"color": (255, 100, 100), "speed": 3, "size": 60},
-    "cat": {"color": (200, 150, 255), "speed": 6, "size": 25},
-    "dog": {"color": (255, 200, 150), "speed": 7, "size": 35},
-    "knight": {"color": (150, 200, 255), "speed": 4, "size": 40},
-    "wizard": {"color": (200, 255, 150), "speed": 4, "size": 35},
-    "scientist": {"color": (150, 255, 200), "speed": 5, "size": 30},
-    "pirate": {"color": (255, 255, 150), "speed": 5, "size": 40}
-}
+CHARACTERS = [
+    {"name": "rabbit", "emoji": "🐰", "color": "#FFB6C1"},
+    {"name": "dragon", "emoji": "🐉", "color": "#FF6347"},
+    {"name": "cat", "emoji": "🐱", "color": "#DDA0DD"},
+    {"name": "dog", "emoji": "🐶", "color": "#FFD700"},
+    {"name": "knight", "emoji": "🤺", "color": "#87CEEB"},
+    {"name": "wizard", "emoji": "🧙", "color": "#98FB98"},
+    {"name": "scientist", "emoji": "🔬", "color": "#20B2AA"},
+    {"name": "pirate", "emoji": "🏴‍☠️", "color": "#FFA500"}
+]
 
 def analyze_story(story):
     """Analyze story and identify programming concepts"""
@@ -213,101 +217,87 @@ def analyze_story(story):
     
     return list(set(detected_concepts))
 
-def generate_pygame_animation(story, concepts):
-    """Generate PyGame animation based on story and concepts"""
+def extract_count_from_story(story):
+    """Extract a number from the story to use in animations"""
+    for word in story.split():
+        if word.isdigit():
+            return min(int(word), 10)
+    return 3  # Default value
+
+def create_animation(story, concepts):
+    """Create a matplotlib animation based on the story and concepts"""
     try:
         # Choose a random character
-        character = random.choice(list(CHARACTERS.keys()))
-        char_details = CHARACTERS[character]
-        
-        # Extract count from story
-        count = 3
-        for word in story.split():
-            if word.isdigit():
-                count = min(int(word), 10)
-                break
-        
-        # Create a temporary file for the animation
-        with tempfile.NamedTemporaryFile(suffix=".gif", delete=False) as tmpfile:
-            gif_path = tmpfile.name
-        
-        # Set up PyGame in headless mode
-        os.environ['SDL_VIDEODRIVER'] = 'dummy'
-        pygame.init()
-        pygame.display.set_mode((1, 1))  # Tiny invisible window
-        
-        # Animation settings
-        WIDTH, HEIGHT = 800, 600
-        screen = pygame.Surface((WIDTH, HEIGHT))
-        clock = pygame.time.Clock()
-        frames = []
-        
-        # Colors
-        BACKGROUND = (25, 25, 50)
-        GROUND = (40, 100, 40)
-        SUN = (255, 255, 200)
-        
-        # Character setup
-        char_x = 100
-        char_y = HEIGHT - 100
-        target_x = 700
-        target_y = HEIGHT - 100
-        hops = 0
-        hop_height = 0
-        
-        # Animation loop
-        for frame in range(120):
-            # Clear screen
-            screen.fill(BACKGROUND)
-            
-            # Draw ground
-            pygame.draw.rect(screen, GROUND, (0, HEIGHT - 50, WIDTH, 50))
-            
-            # Draw sun
-            pygame.draw.circle(screen, SUN, (700, 100), 50)
-            
-            # Draw target
-            pygame.draw.circle(screen, (255, 200, 100), (target_x, target_y), 30)
-            pygame.draw.circle(screen, (255, 150, 50), (target_x, target_y), 20)
-            pygame.draw.circle(screen, (255, 100, 0), (target_x, target_y), 10)
+        character = random.choice(CHARACTERS)
+        count = extract_count_from_story(story)
+        
+        # Create figure and axis
+        fig, ax = plt.subplots(figsize=(10, 6), facecolor='#f0f8ff')
+        ax.set_xlim(0, 10)
+        ax.set_ylim(0, 10)
+        ax.axis('off')
+        
+        # Add title
+        fig.suptitle('Your Story Animation', fontsize=20, color='purple', fontweight='bold')
+        
+        # Add story text
+        ax.text(5, 9, f'"{story[:50]}{"..." if len(story) > 50 else ""}"', 
+                fontsize=14, ha='center', wrap=True, color='#333')
+        
+        # Character position
+        x_pos = 5
+        y_pos = 6
+        
+        # Create character element
+        char_text = ax.text(x_pos, y_pos, character["emoji"], 
+                          fontsize=48, ha='center', color=character["color"])
+        
+        # Target position
+        target_x = 8
+        target_y = 6
+        ax.plot(target_x, target_y, 'ro', markersize=15, alpha=0.5)
+        ax.text(target_x, target_y - 0.8, 'Target', ha='center', fontsize=12)
+        
+        # Add hop counter
+        counter_text = ax.text(2, 8, f'Hops: 0/{count}', fontsize=16, color='#333')
+        
+        # Animation function
+        def animate(frame):
+            nonlocal x_pos, y_pos
             
-            # Move character
-            if hops < count:
-                char_x += char_details["speed"]
+            # Move character toward target
+            if frame < count:
+                x_pos += 0.3
+                
                 # Hop animation
-                hop_height = 50 * np.sin(frame * 0.2)
-                char_y = HEIGHT - 100 - abs(hop_height)
+                if frame % 2 == 0:
+                    y_pos = 6.5
+                else:
+                    y_pos = 6
                 
-                # Check if target reached
-                if char_x >= target_x - 40:
-                    hops += 1
-                    char_x = 100
-            
-            # Draw character
-            pygame.draw.circle(screen, char_details["color"], (int(char_x), int(char_y)), char_details["size"])
+                char_text.set_position((x_pos, y_pos))
+                counter_text.set_text(f'Hops: {frame+1}/{count}')
             
-            # Draw eyes
-            pygame.draw.circle(screen, (255, 255, 255), (int(char_x + 10), int(char_y - 5)), 8)
-            pygame.draw.circle(screen, (0, 0, 0), (int(char_x + 10), int(char_y - 5)), 4)
+            # Return to start after reaching target
+            elif frame == count:
+                x_pos = 5
+                y_pos = 6
+                char_text.set_position((x_pos, y_pos))
             
-            # Draw hop counter
-            font = pygame.font.SysFont(None, 36)
-            text = font.render(f"Hops: {hops}/{count}", True, (255, 255, 255))
-            screen.blit(text, (20, 20))
-            
-            # Draw story text
-            story_text = font.render(story[:40] + ("..." if len(story) > 40 else ""), True, (200, 200, 255))
-            screen.blit(story_text, (WIDTH//2 - story_text.get_width()//2, 50))
-            
-            # Capture frame
-            frame_data = pygame.surfarray.array3d(screen).swapaxes(0, 1)
-            frames.append(frame_data)
-            
-            clock.tick(30)
+            return char_text, counter_text
+        
+        # Create animation
+        ani = FuncAnimation(fig, animate, frames=count*2, interval=500, blit=True)
+        
+        # Save animation as GIF in memory
+        buf = io.BytesIO()
+        ani.save(buf, format='gif', writer='pillow', fps=2)
+        buf.seek(0)
         
-        # Save as GIF
-        imageio.mimsave(gif_path, frames, fps=30)
-        return gif_path
+        # Close the figure to free memory
+        plt.close(fig)
+        
+        return buf
     
     except Exception as e:
         st.error(f"Animation error: {str(e)}")
@@ -343,7 +333,7 @@ def create_story_image(story):
             lines.append(' '.join(current_line))
         
         for i, line in enumerate(lines):
-            ax.text(5, 6.5 - i*0.7, line, fontsize=14, ha='center')
+            ax.text(5, 6.5 - i*0.7, line, fontsize=14, ha='center', color='#333')
         
         # Add decoration
         ax.text(2, 2, '🐰', fontsize=40, ha='center')
@@ -353,12 +343,11 @@ def create_story_image(story):
         
         # Save to buffer
         buf = io.BytesIO()
-        plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0.5)
+        plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0.5, dpi=100)
         buf.seek(0)
-        image = Image.open(buf)
         plt.close()
-        
-        return image
+        return buf
+    
     except Exception as e:
         st.error(f"Image generation error: {str(e)}")
         return None
@@ -373,8 +362,8 @@ def main():
         st.session_state.story = ""
     if 'concepts' not in st.session_state:
         st.session_state.concepts = []
-    if 'animation_path' not in st.session_state:
-        st.session_state.animation_path = None
+    if 'animation' not in st.session_state:
+        st.session_state.animation = None
     if 'active_tab' not in st.session_state:
         st.session_state.active_tab = "story"
     
@@ -397,7 +386,7 @@ def main():
         if st.button("🔄 Reset"):
             st.session_state.story = ""
             st.session_state.concepts = []
-            st.session_state.animation_path = None
+            st.session_state.animation = None
             st.session_state.active_tab = "story"
     
     # Story creation tab
@@ -422,8 +411,8 @@ def main():
                     with st.spinner("🧠 Analyzing your story for coding concepts..."):
                         st.session_state.concepts = analyze_story(story)
                     
-                    with st.spinner("🎬 Creating your animation (this may take a moment)..."):
-                        st.session_state.animation_path = generate_pygame_animation(
+                    with st.spinner("🎬 Creating your animation..."):
+                        st.session_state.animation = create_animation(
                             story, st.session_state.concepts
                         )
                     
@@ -447,7 +436,7 @@ def main():
     elif st.session_state.active_tab == "animation":
         st.header("🎬 Your Story Animation")
         
-        if not st.session_state.animation_path:
+        if not st.session_state.story:
             st.warning("Please create a story first!")
             st.session_state.active_tab = "story"
             st.rerun()
@@ -459,13 +448,13 @@ def main():
         </div>
         """, unsafe_allow_html=True)
         
-        try:
-            st.image(st.session_state.animation_path, use_container_width=True)
-        except Exception as e:
-            st.error(f"Couldn't display animation: {str(e)}")
-            story_image = create_story_image(st.session_state.story)
-            if story_image:
-                st.image(story_image, use_container_width=True)
+        if st.session_state.animation:
+            st.image(st.session_state.animation, use_container_width=True)
+        else:
+            st.warning("Animation couldn't be generated. Showing story image instead.")
+            story_img = create_story_image(st.session_state.story)
+            if story_img:
+                st.image(story_img, use_container_width=True)
         
         st.success("✨ Animation created successfully!")
         st.caption("This animation was generated with Python code based on your story!")
@@ -506,42 +495,65 @@ def main():
         st.write("Here's the Python code that created your animation:")
         
         # Sample code (in a real app, you would generate this based on the story)
-        sample_code = """
-# Story: {story}
+        count = extract_count_from_story(st.session_state.story)
+        character = random.choice(CHARACTERS)
+        
+        sample_code = f"""
+# Story: {st.session_state.story[:50]}{'...' if len(st.session_state.story) > 50 else ''}
 
-import pygame
+import matplotlib.pyplot as plt
+from matplotlib.animation import FuncAnimation
 import numpy as np
-import imageio
 
-def create_animation(story):
-    # Setup PyGame
-    pygame.init()
-    WIDTH, HEIGHT = 800, 600
-    screen = pygame.Surface((WIDTH, HEIGHT))
-    
-    # Animation parameters based on story
-    character = "{character}"
-    count = {count}
-    
-    # Animation loop
-    frames = []
-    for frame in range(120):
-        # Draw background, character, etc.
-        # ... animation code ...
-        
-        # Capture frame
-        frame_data = pygame.surfarray.array3d(screen)
-        frames.append(frame_data)
-    
-    # Save animation
-    imageio.mimsave('animation.gif', frames, fps=30)
+# Setup the figure
+fig, ax = plt.subplots(figsize=(10, 6))
+ax.set_xlim(0, 10)
+ax.set_ylim(0, 10)
+ax.axis('off')
 
-create_animation("{story}")
-        """.format(
-            story=st.session_state.story[:50] + ("..." if len(st.session_state.story) > 50 else ""),
-            character=random.choice(list(CHARACTERS.keys())),
-            count=min([int(word) for word in st.session_state.story.split() if word.isdigit()] or [3])
-        )
+# Add story text
+ax.text(5, 9, "{st.session_state.story[:50]}{'...' if len(st.session_state.story) > 50 else ''}", 
+        fontsize=14, ha='center', wrap=True)
+
+# Add character
+char = ax.text(5, 6, "{character['emoji']}", fontsize=48, ha='center', color='{character['color']}')
+
+# Add target
+ax.plot(8, 6, 'ro', markersize=15, alpha=0.5)
+ax.text(8, 5.2, 'Target', ha='center', fontsize=12)
+
+# Add counter
+counter = ax.text(2, 8, 'Hops: 0/{count}', fontsize=16)
+
+# Animation function
+def animate(frame):
+    x = char.get_position()[0]
+    
+    # Move character toward target
+    if frame < {count}:
+        new_x = x + 0.3
+        
+        # Hop animation
+        if frame % 2 == 0:
+            new_y = 6.5
+        else:
+            new_y = 6
+            
+        char.set_position((new_x, new_y))
+        counter.set_text(f'Hops: {{frame+1}}/{count}')
+    
+    # Return to start after reaching target
+    elif frame == {count}:
+        char.set_position((5, 6))
+    
+    return char, counter
+
+# Create and save animation
+ani = FuncAnimation(fig, animate, frames={count*2}, interval=500, blit=True)
+ani.save('animation.gif', writer='pillow', fps=2)
+
+plt.close()
+        """
         
         st.code(sample_code, language="python")