Update app.py

app.py

@@ -100,6 +100,63 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
+# 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
+
+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
+        )
+        
+        if response.status_code == 200:
+            return response.json()
+        else:
+            return None
+    except:
+        return None
+
 # 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)
@@ -121,6 +178,16 @@ with st.expander("✨ How It Works (Like Baking a Cake) 🎂"):
     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 = ""
+
 # Main content
 col1, col2 = st.columns([1, 1])
 
@@ -131,73 +198,110 @@ with col1:
         "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"
+        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"):
-        st.session_state.generate_clicked = True
+        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)
 
 with col2:
     st.markdown('<div class="story-box">', unsafe_allow_html=True)
     st.markdown("### 🎮 Your Game Animation")
     
-    if 'generate_clicked' in st.session_state and story_text:
+    if st.session_state.generating and st.session_state.task_id:
         with st.spinner("🧙‍♂️ Cooking your story in the magic oven..."):
-            # Simulate processing time
-            time.sleep(3)
-            
-            # Generate animation (would use actual API call in production)
-            st.session_state.animation_generated = True
-            st.session_state.story_text = story_text
+            status = get_animation_status(st.session_state.task_id)
             
-            # Display animation placeholder
-            st.video("https://cdn.pixabay.com/video/2023/01/20/155564-810349295_large.mp4")
-            
-            st.success("🎉 Your animation is ready! Tavus will explain the magic now!")
-    elif 'animation_generated' in st.session_state:
-        st.video("https://cdn.pixabay.com/video/2023/01/20/155564-810349295_large.mp4")
+            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.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)
 
 # Tavus explanation section
-if 'animation_generated' in st.session_state and 'story_text' in st.session_state:
+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"]
+    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:
-        action = found_actions[0]
-        explanation = {
-            "zoom": "makes your spaceship move super fast!",
-            "shoot": "creates laser beams to defeat enemies!",
-            "fly": "keeps your character moving through the air!",
-            "move": "changes position on the screen!",
-            "jump": "makes your character leap upwards!",
-            "run": "makes your character move faster!",
-            "attack": "lets your hero fight the bad guys!"
-        }
+        # 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:")
         
-        code_snippet = {
-            "zoom": "spaceship.accelerate(speed=10)",
-            "shoot": "laser = spaceship.fire_weapon()",
-            "fly": "character.apply_gravity(False)",
-            "move": "player.move(direction='right')",
-            "jump": "hero.jump(height=100)",
-            "run": "player.speed = player.speed * 2",
-            "attack": "sword.swing(damage=15)"
-        }
+        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(f"See how your story became real code? When you said **'{action}'**, we used:")
-        st.markdown(f'<div class="code-block">{code_snippet[action]}</div>', unsafe_allow_html=True)
-        st.markdown(f"This code {explanation[action]} That's the magic of coding - turning words into actions!")
+        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!")