Update app.py

app.py

@@ -7,9 +7,10 @@ import subprocess
 import tempfile
 import time
 
+# Load API key
 load_dotenv()
 
-# Initialize Novita API client
+# --- Code Converter ---
 class StoryToCodeConverter:
     def __init__(self):
         api_key = os.getenv("NOVITA_API_KEY")
@@ -26,11 +27,9 @@ Story: {story}
 
 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}]
-        )
-        content = response.choices[0].message.content
+        # 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):
@@ -39,11 +38,11 @@ Only return the Python code wrapped in triple backticks like ```python ... ```
         return content
 
 
-# Safe execution
+# --- Code Executor ---
 class CodeExecutor:
     def execute_safely(self, code):
         try:
-            ast.parse(code)  # Safety check
+            ast.parse(code)  # Basic safety check
             with tempfile.NamedTemporaryFile(delete=False, suffix=".py") as tmp:
                 tmp.write(code.encode("utf-8"))
                 tmp_path = tmp.name
@@ -54,7 +53,7 @@ class CodeExecutor:
             return False
 
 
-# ---------- Streamlit UI ----------
+# --- Streamlit Interface ---
 st.set_page_config(page_title="CodeTales", page_icon="✨")
 
 with st.sidebar:
@@ -78,8 +77,8 @@ if st.button("✨ Bake My Story Cake!"):
         st.warning("Please write a story first!")
     else:
         with st.spinner("Baking your story cake... 🍰"):
-            converter = StoryToCodeConverter()
             try:
+                converter = StoryToCodeConverter()
                 code = converter.generate_code(story)
             except Exception as e:
                 st.error(f"AI oven broke down! {e}")