Updating app to show venues on a map

app.py

@@ -1,11 +1,40 @@
 #!/usr/bin/env python
 import sys
+import json
 import gradio as gr
-from crew import SurpriseTravelCrew
+import plotly.graph_objects as go
+import logging
+from crew import SurpriseTravelCrew, AddressSummaryCrew
 
+def filter_map(text_list, lat, lon):
+    fig = go.Figure(go.Scattermapbox(
+            lat=lat,
+            lon=lon,
+            mode='markers',
+            marker=go.scattermapbox.Marker(
+                size=11
+            ),
+            hovertext=text_list
+        ))
+
+    fig.update_layout(
+        mapbox_style="open-street-map",
+        hovermode='closest',
+        mapbox=dict(
+            bearing=0,
+            center=go.layout.mapbox.Center(
+                lat=lat[1],
+                lon=lon[1]
+            ),
+            pitch=0,
+            zoom=10
+        ),
+    )
+    return fig
 
 def run(origin, destination, age, trip_duration, children, budget):
     # Replace with your inputs, it will automatically interpolate any tasks and agents information
+    logger.info(f"Origin: {origin}, Destination: {destination}, Age: {age}, Duration: {trip_duration}, Children: {children}, Daily Budget: {budget}")
     inputs = {
         'origin': origin,
         'destination': destination,
@@ -15,17 +44,37 @@ def run(origin, destination, age, trip_duration, children, budget):
         'budget': budget
     }
     result = SurpriseTravelCrew().crew().kickoff(inputs=inputs)
-    return (result)
+    inputs_for_address = {
+        'text': str(result)
+    }
+
+    addresses = AddressSummaryCrew().crew().kickoff(inputs=inputs_for_address)
+    json_addresses = None
+    if addresses.json_dict:
+        json_addresses = addresses.json_dict
+    if not json_addresses:
+        try:
+            json_addresses = json.loads(addresses.raw)
+        except json.JSONDecodeError as e:
+            print ("Error loading Crew Output for addresses")
+            return (result, None)
+    fig = filter_map(json_addresses["name"], json_addresses["lat"], json_addresses["lon"])
+    return (result, fig)
 
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
 
 demo = gr.Interface(
     title="Plan your itinerary with the help of AI",
-    description="Use this app to create a detailed itinerary on how to explore a new place. Itinerary is customized to your taste",
+    description="Use this app to create a detailed itinerary on how to explore a new place. Itinerary is customized to your taste.",
     fn=run,
     inputs=["text", "text", gr.Slider(value=30, minimum=15, maximum=90, step=5), 
             gr.Slider(value=5, minimum=1, maximum=14, step=1),
             gr.Checkbox(),
             gr.Slider(value=100, minimum=20, maximum=1000, step=20)],
-    outputs=["text"],
+    outputs=[
+        gr.Textbox(label="Complete Itinerary", show_copy_button=True, autoscroll=False),
+        gr.Plot(label="Venues on a Map")
+    ]
 )
 demo.launch()

config/address_agents.yaml

@@ -0,0 +1,8 @@
+address_summarizer:
+  role: >
+    Summarize Addresses into Latitude and Longitude coordinates 
+  goal: >
+    You need to extract addresses from a provided text and convert those addresses into latitude and longitude coordinates so that they can be easily shown on a map
+  backstory: >
+    You are skilled at extracting addresses from lines of text into latitude and longitude coordinates
+

config/address_tasks.yaml

@@ -0,0 +1,6 @@
+address_compilation_task:
+  description: >
+    Extract the list of addresses from {text} in convert them to latitudes and longitudes.
+
+  expected_output: >
+    Output is expected in simple JSON format with three lists. The first list should contain the name of the venue or the name of the restaurant and should be indexed by the keyword "name". The second list should contain the latitudes and indexed with the keyword "lat". The third list should contain the longitudes and indexed with the keyword "lon".

crew.py

@@ -48,7 +48,6 @@ class SurpriseTravelCrew():
             llm=llm,
             max_iter=1,
             tools=[SerperDevTool(), ScrapeWebsiteTool()], # Example of custom tool, loaded at the beginning of file
-            verbose=True,
             allow_delegation=False,
         )
 
@@ -59,7 +58,6 @@ class SurpriseTravelCrew():
             llm=llm,
             max_iter=1,
             tools=[SerperDevTool(), ScrapeWebsiteTool()],
-            verbose=True,
             allow_delegation=False,
         )
 
@@ -69,7 +67,6 @@ class SurpriseTravelCrew():
             config=self.agents_config['itinerary_compiler'],
             llm=llm,
             max_iter=1,
-            verbose=True,
             allow_delegation=False,
         )
 
@@ -108,6 +105,40 @@ class SurpriseTravelCrew():
             agents=self.agents, # Automatically created by the @agent decorator
             tasks=self.tasks, # Automatically created by the @task decorator
             process=Process.sequential,
-            verbose=True,
             # process=Process.hierarchical, # In case you want to use that instead https://docs.crewai.com/how-to/Hierarchical/
         )
+
+
+@CrewBase
+class AddressSummaryCrew():
+    """Address Summary crew"""
+    agents_config = 'config/address_agents.yaml'
+    tasks_config = 'config/address_tasks.yaml'
+
+    @agent
+    def address_summarizer(self) -> Agent:
+        return Agent(
+            config=self.agents_config['address_summarizer'],
+            llm=llm,
+            max_iter=1,
+            allow_delegation=False,
+        )
+
+    @task
+    def address_compilation_task(self) -> Task:
+        return Task(
+            config=self.tasks_config['address_compilation_task'],
+            llm=llm,
+            max_iter=1,
+            agent=self.address_summarizer(),
+        )
+
+    @crew
+    def crew(self) -> Crew:
+        """Creates the AddressSummary crew"""
+        crew = Crew(
+            agents=self.agents, # Automatically created by the @agent decorator
+            tasks=self.tasks, # Automatically created by the @task decorator
+            process=Process.sequential,
+        )
+        return crew