Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import openai
 import gradio as gr
 import json
+import plotly.graph_objects as go
 
 class ArtExplorer:
     def __init__(self):
@@ -14,6 +15,28 @@ class ArtExplorer:
             "selections": {}
         }
 
+    def create_map(self, locations):
+        """Create a Plotly map figure from location data"""
+        fig = go.Figure(go.Scattermapbox(
+            lat=[loc.get('lat') for loc in locations],
+            lon=[loc.get('lon') for loc in locations],
+            mode='markers',
+            marker=go.scattermapbox.Marker(size=10),
+            text=[loc.get('name') for loc in locations],
+            hovertemplate='<b>%{text}</b><extra></extra>'
+        ))
+
+        fig.update_layout(
+            mapbox_style="open-street-map",
+            mapbox=dict(
+                center=dict(lat=48.8566, lon=2.3522),  # Default to Paris
+                zoom=4
+            ),
+            margin=dict(r=0, t=0, l=0, b=0),
+            showlegend=False
+        )
+        return fig
+
     def get_llm_response(self, query: str, zoom_context: dict = None) -> dict:
         system_prompt = """You are an art history expert. Generate a structured JSON configuration for an interactive art exploration interface."""
         
@@ -23,6 +46,11 @@ class ArtExplorer:
         Current Selections: {json.dumps(zoom_context) if zoom_context else 'None'}
         
         Return a JSON object with configurations for temporal, geographical, and style axes.
+        Include exact coordinates for geographical locations in this format:
+        "locations": [
+            {{"name": "Paris", "lat": 48.8566, "lon": 2.3522, "description": "Center of French art"}},
+            {{"name": "Florence", "lat": 43.7696, "lon": 11.2558, "description": "Renaissance art hub"}}
+        ]
         """
 
         response = self.client.chat.completions.create(
@@ -41,7 +69,6 @@ class ArtExplorer:
         with gr.Blocks() as demo:
             gr.Markdown("# Art History Explorer")
             
-            # Query and initial search
             with gr.Row():
                 query = gr.Textbox(
                     label="Enter your art history query",
@@ -49,7 +76,6 @@ class ArtExplorer:
                 )
                 search_btn = gr.Button("Explore")
 
-            # Display areas for each axis
             with gr.Row():
                 # Temporal axis
                 with gr.Column():
@@ -64,14 +90,10 @@ class ArtExplorer:
 
                 # Geographical axis
                 with gr.Column():
-                    map_select = gr.Map(
-                        label="Geographic Location",
-                        interactive=True
-                    )
+                    map_plot = gr.Plot(label="Geographic Location")
                     geo_explanation = gr.Markdown()
                     geo_zoom = gr.Button("🔍 Zoom Geography")
 
-            # Style selection
             with gr.Row():
                 style_select = gr.Dropdown(
                     multiselect=True,
@@ -85,9 +107,12 @@ class ArtExplorer:
                 self.current_state["selections"] = {}
                 self.current_state["zoom_level"] = 0
                 
+                # Create map figure from locations
+                map_fig = self.create_map(config["geographical"]["locations"])
+                
                 return {
                     time_slider: config["temporal"]["range"],
-                    map_select: config["geographical"]["locations"],
+                    map_plot: map_fig,
                     style_select: gr.Dropdown(choices=config["style"]["options"]),
                     time_explanation: config["temporal"]["explanation"],
                     geo_explanation: config["geographical"]["explanation"],
@@ -110,8 +135,9 @@ class ArtExplorer:
                         time_explanation: config["temporal"]["explanation"]
                     })
                 elif axis_name == "geographical":
+                    map_fig = self.create_map(config["geographical"]["locations"])
                     updates.update({
-                        map_select: config["geographical"]["locations"],
+                        map_plot: map_fig,
                         geo_explanation: config["geographical"]["explanation"]
                     })
                 elif axis_name == "style":
@@ -125,7 +151,7 @@ class ArtExplorer:
             search_btn.click(
                 fn=initial_search,
                 inputs=[query],
-                outputs=[time_slider, map_select, style_select,
+                outputs=[time_slider, map_plot, style_select,
                         time_explanation, geo_explanation, style_explanation]
             )
 
@@ -137,19 +163,9 @@ class ArtExplorer:
 
             geo_zoom.click(
                 fn=lambda q, v: zoom_axis(q, "geographical", v),
-                inputs=[query, map_select],
-                outputs=[map_select, geo_explanation]
+                inputs=[query, map_plot],
+                outputs=[map_plot, geo_explanation]
             )
 
             style_zoom.click(
-                fn=lambda q, v: zoom_axis(q, "style", v),
-                inputs=[query, style_select],
-                outputs=[style_select, style_explanation]
-            )
-
-        return demo
-
-if __name__ == "__main__":
-    explorer = ArtExplorer()
-    demo = explorer.create_interface()
-    demo.launch()
+                fn=lambda q, v: zoom_axis