Update app.py

app.py

@@ -39,6 +39,28 @@ class ArtExplorer:
       )
       return fig
 
+  def clean_json_string(self, json_str: str) -> str:
+      """Clean and prepare JSON string for parsing"""
+      # Remove any leading/trailing whitespace and newlines
+      json_str = json_str.strip()
+      
+      # Remove any BOM or special characters at the start
+      if json_str.startswith('\ufeff'):
+          json_str = json_str[1:]
+          
+      # Ensure it starts with {
+      if not json_str.startswith('{'):
+          start_idx = json_str.find('{')
+          if start_idx != -1:
+              json_str = json_str[start_idx:]
+              
+      # Ensure it ends with }
+      end_idx = json_str.rfind('}')
+      if end_idx != -1:
+          json_str = json_str[:end_idx+1]
+          
+      return json_str
+
   def get_llm_response(self, query: str, zoom_context: dict = None) -> dict:
       try:
           current_zoom_states = {
@@ -70,16 +92,23 @@ class ArtExplorer:
           )
           print("Received response from LLM")
           
-          # Clean and parse the response
-          content = response.choices[0].message.content.strip()
+          # Get the response content and clean it
+          content = response.choices[0].message.content
+          print(f"Raw LLM response: {content}")
+          
+          # Clean the JSON string
+          cleaned_content = self.clean_json_string(content)
+          print(f"Cleaned JSON string: {cleaned_content}")
+          
           try:
-              result = json.loads(content)
+              result = json.loads(cleaned_content)
+              print("Successfully parsed JSON")
+              return result
           except json.JSONDecodeError as e:
               print(f"JSON parsing error: {str(e)}")
-              print(f"Raw content: {content}")
+              print(f"Failed to parse JSON: {cleaned_content}")
               return self.get_default_response()
               
-          return result
       except Exception as e:
           print(f"Error in LLM response: {str(e)}")
           return self.get_default_response()
@@ -118,7 +147,7 @@ class ArtExplorer:
 
           with gr.Row():
               style_select = gr.Dropdown(
-                  choices=["Classical", "Modern"],  # Set initial choices
+                  choices=["Classical", "Modern"],  # Initial choices
                   multiselect=True,
                   label="Artistic Styles",
                   allow_custom_value=True  # Allow new values