Update app.py

app.py

@@ -1,81 +1,331 @@
-import os
+# app.py
 import streamlit as st
-from novita_client import NovitaClient
-from dotenv import load_dotenv
-import ast
+import requests
 import time
+import os
+from dotenv import load_dotenv
 
+# Load environment variables
 load_dotenv()
 
-class StoryToCodeConverter:
-    def __init__(self):
-        self.client = NovitaClient(os.getenv('NOVITA_API_KEY'))
+# Set up the page
+st.set_page_config(
+    page_title="CodeTales ✨",
+    page_icon="🚀",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# 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. 
-        Include code comments that explain basic programming concepts to kids. Story: {story}
+# 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
+        )
         
-        Return ONLY the Python code wrapped in triple backticks."""
-        response = self_client.chat_completions.create(
-            model="gpt-4",
-            messages=[{"role": "user", "content": prompt}]
+        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
+
+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
         )
-        return self._extract_code(response.choices[0].message.content)
+        
+        if response.status_code == 200:
+            return response.json()
+        else:
+            return None
+    except:
+        return None
 
-    def _extract_code(self, text):
-        if '```python' in text:
-            return text.split('```python')[1].split('```')[0]
-        return text
+# 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)
 
-class CodeExecutor:
-    def execute_safely(self, code):
-        restricted_namespace = {
-            '__builtins__': {
-                'range': range,
-                'print': print,
-                'len': len
-            }
-        }
-        try:
-            ast.parse(code)
-            exec(code, restricted_namespace)
-            return True
-        except Exception as e:
-            st.error(f"Oops! Our robot chef burned the cake: {str(e)}")
-            return False
+# 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!"*
+    """)
 
-# Streamlit UI
-st.set_page_config(page_title="CodeTales", page_icon="✨")
+# 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 = ""
 
-with st.sidebar:
-    st.header("Magic Ingredients 🧙♂️")
-    st.markdown("1. Write your story recipe ✍️\n2. We bake it into code 🧑🍳\n3. Play your game! 🎮")
+# Main content
+col1, col2 = st.columns([1, 1])
 
-st.title("CodeTales: Storytime Coding Magic ✨")
+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)
 
-story = st.text_area("Write your story recipe here (English only):", height=150,
-                    placeholder="Once upon a time, a rocket zoomed past rainbow stars...")
+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)
 
-if st.button("Bake My Story Cake! 🎂", use_container_width=True):
-    with st.spinner("Baking your code cake... 🧑🍳"):
-        converter = StoryToCodeConverter()
-        code = converter.generate_code(story)
+# 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:")
         
-        st.subheader("Your Secret Code Recipe 📜")
-        st.code(code)
+        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!")
+    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)
 
-        executor = CodeExecutor()
-        if executor.execute_safely(code):
-            st.balloons()
-            
-            with st.expander("Robot Chef's Coding Kitchen 🧑🍳"):
-                st.markdown("""
-                **How the magic works:**
-                - `move()` → Makes characters dance! 💃
-                - `color()` → Paints with rainbow colors 🌈
-                - `loop` → Repeats fun patterns 🔁
-                """)
+# Benefits section
+st.markdown("""
+## ❤ Why Everyone Will Love CodeTales
 
-            st.success("🎉 Your coding cake is ready! Keep storytelling to learn more magic!")
+| 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 |
+""")
 
-st.caption("Made with ❤️ by CodeTales Kitchen | Perfect for ages 6-12")
+# 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)