Update app.py

app.py

@@ -1,80 +1,82 @@
 import os
 import streamlit as st
 from dotenv import load_dotenv
-from novita_client import NovitaClient
+from openai import OpenAI  # Novita-compatible OpenAI interface
 import ast
 import subprocess
 import tempfile
-import time
 
-# Load API key
 load_dotenv()
 
-# --- Code Converter ---
+# --- Story-to-Code Converter ---
 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)
+        self.client = OpenAI(
+            api_key=api_key,
+            base_url="https://api.novita.ai/v3/openai"
+        )
 
-    def generate_code(self, story):
+    def generate_code(self, story: str) -> str:
         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.
+Include comments explaining loops, movement, colors—easy for kids.
 
-Story: {story}
+Story:
+\"\"\"{story}\"\"\"
 
-Only return the Python code wrapped in triple backticks like ```python ... ```
+Wrap the code in triple backticks like ```python ... ```
 """
-        # Using Novita's correct `generate` method
-        response = self.client.generate(prompt=prompt, model="gpt-4")
-        content = response.get("text", "")  # Adjust if response structure changes
-        return self._extract_code(content)
-
-    def _extract_code(self, content):
+        response = self.client.chat.completions.create(
+            model="meta-llama/Llama-4-Scout-17B-16E-Instruct",  # or another supported model
+            messages=[
+                {"role": "system", "content": "You are a friendly teacher for kids."},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=512
+        )
+        return self._extract_code(response.choices[0].message.content)
+
+    def _extract_code(self, content: str) -> str:
         if "```python" in content:
             return content.split("```python")[1].split("```")[0]
         return content
 
-
-# --- Code Executor ---
+# --- Safe Code Executor ---
 class CodeExecutor:
-    def execute_safely(self, code):
+    def execute_safely(self, code: str) -> bool:
         try:
-            ast.parse(code)  # Basic safety check
+            ast.parse(code)
             with tempfile.NamedTemporaryFile(delete=False, suffix=".py") as tmp:
                 tmp.write(code.encode("utf-8"))
-                tmp_path = tmp.name
-            subprocess.Popen(["python", tmp_path])
+                path = tmp.name
+            subprocess.Popen(["python", path])
             return True
         except Exception as e:
             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 (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!
+1. Kids write a story.  
+2. AI (Novita) turns it into Python-turtle code.  
+3. You see it! 🎮  
+4. Robot Teacher explains concepts 🎈
 """)
 
 st.title("✨ CodeTales: Storytime + Coding Magic")
 
-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...")
+story = st.text_area("Write your story here:", height=200,
+                     placeholder="A turtle walks forward, draws a rainbow circle...")
 
 if st.button("✨ Bake My Story Cake!"):
     if not story.strip():
-        st.warning("Please write a story first!")
+        st.warning("Please enter a story!")
     else:
         with st.spinner("Baking your story cake... 🍰"):
             try:
@@ -85,22 +87,19 @@ if st.button("✨ Bake My Story Cake!"):
                 st.stop()
 
             st.subheader("📜 Your Code Recipe")
-            st.code(code, language='python')
+            st.code(code, language="python")
 
             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!")
+- `turtle.forward(100)`: moves forward  
+- `turtle.color("red")`: changes color  
+- `for` loops: repeat actions  
+""")
+                st.success("🎉 Your animation came to life!")
             else:
-                st.error("Oops! Couldn't bake the animation.")
-else:
-    st.caption("Turn your words into animated code with magic 🪄")
+                st.error("Oops, something went wrong.")
 
-st.markdown("---")
-st.caption("Made with ❤️ for young coders | Powered by Novita + Python Turtle")
+st.caption("Made with ❤️ for kids | Powered by Novita AI + Python Turtle")