Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Game-Based Learning with Free Models
+# app.py - Fixed Version with Free Models
 import streamlit as st
 import os
 import time
@@ -10,8 +10,7 @@ import re
 from PIL import Image
 import io
 import matplotlib.pyplot as plt
-from transformers import pipeline
-import torch
+from transformers import pipeline, set_seed
 
 # Configure Streamlit page
 st.set_page_config(
@@ -201,7 +200,7 @@ def init_session_state():
     if 'game_explanation' not in st.session_state:
         st.session_state.game_explanation = ""
     if 'game_preview' not in st.session_state:
-        st.session_state.game_preview = ""
+        st.session_state.game_preview = None
     if 'active_tab' not in st.session_state:
         st.session_state.active_tab = "story"
     if 'loading' not in st.session_state:
@@ -251,13 +250,12 @@ CONCEPTS = {
     }
 }
 
-# Load models with caching
-@st.cache_resource
+# Load text generation model
+@st.cache_resource(show_spinner=False)
 def load_text_generator():
-    """Load a free text generation model"""
+    """Load a lightweight text generation model"""
     try:
-        # Using a smaller model for better performance
-        return pipeline("text-generation", model="gpt2")
+        return pipeline('text-generation', model='gpt2', framework='pt', device=-1)
     except:
         return None
 
@@ -292,104 +290,67 @@ def analyze_story(story):
 # Generate game scenario using free models
 def generate_game_scenario(story, concepts):
     """Generate a game scenario based on the story and concepts"""
-    try:
-        # Create prompt
-        prompt = (
-            f"Create a children's educational game based on this story: {story}\n"
-            f"Game should teach these programming concepts: {', '.join(concepts)}\n"
-            "Structure response with:\n"
-            "Game Title: ...\n"
-            "Game Objective: ...\n"
-            "Characters: ...\n"
-            "Game Mechanics: ...\n"
-            "Coding Concepts: ...\n"
-            "Visual Description: ...\n"
-        )
-        
-        # Generate with free model
-        generator = load_text_generator()
-        if generator:
-            result = generator(
-                prompt, 
-                max_length=400, 
-                do_sample=True, 
-                temperature=0.7,
-                num_return_sequences=1
-            )
-            return result[0]['generated_text']
-    except Exception as e:
-        st.error(f"Text generation error: {str(e)}")
-    
-    # Fallback scenario
-    concept_names = [CONCEPTS[c]["name"] for c in concepts]
-    return (
-        f"Game Title: {story[:20]} Adventure\n\n"
-        f"Game Objective: Complete challenges based on the story: {story[:100]}...\n\n"
-        "Characters: Hero character based on the story\n\n"
-        "Game Mechanics: Navigate through levels, collect items, solve puzzles\n\n"
-        f"Coding Concepts: This game teaches {', '.join(concept_names)} through gameplay\n\n"
-        "Visual Description: Colorful 3D world with characters and obstacles"
-    )
+    # Create a simple template-based scenario
+    concept_names = [CONCEPTS[c]['name'] for c in concepts]
+    concept_list = ", ".join(concept_names)
+    
+    return f"""
+Game Title: {story[:15]} Adventure
+Game Objective: Complete challenges based on the story: {story[:100]}...
+Characters: 
+- Hero: The main character from your story
+- Helper: A friendly guide who explains coding concepts
+- Villain: A character that creates obstacles (if applicable)
+
+Game Mechanics:
+1. The player controls the hero using arrow keys
+2. Collect items mentioned in the story
+3. Avoid obstacles and solve simple puzzles
+4. Helper characters appear to teach {concept_list}
+
+Coding Concepts: This game teaches {concept_list} through:
+- Using loops to repeat actions
+- Making decisions with conditionals
+- Creating reusable functions
+- Tracking progress with variables
+- Managing collections with lists
+
+Visual Description: Colorful 3D world with cartoon-style characters, vibrant landscapes, and magical effects.
+"""
 
 # Generate game code explanation
 def generate_game_explanation(story, concepts, game_scenario):
     """Generate explanation of game code"""
-    try:
-        # Create prompt
-        prompt = (
-            f"Explain how a game based on this story teaches programming concepts to children: {story}\n"
-            f"Game Scenario: {game_scenario}\n"
-            f"Concepts: {', '.join(concepts)}\n"
-            "Use simple language and analogies for kids aged 6-12."
-        )
-        
-        # Generate with free model
-        generator = load_text_generator()
-        if generator:
-            result = generator(
-                prompt, 
-                max_length=300, 
-                do_sample=True, 
-                temperature=0.7,
-                num_return_sequences=1
-            )
-            return result[0]['generated_text']
-    except:
-        pass
-    
-    # Fallback explanation
+    # Create simple explanation based on concepts
     concept_explanations = "\n".join(
         [f"- {CONCEPTS[c]['name']}: {CONCEPTS[c]['game_example']}" for c in concepts]
     )
-    return (
-        f"In this game based on your story '{story[:20]}...', we use programming concepts to make it work:\n"
-        f"{concept_explanations}\n\n"
-        "The code brings your story to life in a 3D game world!"
-    )
+    return f"""
+In this game based on your story "{story[:20]}...", we use programming concepts to make it work:
+
+{concept_explanations}
+
+As you play the game, think about:
+1. How the game uses these concepts to create challenges
+2. How you might change the code to make the game easier or harder
+
+The code brings your story to life in a 3D game world!
+"""
 
 # Generate simple game code
 def generate_game_code(story, concepts):
     """Generate simple PyGame code for the game"""
-    # Extract game elements from story
-    character = "hero"
-    goal = "treasure"
-    obstacles = ["rock", "tree", "river"]
-    
-    if "knight" in story.lower():
-        character = "knight"
-        goal = "dragon"
-    elif "space" in story.lower():
-        character = "astronaut"
-        goal = "planet"
-    elif "jungle" in story.lower():
-        character = "explorer"
-        goal = "temple"
-    
-    # Generate dynamic game code
+    # Generate a unique game based on story keywords
+    keywords = re.findall(r'\b\w{4,}\b', story)[:3]
+    player_char = keywords[0] if keywords else "hero"
+    collect_item = keywords[1] if len(keywords) > 1 else "star"
+    obstacle = keywords[2] if len(keywords) > 2 else "rock"
+    
     return f"""
 # {story[:30]} Adventure Game
 import pygame
 import random
+import sys
 
 # Initialize pygame
 pygame.init()
@@ -397,14 +358,15 @@ pygame.init()
 # Game setup
 WIDTH, HEIGHT = 800, 600
 screen = pygame.display.set_mode((WIDTH, HEIGHT))
-pygame.display.set_caption("{story[:20]} Adventure")
+pygame.display.set_caption("{player_char.capitalize()} Adventure")
 clock = pygame.time.Clock()
 
 # Colors
 BACKGROUND = (173, 216, 230)  # Light blue
 PLAYER_COLOR = (255, 0, 0)     # Red
-GOAL_COLOR = (0, 255, 0)       # Green
+GOAL_COLOR = (255, 223, 0)     # Gold
 OBSTACLE_COLOR = (139, 69, 19) # Brown
+TEXT_COLOR = (0, 0, 0)         # Black
 
 # Player setup
 player_size = 50
@@ -481,28 +443,29 @@ while running:
                     (goal_x, goal_y, goal_size, goal_size))
     
     # Display score
-    score_text = font.render(f"Stars: {{score}}", True, (0, 0, 0))
+    score_text = font.render(f"{collect_item.capitalize()}s: {{score}}", True, TEXT_COLOR)
     screen.blit(score_text, (20, 20))
     
-    # Display character type
-    char_text = font.render(f"Character: {character.title()}", True, (0, 0, 0))
-    screen.blit(char_text, (20, 60))
+    # Display story title
+    title_text = font.render(f"{player_char.capitalize()} Adventure", True, TEXT_COLOR)
+    screen.blit(title_text, (WIDTH // 2 - 100, 20))
+    
+    # Display instructions
+    help_text = font.render("Arrow keys to move - Collect the gold squares!", True, TEXT_COLOR)
+    screen.blit(help_text, (WIDTH // 2 - 200, HEIGHT - 40))
     
     # Update display
     pygame.display.flip()
     clock.tick(60)
 
 pygame.quit()
+sys.exit()
 """
 
 # Generate game preview visualization
-def generate_game_preview(game_scenario):
+def generate_game_preview():
     """Generate a visual preview of the game"""
     try:
-        # Extract title from game scenario
-        title_match = re.search(r"Game Title: (.+)", game_scenario)
-        title = title_match.group(1) if title_match else "Your Adventure"
-        
         # Create a simple visualization
         fig, ax = plt.subplots(figsize=(10, 6))
         ax.set_facecolor('#a2d2ff')
@@ -510,33 +473,37 @@ def generate_game_preview(game_scenario):
         ax.set_ylim(0, 6)
         
         # Draw game elements
-        ax.text(5, 5, title, fontsize=20, ha='center', color='#9b5de5')
+        ax.text(5, 5, "Your Adventure", fontsize=20, ha='center', color='#9b5de5')
         ax.plot([1, 9], [3, 3], 'k-', linewidth=2)  # Ground
         
         # Player character
         ax.plot(2, 3.5, 'ro', markersize=15)
+        ax.text(2, 4, 'Player', ha='center')
         
         # Goal
-        ax.plot(8, 3.5, 'go', markersize=15)
+        ax.plot(8, 3.5, 'yo', markersize=15)
+        ax.text(8, 4, 'Goal', ha='center')
         
         # Obstacles
         for i in range(3):
             x = random.uniform(3, 7)
             y = random.uniform(3.2, 4)
             ax.plot(x, y, 's', color='#8d99ae', markersize=12)
+            ax.text(x, y+0.3, 'Obstacle', ha='center', fontsize=8)
         
         # Path
         ax.plot([2, 8], [3.5, 3.5], 'y--', linewidth=2)
         
         ax.axis('off')
-        ax.set_title("Game Preview", fontsize=16)
+        ax.set_title("Game Preview - Move player to collect goals while avoiding obstacles", fontsize=14)
         
         # Save to bytes
         buf = io.BytesIO()
         plt.savefig(buf, format='png', dpi=100, bbox_inches='tight')
         buf.seek(0)
         return buf
-    except:
+    except Exception as e:
+        st.error(f"Preview generation error: {str(e)}")
         return None
 
 # Main application function
@@ -568,7 +535,7 @@ def main():
             st.session_state.game_scenario = ""
             st.session_state.game_code = ""
             st.session_state.game_explanation = ""
-            st.session_state.game_preview = ""
+            st.session_state.game_preview = None
             st.session_state.active_tab = "story"
     st.markdown('</div>', unsafe_allow_html=True)
     
@@ -612,9 +579,7 @@ def main():
                         )
                     
                     with st.spinner("🖼️ Generating game preview..."):
-                        preview = generate_game_preview(st.session_state.game_scenario)
-                        if preview:
-                            st.session_state.game_preview = preview
+                        st.session_state.game_preview = generate_game_preview()
                     
                     st.session_state.active_tab = "game"
                     st.session_state.loading = False
@@ -660,7 +625,7 @@ def main():
         if st.session_state.game_preview:
             st.image(st.session_state.game_preview, use_container_width=True)
         else:
-            st.warning("Game preview couldn't be generated. Showing example instead.")
+            st.info("Game preview visualization")
             st.image("https://i.imgur.com/7zQY1eE.gif", use_container_width=True)
         
         # Game explanation
@@ -751,7 +716,7 @@ def main():
             <div class="game-card">
             <ul>
                 <li>Move your character (red square) with arrow keys</li>
-                <li>Collect the green stars to increase your score</li>
+                <li>Collect the gold squares to increase your score</li>
                 <li>Avoid the brown obstacles</li>
                 <li>See how programming concepts make the game work!</li>
             </ul>