Update art_explorer.py

art_explorer.py

@@ -1,7 +1,7 @@
 from typing import Optional, List, Dict
 import json
 from openai import OpenAI
-from prompts import SYSTEM_PROMPT, format_exploration_prompt, DEFAULT_RESPONSE
+from .prompts import SYSTEM_PROMPT, format_exploration_prompt, DEFAULT_RESPONSE
 
 class ExplorationPathGenerator:
   def __init__(self, api_key: str):
@@ -49,7 +49,15 @@ class ExplorationPathGenerator:
           )
           
           result = response.choices[0].message.content
-          return json.loads(result)
+          try:
+              parsed_result = json.loads(result)
+              print("Generated response:", json.dumps(parsed_result, indent=2))  # Debug print
+              return parsed_result
+          except json.JSONDecodeError as e:
+              print(f"Error parsing JSON response: {e}")
+              print("Raw response:", result)  # Debug print
+              return DEFAULT_RESPONSE
+              
       except Exception as e:
-          print(f"Error generating exploration path: {e}")
+          print(f"Error in API call: {e}")  # Debug print
           return DEFAULT_RESPONSE