Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 import json
 import plotly.graph_objects as go
 
-
 CONTEXTUAL_ZOOM_PROMPT = """
 You are an expert art historian specializing in interactive exploration. Analyze the query and generate contextually aware zoom configurations with explanations.
 
@@ -98,12 +97,12 @@ Current Zoom States: {
         "current_zoom": {
             "level": "period",
             "range": [1799, 1815],
-            "explanation": "Napoleon's reign as First Consul and Emperor"[1]
+            "explanation": "Napoleon's reign as First Consul and Emperor"
         },
         "available_zooms": {
             "in": {
                 "range": [1812, 1815],
-                "explanation": "Focus on final campaigns and artistic responses"[1]
+                "explanation": "Focus on final campaigns and artistic responses"
             }
         }
     },
@@ -114,11 +113,11 @@ Current Zoom States: {
             "locations": [
                 {
                     "name": "France",
-                    "relevance": "Center of Napoleonic art production"[2]
+                    "relevance": "Center of Napoleonic art production"
                 },
                 {
                     "name": "Spain",
-                    "relevance": "Goya's war paintings perspective"[1]
+                    "relevance": "Goya's war paintings perspective"
                 }
             ]
         }
@@ -135,8 +134,6 @@ Current Zoom States: {
 
 Generate only the JSON response, maintaining strict JSON format."""
 
-
-
 class ArtExplorer:
     def __init__(self):
         self.client = openai.OpenAI(
@@ -152,7 +149,7 @@ class ArtExplorer:
         """Create a Plotly map figure from location data"""
         if not locations:
             locations = [{"name": "Paris", "lat": 48.8566, "lon": 2.3522}]
-            
+
         fig = go.Figure(go.Scattermapbox(
             lat=[loc.get('lat') for loc in locations],
             lon=[loc.get('lon') for loc in locations],
@@ -179,18 +176,20 @@ class ArtExplorer:
                 "style": {"level": self.current_state["zoom_level"], "selection": ""},
                 "subject": {"level": self.current_state["zoom_level"], "selection": ""}
             }
-            
+
             if zoom_context:
                 for key, value in zoom_context.items():
                     if key in current_zoom_states:
                         current_zoom_states[key]["selection"] = value
+
             messages=[
                 {"role": "system", "content": "You are an expert art historian specializing in interactive exploration."},
                 {"role": "user", "content": CONTEXTUAL_ZOOM_PROMPT.format(
                     user_query=query,
                     current_zoom_states=json.dumps(current_zoom_states, indent=2)
                 )}
-            ],
+            ]
+
             print(messages)
             response = self.client.chat.completions.create(
                 model="mixtral-8x7b-32768",
@@ -201,7 +200,6 @@ class ArtExplorer:
             print(response)
             result = json.loads(response.choices[0].message.content)
             return result
-            
         except Exception as e:
             print(f"Error in LLM response: {str(e)}")
             return self.get_default_response()
@@ -293,13 +291,10 @@ class ArtExplorer:
 
             def initial_search(query):
                 config = self.get_llm_response(query)
-                
-                # Access the correct nested structure
                 temporal_config = config["axis_configurations"]["temporal"]["current_zoom"]
                 geographical_config = config["axis_configurations"]["geographical"]["current_zoom"]
                 style_config = config["axis_configurations"]["style"]["current_zoom"]
                 
-                # Create map figure
                 map_fig = self.create_map(geographical_config["locations"])
                 
                 return {
@@ -317,7 +312,6 @@ class ArtExplorer:
                     query,
                     zoom_context={axis_name: current_value}
                 )
-                
                 axis_config = config["axis_configurations"][axis_name]["current_zoom"]
                 
                 if axis_name == "temporal":
@@ -342,11 +336,11 @@ class ArtExplorer:
                 fn=initial_search,
                 inputs=[query],
                 outputs=[
-                    time_slider, 
-                    map_plot, 
+                    time_slider,
+                    map_plot,
                     style_select,
-                    time_explanation, 
-                    geo_explanation, 
+                    time_explanation,
+                    geo_explanation,
                     style_explanation
                 ]
             )