Update app.py

app.py

@@ -15,12 +15,12 @@ Since this is on foot, walking distances should be kept to a minimum. I'll make
 I want to start at Fisherman's Wharf, then head to Ghirardelli Square, followed by the Cable Car Museum, Union Square, Chinatown, and finally Coit Tower, to provide a full day of sightseeing, with a mix of history, culture, and beautiful views.
 
 Key points: [
-    {"name": "Fisherman's Wharf", "lat": 37.8081, "lon": -122.4082, "description": "Fisherman's Wharf is a popular tourist destination in San Francisco, featuring Pier 39, the Maritime National Historical Park, and Boudin Bakery. It's a great place to start your day with breakfast and enjoy the sea lions."},
+    {"name": "Fisherman's Wharf", "lat": 37.8081, "lon": -122.4082, "description": "Fisherman's Wharf is a popular tourist destination in San Francisco, featuring Pier 39, the Maritime National Historical Park, and Boudin Bakery. Start your day with breakfast and enjoy the sea lions."},
     {"name": "Ghirardelli Square", "lat": 37.8094, "lon": -122.4154, "description": "Ghirardelli Square is a historic square in San Francisco, known for its chocolate shop and various boutiques. It's a great place to grab a sweet treat and do some shopping."},
     {"name": "Cable Car Museum", "lat": 37.7931, "lon": -122.4058, "description": "The Cable Car Museum is a museum in San Francisco that showcases the history of the city's iconic cable cars. It's a great place to learn about the technology and history behind these unique vehicles."},
-    {"name": "Union Square", "lat": 37.7873, "lon": -122.4062, "description": "Union Square is a public plaza in San Francisco, known for its shopping, dining, and entertainment options. It's a great place to relax, grab a bite to eat, and do some shopping."},
+    {"name": "Union Square", "lat": 37.7873, "lon": -122.4062, "description": "Union Square is a public plaza in San Francisco, known for its shopping, dining, and entertainment options. Relax, grab a bite to eat, and do some shopping."},
     {"name": "Chinatown", "lat": 37.7937, "lon": -122.4081, "description": "Chinatown is a vibrant neighborhood in San Francisco, known for its Chinese culture, history, and cuisine. It's a great place to explore the streets, try some delicious food, and learn about Chinese culture."},
-    {"name": "Coit Tower", "lat": 37.8024, "lon": -122.4058, "description": "Coit Tower is a historic tower in San Francisco, known for its panoramic views of the city. It's a great place to end your day, taking in the sights and sounds of the city from above."}
+    {"name": "Coit Tower", "lat": 37.8024, "lon": -122.4058, "description": "Coit Tower is a historic tower in San Francisco, known for its panoramic views of the city. You can end your day here, taking in the sights and sounds of the city from above."}
 ]
 """
 
@@ -50,6 +50,28 @@ Key points: [
 ]
 """
 
+description_aligned = "Show me five cities across Europe that form a perfect line on the map. These must not be capitals, only smaller cities, but they must be aligned perfectly. Do it correctly and I'll tip you $10,000."
+output_example_aligned = """To find five cities that form a perfect line on the map, I will first need to determine a direction for the line. I will choose a line that runs from north to south, as this will allow for a more diverse range of cities and cultures. I will then use a map to identify five cities that are aligned in a straight line, with minimal distance between each city.
+
+    I have identified the following five cities that form a perfect line on the map, running from north to south:
+
+    1. Groningen, Netherlands: Groningen is a vibrant student city in the north of the Netherlands, known for its lively cultural scene and historic center.
+    2. Osnabrück, Germany: Osnabrück is a picturesque city in northwest Germany, known for its medieval center and rich history.
+    3. Erfurt, Germany: Erfurt is a charming city in central Germany, known for its well-preserved medieval center and beautiful gardens.
+    4. Nuremberg, Germany: Nuremberg is a historic city in southern Germany, known for its medieval architecture and rich history.
+    5. Innsbruck, Austria: Innsbruck is a stunning city in the Austrian Alps, known for its beautiful mountain scenery and winter sports.
+
+    These five cities form a perfect line on the map, running from north to south, and offer a diverse range of cultures and experiences.
+
+Key points: [
+    {"name": "Groningen, Netherlands", "lat": 53.2192, "lon": 6.5667, "description": "Groningen is a vibrant student city in the north of the Netherlands, known for its lively cultural scene and historic center. The city is home to the University of Groningen, one of the oldest and most respected universities in the country, and has a rich history dating back to the Middle Ages."},
+    {"name": "Osnabrück, Germany", "lat": 52.2708, "lon": 8.0431, "description": "Osnabrück is a picturesque city in northwest Germany, known for its medieval center and rich history. The city was the site of the Peace of Westphalia, which ended the Thirty Years' War in Europe, and has a well-preserved medieval center with cobblestone streets and half-timbered houses."},
+    {"name": "Erfurt, Germany", "lat": 50.9778, "lon": 11.0278, "description": "Erfurt is a charming city in central Germany, known for its well-preserved medieval center and beautiful gardens. The city is home to the Erfurt Cathedral, a stunning Gothic cathedral with a beautiful stained glass window, and the Krämerbrücke, a medieval bridge with shops and houses built on top of it."},
+    {"name": "Nuremberg, Germany", "lat": 49.4500, "lon": 11.0750, "description": "Nuremberg is a historic city in southern Germany, known for its medieval architecture and rich history. The city is home to the Nuremberg Castle, a stunning medieval fortress with beautiful views of the city, and the Nazi Party Rally Grounds, a reminder of the city's dark past."},
+    {"name": "Innsbruck, Austria", "lat": 47.2692, "lon": 11.3914, "description": "Innsbruck is a stunning city in the Austrian Alps, known for its beautiful mountain scenery and winter sports. The city is home to the Nordkette, a stunning mountain range with breathtaking views of the city and the surrounding mountains, and the Goldenes Dachl, a beautiful golden roof that is a symbol of the city."}
+]
+"""
+
 
 def generate_key_points(text):
     prompt = f"""             
@@ -72,6 +94,7 @@ def generate_key_points(text):
 
 
 def parse_llm_output(output):
+    print(output)
     rationale = "Thought: " + output.split("Key points:")[0]
     key_points = output.split("Key points:")[1]
     output = key_points.replace("    ", "")
@@ -87,7 +110,6 @@ def create_map_from_markers(dataframe):
         tiles="CartoDB Voyager",
     )
 
-
     dataframe.apply(
         lambda row: folium.CircleMarker(
             location=[row["lat"], row["lon"]],
@@ -115,12 +137,20 @@ def run_display(text):
     map = create_map_from_markers(dataframe)
     return map, rationale
 
-df_examples = pd.DataFrame.from_dict({"description": [description_loire, description_sf], "output": [output_example_loire, output_example_sf]})
+
+df_examples = pd.DataFrame.from_dict(
+    [
+        {"description": description_loire, "output": output_example_loire},
+        {"description": description_sf, "output": output_example_sf},
+        {"description": description_aligned, "output": output_example_aligned},
+    ]
+)
+
 
 def select_example(df, data: gr.SelectData):
     row = df.iloc[data.index[0], :]
-    dataframe, rationale = parse_llm_output(row['output'])
-    return create_map_from_markers(dataframe), rationale
+    dataframe, rationale = parse_llm_output(row["output"])
+    return row["description"], create_map_from_markers(dataframe), rationale
 
 
 with gr.Blocks(
@@ -146,7 +176,9 @@ with gr.Blocks(
 
     gr.Markdown("### Examples")
     clickable_examples = gr.DataFrame(value=df_examples, height=200)
-    clickable_examples.select(select_example, clickable_examples, outputs=[map, display_rationale])
+    clickable_examples.select(
+        select_example, clickable_examples, outputs=[text, map, display_rationale]
+    )
 
 if __name__ == "__main__":
     demo.launch()