Update app.py

app.py

@@ -1,14 +1,13 @@
 import os
 import streamlit as st
 from dotenv import load_dotenv
-from openai import OpenAI  # Novita-compatible OpenAI interface
+from openai import OpenAI
 import ast
 import subprocess
 import tempfile
 
 load_dotenv()
 
-# --- Story-to-Code Converter ---
 class StoryToCodeConverter:
     def __init__(self):
         api_key = os.getenv("NOVITA_API_KEY")
@@ -22,7 +21,7 @@ class StoryToCodeConverter:
     def generate_code(self, story: str) -> str:
         prompt = f"""
 Convert this children's story into Python animation code using turtle graphics.
-Include comments explaining loops, movement, colors—easy for kids.
+Include comments that explain loops, movement, colors—easy for kids.
 
 Story:
 \"\"\"{story}\"\"\"
@@ -30,12 +29,12 @@ Story:
 Wrap the code in triple backticks like ```python ... ```
 """
         response = self.client.chat.completions.create(
-            model="meta-llama/Llama-4-Scout-17B-16E-Instruct",  # or another supported model
+            model="meta-llama/llama-3.1-8b-instruct",
             messages=[
-                {"role": "system", "content": "You are a friendly teacher for kids."},
+                {"role": "system", "content": "You are a friendly teacher and coder for kids."},
                 {"role": "user", "content": prompt}
             ],
-            max_tokens=512
+            max_tokens=512,
         )
         return self._extract_code(response.choices[0].message.content)
 
@@ -44,7 +43,7 @@ Wrap the code in triple backticks like ```python ... ```
             return content.split("```python")[1].split("```")[0]
         return content
 
-# --- Safe Code Executor ---
+
 class CodeExecutor:
     def execute_safely(self, code: str) -> bool:
         try:
@@ -58,27 +57,26 @@ class CodeExecutor:
             st.error(f"Robot Chef burned the cake: {e}")
             return False
 
-# --- Streamlit Interface ---
+
 st.set_page_config(page_title="CodeTales", page_icon="✨")
 with st.sidebar:
     st.header("🍰 How It Works")
     st.markdown("""
-1. Kids write a story.  
-2. AI (Novita) turns it into Python-turtle code.  
-3. You see it! 🎮  
-4. Robot Teacher explains concepts 🎈
+1. 📝 Kids write a story  
+2. 🔮 Novita AI turns it into Python‑turtle code  
+3. 🎮 Animation plays  
+4. 🤖 Robot Teacher explains concepts  
 """)
 
 st.title("✨ CodeTales: Storytime + Coding Magic")
-
 story = st.text_area("Write your story here:", height=200,
-                     placeholder="A turtle walks forward, draws a rainbow circle...")
+                     placeholder="A turtle draws a rainbow circle...")
 
 if st.button("✨ Bake My Story Cake!"):
     if not story.strip():
         st.warning("Please enter a story!")
     else:
-        with st.spinner("Baking your story cake... 🍰"):
+        with st.spinner("Baking your story code... 🍰"):
             try:
                 converter = StoryToCodeConverter()
                 code = converter.generate_code(story)
@@ -94,12 +92,12 @@ if st.button("✨ Bake My Story Cake!"):
                 st.balloons()
                 with st.expander("🤖 Robot Teacher Says:"):
                     st.markdown("""
-- `turtle.forward(100)`: moves forward  
-- `turtle.color("red")`: changes color  
-- `for` loops: repeat actions  
+- `turtle.forward(100)` – Moves forward  
+- `turtle.color("red")` – Changes color  
+- `for` loops – Repeat patterns  
 """)
-                st.success("🎉 Your animation came to life!")
+                st.success("🎉 Animation ready!")
             else:
                 st.error("Oops, something went wrong.")
 
-st.caption("Made with ❤️ for kids | Powered by Novita AI + Python Turtle")
+st.caption("Made with ❤️ for young coders | Powered by Novita AI + Python Turtle")