Update app.py

app.py

@@ -69,7 +69,16 @@ class ArtExplorer:
               max_tokens=2048
           )
           print("Received response from LLM")
-          result = json.loads(response.choices[0].message.content)
+          
+          # Clean and parse the response
+          content = response.choices[0].message.content.strip()
+          try:
+              result = json.loads(content)
+          except json.JSONDecodeError as e:
+              print(f"JSON parsing error: {str(e)}")
+              print(f"Raw content: {content}")
+              return self.get_default_response()
+              
           return result
       except Exception as e:
           print(f"Error in LLM response: {str(e)}")
@@ -109,17 +118,16 @@ class ArtExplorer:
 
           with gr.Row():
               style_select = gr.Dropdown(
+                  choices=["Classical", "Modern"],  # Set initial choices
                   multiselect=True,
-                  label="Artistic Styles"
+                  label="Artistic Styles",
+                  allow_custom_value=True  # Allow new values
               )
               style_explanation = gr.Markdown()
               style_zoom = gr.Button("🔍 Zoom Styles")
 
           def initial_search(query):
-              """
-              Handle the initial search query and return properly formatted outputs for Gradio
-              Returns individual values instead of a dictionary for Gradio components
-              """
+              """Handle the initial search query"""
               config = self.get_llm_response(query)
               temporal_config = config["axis_configurations"]["temporal"]["current_zoom"]
               geographical_config = config["axis_configurations"]["geographical"]["current_zoom"]
@@ -127,7 +135,7 @@ class ArtExplorer:
               
               map_fig = self.create_map(geographical_config["locations"])
               
-              # Return individual values in the order expected by Gradio
+              # Return values directly in the order expected by Gradio
               return (
                   temporal_config["range"],  # time_slider
                   map_fig,                   # map_plot
@@ -138,9 +146,7 @@ class ArtExplorer:
               )
 
           def zoom_axis(query, axis_name, current_value):
-              """
-              Handle zoom events for any axis with proper return formatting
-              """
+              """Handle zoom events for any axis"""
               self.current_state["zoom_level"] += 1
               config = self.get_llm_response(
                   query,
@@ -208,7 +214,7 @@ def main():
   demo.launch(
       server_name="0.0.0.0",
       server_port=7860,
-      share=True  # Set to True to create a public link
+      share=True
   )
 
 if __name__ == "__main__":