Update utils/planner.py

utils/planner.py

@@ -1,26 +1,26 @@
 # utils/planner.py
 
-import openai
 import os
+import json
+import logging
+from openai import OpenAI
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 SYSTEM_INSTRUCTIONS = """
 You are a scene planning assistant for an AI image generation system.
-
 Your job is to take the user's prompt and return a structured JSON with:
 - scene (environment, setting)
 - subject (main actor)
 - objects (main product or items)
 - layout (foreground/background elements and their placement)
 - rules (validation rules to ensure visual correctness)
-
 Respond ONLY in raw JSON format. Do NOT include explanations.
 """
 
 def extract_scene_plan(prompt: str) -> dict:
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-4-0125-preview",
             messages=[
                 {"role": "system", "content": SYSTEM_INSTRUCTIONS},
@@ -29,11 +29,16 @@ def extract_scene_plan(prompt: str) -> dict:
             temperature=0.3,
             max_tokens=500
         )
-        json_output = response["choices"][0]["message"]["content"]
-        return eval(json_output)  # Converts JSON string to Python dict
+
+        content = response.choices[0].message.content.strip()
+
+        # Remove common formatting issues
+        content = content.strip("`").replace("json", "")
+
+        return json.loads(content)
 
     except Exception as e:
-        print("LLM Error:", e)
+        logging.exception("LLM Error:")
         return {
             "scene": None,
             "subject": None,