Update app.py

app.py

@@ -1,331 +1,107 @@
-# app.py
-import streamlit as st
-import requests
-import time
 import os
+import streamlit as st
 from dotenv import load_dotenv
+from novita_client import NovitaClient
+import ast
+import subprocess
+import tempfile
+import time
 
-# Load environment variables
 load_dotenv()
 
-# Set up the page
-st.set_page_config(
-    page_title="CodeTales ✨",
-    page_icon="🚀",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
+# Initialize Novita API client
+class StoryToCodeConverter:
+    def __init__(self):
+        api_key = os.getenv("NOVITA_API_KEY")
+        if not api_key:
+            raise ValueError("NOVITA_API_KEY not found in environment")
+        self.client = NovitaClient(api_key)
 
-# Custom CSS
-st.markdown("""
-<style>
-@import url('https://fonts.googleapis.com/css2?family=Comic+Neue:wght@700&display=swap');
-            
-.stApp {
-    background: linear-gradient(135deg, #6e8efb, #a777e3);
-}
-            
-.header {
-    font-family: 'Comic Neue', cursive;
-    color: white;
-    text-align: center;
-    font-size: 3.5rem;
-    text-shadow: 3px 3px 0 #000;
-    padding: 1rem;
-}
-            
-.subheader {
-    font-family: 'Comic Neue', cursive;
-    color: #ffeb3b;
-    text-align: center;
-    font-size: 1.8rem;
-    margin-bottom: 2rem;
-}
-            
-.story-box {
-    background-color: rgba(255, 255, 255, 0.9);
-    border-radius: 20px;
-    padding: 2rem;
-    box-shadow: 0 8px 16px rgba(0,0,0,0.2);
-    margin-bottom: 2rem;
-}
-            
-.robot-speech {
-    background-color: #4caf50;
-    color: white;
-    border-radius: 20px;
-    padding: 1.5rem;
-    font-size: 1.2rem;
-    position: relative;
-    margin-top: 2rem;
-}
-            
-.robot-speech:after {
-    content: '';
-    position: absolute;
-    top: -20px;
-    left: 50px;
-    border: 10px solid transparent;
-    border-bottom-color: #4caf50;
-}
-            
-.generate-btn {
-    background: #ff5722 !important;
-    color: white !important;
-    font-weight: bold !important;
-    font-size: 1.2rem !important;
-    padding: 0.7rem 2rem !important;
-    border-radius: 50px !important;
-    border: none !important;
-    box-shadow: 0 4px 8px rgba(0,0,0,0.3) !important;
-    transition: all 0.3s !important;
-    margin-top: 1rem;
-}
-            
-.generate-btn:hover {
-    transform: scale(1.05) !important;
-    box-shadow: 0 6px 12px rgba(0,0,0,0.4) !important;
-}
-            
-.code-block {
-    background: #2d2d2d;
-    color: #f8f8f2;
-    padding: 1rem;
-    border-radius: 10px;
-    font-family: monospace;
-    font-size: 1.1rem;
-    margin: 1rem 0;
-    overflow-x: auto;
-}
-</style>
-""", unsafe_allow_html=True)
+    def generate_code(self, story):
+        prompt = f"""
+Convert this children's story into Python animation code using turtle graphics.
+Add code comments that explain the basic concepts to kids, like loops, movement, and colors.
 
-# Novita.ai API functions
-def generate_animation(prompt):
-    """Generate animation using Novita.ai API"""
-    api_key = os.getenv("NOVITA_API_KEY")
-    if not api_key:
-        st.error("Novita.ai API key not found. Please add it to your .env file.")
-        return None
-    
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-    
-    payload = {
-        "prompt": prompt,
-        "model": "animov-0.1.1",
-        "width": 512,
-        "height": 512,
-        "fps": 8,
-        "seconds": 4
-    }
-    
-    try:
-        response = requests.post(
-            "https://api.novita.ai/v1/video",
-            headers=headers,
-            json=payload
-        )
-        
-        if response.status_code == 200:
-            task_id = response.json()["task_id"]
-            return task_id
-        else:
-            st.error(f"API Error: {response.status_code} - {response.text}")
-            return None
-    except Exception as e:
-        st.error(f"Connection Error: {str(e)}")
-        return None
+Story: {story}
 
-def get_animation_status(task_id):
-    """Check status of animation generation"""
-    api_key = os.getenv("NOVITA_API_KEY")
-    headers = {"Authorization": f"Bearer {api_key}"}
-    
-    try:
-        response = requests.get(
-            f"https://api.novita.ai/v1/video/{task_id}",
-            headers=headers
+Only return the Python code wrapped in triple backticks like ```python ... ```
+"""
+        response = self.client.chat_completions.create(
+            model="gpt-4",
+            messages=[{"role": "user", "content": prompt}]
         )
-        
-        if response.status_code == 200:
-            return response.json()
-        else:
-            return None
-    except:
-        return None
+        content = response.choices[0].message.content
+        return self._extract_code(content)
 
-# 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 AI magic!</div>', unsafe_allow_html=True)
+    def _extract_code(self, content):
+        if "```python" in content:
+            return content.split("```python")[1].split("```")[0]
+        return content
 
-# 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 (like translating English to Computer)
-    
-    **3. Animation Pops Out (The Cake!) 🎮**  
-    See your spaceship flying and shooting instantly!
-    
-    **4. Robot Teacher Explains 🤖**  
-    Tavus shows: *"See? 'spaceship.move_right()' makes it fly! That's coding!"*
-    """)
 
-# Initialize session state
-if 'task_id' not in st.session_state:
-    st.session_state.task_id = None
-if 'video_url' not in st.session_state:
-    st.session_state.video_url = None
-if 'generating' not in st.session_state:
-    st.session_state.generating = False
-if 'story_text' not in st.session_state:
-    st.session_state.story_text = ""
+# Safe execution
+class CodeExecutor:
+    def execute_safely(self, code):
+        try:
+            ast.parse(code)  # Safety check
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".py") as tmp:
+                tmp.write(code.encode("utf-8"))
+                tmp_path = tmp.name
+            subprocess.Popen(["python", tmp_path])
+            return True
+        except Exception as e:
+            st.error(f"Robot Chef burned the cake: {e}")
+            return False
 
-# Main content
-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_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=st.session_state.story_text
-    )
-    
-    # Generate button with animation
-    if st.button("✨ Generate Animation!", use_container_width=True, key="generate", type="primary"):
-        if story_text.strip():
-            st.session_state.story_text = story_text
-            st.session_state.generating = True
-            st.session_state.task_id = generate_animation(story_text)
-            st.session_state.video_url = None
-            st.rerun()
-        else:
-            st.warning("Please enter a story first!")
-    st.markdown('</div>', unsafe_allow_html=True)
+# ---------- Streamlit UI ----------
+st.set_page_config(page_title="CodeTales", page_icon="✨")
 
-with col2:
-    st.markdown('<div class="story-box">', unsafe_allow_html=True)
-    st.markdown("### 🎮 Your Game Animation")
-    
-    if st.session_state.generating and st.session_state.task_id:
-        with st.spinner("🧙‍♂️ Cooking your story in the magic oven..."):
-            status = get_animation_status(st.session_state.task_id)
-            
-            if status and status.get("status") == "SUCCESS":
-                st.session_state.video_url = status["video_url"]
-                st.session_state.generating = False
-                st.rerun()
-            elif status and status.get("status") in ["FAILED", "CANCELLED"]:
-                st.error("Oh no! The magic oven malfunctioned. Try again with a different story!")
-                st.session_state.generating = False
-            else:
-                # Still processing
-                time.sleep(3)
-                st.rerun()
-    
-    if st.session_state.video_url:
-        st.video(st.session_state.video_url)
-        st.success("🎉 Your animation is ready! Tavus will explain the magic now!")
-    elif st.session_state.story_text:
-        st.image("https://img.freepik.com/free-vector/hand-drawn-colorful-space-background_23-2148821798.jpg", 
-                 use_column_width=True)
-        st.info("👆 Click Generate to see your story come to life!")
-    else:
-        st.image("https://img.freepik.com/free-vector/hand-drawn-colorful-space-background_23-2148821798.jpg", 
-                 use_column_width=True)
-        st.info("👆 Write your story and click Generate to see the magic!")
-    
-    st.markdown('</div>', unsafe_allow_html=True)
+with st.sidebar:
+    st.header("🍰 How It Works")
+    st.markdown("""
+1. 📝 Kids write a story (like a recipe!)  
+2. 🔮 AI turns it into **Python code**  
+3. 🎮 Animation appears like magic!  
+4. 🤖 Robot Teacher explains the code!
+
+Made for ages 6–12. Fun, visual, and interactive!
+""")
+
+st.title("✨ CodeTales: Storytime + Coding Magic")
 
-# Tavus explanation section
-if st.session_state.video_url and st.session_state.story_text:
-    st.markdown('<div class="robot-speech">', unsafe_allow_html=True)
-    st.markdown("### 🤖 Tavus the Robot Teacher says:")
-    
-    # Extract action words from story
-    action_words = ["zoom", "shoot", "fly", "move", "jump", "run", "attack", "laser", "alien", "spaceship"]
-    found_actions = [word for word in action_words if word in 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')")
-        
-        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}")
-        
-        st.markdown("That's the magic of coding - turning words into actions!")
+story = st.text_area("Write Your Story Recipe 📝", height=200,
+                     placeholder="Once upon a time, a rocket zoomed past rainbow stars and met an alien...")
+
+if st.button("✨ Bake My Story Cake!"):
+    if not story.strip():
+        st.warning("Please write a story first!")
     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("</div>", unsafe_allow_html=True)
+        with st.spinner("Baking your story cake... 🍰"):
+            converter = StoryToCodeConverter()
+            try:
+                code = converter.generate_code(story)
+            except Exception as e:
+                st.error(f"AI oven broke down! {e}")
+                st.stop()
 
-# Benefits section
-st.markdown("""
-## ❤ Why Everyone Will Love CodeTales
+            st.subheader("📜 Your Code Recipe")
+            st.code(code, language='python')
 
-| 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 |
-""")
+            executor = CodeExecutor()
+            if executor.execute_safely(code):
+                st.balloons()
+                with st.expander("🤖 Robot Teacher Says:"):
+                    st.markdown("""
+- `turtle.forward(100)` – Moves the rocket forward 🚀  
+- `turtle.color("red")` – Paints with colors 🎨  
+- `for loop` – Repeats an action like magic 🔁  
+                    """)
+                st.success("🎉 Your story came to life!")
+            else:
+                st.error("Oops! Couldn't bake the animation.")
+else:
+    st.caption("Turn your words into animated code with magic 🪄")
 
-# Footer
 st.markdown("---")
-st.markdown("""
-<center>
-    <p style="color:white; font-size:1.1rem;">
-        ✨ Made with magic by CodeTales Team ✨<br>
-        Transforming stories into games since 2023
-    </p>
-</center>
-""", unsafe_allow_html=True)
+st.caption("Made with ❤️ for young coders | Powered by Novita + Python Turtle")