Update app.py

app.py

@@ -7,6 +7,16 @@ import pandas as pd
 from gradio_folium import Folium
 import folium
 from huggingface_hub import InferenceClient
+from geopy.geocoders import Nominatim
+
+geolocator = Nominatim(user_agent="HF-trip-planner")
+
+def get_coordinates(address):
+    location = geolocator.geocode(address)
+    if location:
+        return (location.latitude, location.longitude)
+    else:
+        return None
 
 repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 llm_client = InferenceClient(model=repo_id, timeout=180)
@@ -22,7 +32,7 @@ Key points: [
     {"name": "Cable Car Museum", "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, San Francisco", "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, San Francisco", "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, San Francisco", "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."}
+    {"name": "Coit Tower, San Francisco", "description": "Coit Tower is a tower in San Francisco with panoramic views of the city. You can end your day here, taking in the sights and sounds of the city from above."}
 ]
 """
 
@@ -44,11 +54,11 @@ Key points: [
     {"name": "Château de Chambord", "description": "Château de Chambord is the largest and most recognizable château in the Loire Valley, known for its French Renaissance architecture and beautiful gardens."},
     {"name": "Château de Cheverny", "description": "Château de Cheverny is a small but charming château with beautiful gardens, known for its French classical architecture and rich history."},
     {"name": "Château de Chaumont-sur-Loire", "description": "Château de Chaumont-sur-Loire is a château with a stunning view of the Loire River, known for its beautiful gardens and rich history."},
-    {"name": "Château de Chenonceau", "description": "Château de Chenonceau is a stunning château built over the River Cher, known for its beautiful gardens and rich history."},
-    {"name": "Château d'Amboise", "description": "Château d'Amboise is a historic château with ties to the French Renaissance, known for its beautiful gardens and stunning views of the Loire Valley."},
+    {"name": "Château de Chenonceau", "description": "Château de Chenonceau, known for its beautiful gardens and rich history, is a stunning château built over the River Cher."},
+    {"name": "Château d'Amboise", "description": "Château d'Amboise has ties to the French Renaissance: this château is known for its beautiful gardens and stunning views of the Loire Valley."},
     {"name": "Château de Villandry", "description": "Château de Villandry is known for its beautiful gardens, featuring a water garden, ornamental flower gardens, and vegetable gardens."},
     {"name": "Château d'Azay-le-Rideau", "description": "Château d'Azay-le-Rideau is a charming château with a moat, known for its French Renaissance architecture and beautiful gardens."},
-    {"name": "Château d'Ussé", "description": "Château d'Ussé is a fairy-tale like château said to have inspired Sleeping Beauty, known for its beautiful gardens and stunning views of the Indre Valley."},
+    {"name": "Château d'Ussé", "description": "Château d'Ussé is a fairy-tale like château that displays beautiful gardens and stunning views of the Indre Valley. It is said to have inspired Sleeping Beauty,"},
 ]
 """
 
@@ -100,10 +110,40 @@ Key points: [
     {"name": "Las Vegas Chinatown, Las Vegas, NV", "description": "Las Vegas Chinatown is a vibrant neighborhood located just west of the Las Vegas Strip, known for its beautiful architecture, traditional Chinese markets, and delicious food. It is a popular destination for tourists and locals alike, offering a unique blend of Chinese culture and American history."}
 ]
 """
+
+description_taiwan= "A two-week trip on train around Taiwan."
+output_example_taiwan = """Since this is a train trip, I will make sure to plan the trip around train stations and routes. I will also try to minimize the distance between locations and plan the trip so that it can be done in two weeks. I will also try to include a variety of activities and sights to see, such as cultural experiences, natural wonders, and historical sites.
+
+    Here is a suggested itinerary for a two-week train trip around Taiwan:
+
+    1. Start in Taipei, the capital city of Taiwan. Visit the National Palace Museum, Taipei 101, and the Chiang Kai-shek Memorial Hall.
+    2. Take the train to Hualien, a coastal city on the east coast of Taiwan. Visit the Taroko National Park, known for its stunning marble cliffs and hiking trails.
+    3. Take the train to Taitung, a city on the southeast coast of Taiwan. Visit the Taitung Forest Park and the Green Island, a volcanic island known for its clear waters and coral reefs.
+    4. Take the train to Kaohsiung, a city on the southwest coast of Taiwan. Visit the Fo Guang Shan Buddha Museum, the Love River, and the Pier-2 Art Center.
+    5. Take the train to Tainan, a city on the southwest coast of Taiwan. Visit the Anping Old Fort, the Chimei Museum, and the National Museum of Taiwanese Literature.
+    6. Take the train to Chiayi, a city on the west coast of Taiwan. Visit the Alishan National Scenic Area, known for its mountain scenery and sunrise views.
+    7. Take the train to Taichung, a city in the west-central part of Taiwan. Visit the National Taiwan Museum of Fine Arts, the Taichung Park, and the Miyahara Ice Cream Shop.
+    8. Take the train to Hsinchu, a city in the northwest part of Taiwan. Visit the Hsinchu Science Park and the Hsinchu Zoo.
+    9. Take the train back to Taipei. Visit the Taipei Zoo, the Maokong Gondola, and the Yangmingshan National Park.
+
+Key points: [
+    {"name": "Taipei, Taiwan", "description": "Taipei is the capital city of Taiwan, known for its vibrant culture, rich history, and modern skyline. Visit the National Palace Museum, Taipei 101, and the Chiang Kai-shek Memorial Hall."},
+    {"name": "Hualien, Taiwan", "description": "Hualien is a coastal city on the east coast of Taiwan, known for its stunning natural beauty and outdoor activities. Visit the Taroko National Park, known for its marble cliffs and hiking trails."},
+    {"name": "Taitung, Taiwan", "description": "Taitung is a city on the southeast coast of Taiwan, known for its beautiful beaches and outdoor activities. Visit the Taitung Forest Park and the Green Island, a volcanic island known for its clear waters and coral reefs."},
+    {"name": "Kaohsiung, Taiwan", "description": "Kaohsiung is a city on the southwest coast of Taiwan, known for its vibrant culture, rich history, and modern skyline. Visit the Fo Guang Shan Buddha Museum, the Love River, and the Pier-2 Art Center."},
+    {"name": "Tainan, Taiwan", "description": "Tainan is a city on the southwest coast of Taiwan, known for its rich history, cultural heritage, and delicious food. Visit the Anping Old Fort, the Chimei Museum, and the National Museum of Taiwanese Literature."},
+    {"name": "Chiayi, Taiwan", "description": "Chiayi is a city on the west coast of Taiwan, known for its stunning mountain scenery and outdoor activities. Visit the Alishan National Scenic Area, known for its mountain scenery and sunrise views."},
+    {"name": "Taichung, Taiwan", "description": "Taichung is a city in the west-central part of Taiwan, known for its vibrant culture, rich history, and modern skyline. Visit the National Taiwan Museum of Fine Arts, the Taichung Park, and the Miyahara Ice Cream Shop."},
+    {"name": "Hsinchu, Taiwan", "description": "Hsinchu is a city in the northwest part of Taiwan, known for its high-tech industry and beautiful natural scenery. Visit the Hsinchu Science Park and the Hsinchu Zoo."},
+]
+"""
+
+
 def generate_key_points(text):
     prompt = f"""             
     Please generate a set of key geographical points for the following description: {text}, as a json list of less than 10 dictionnaries with the following keys: 'name', 'description'. Precise the full location in the 'name' if there is a possible ambiguity.
     Generally try to minimze the distance between locations. Always think of the transportation means that you want to use, and the timing: morning, afternoon, where to sleep.
+    Only generate a 'Thought:' and a 'Key points:' sections, nothing else.
 
     For instance:
     Description: {description_sf}
@@ -175,9 +215,9 @@ def run_display(text):
 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},
         {"description": description_chinatown, "output": output_example_chinatown},
+        {"description": description_taiwan, "output": output_example_taiwan},
     ]
 )
 
@@ -209,7 +249,7 @@ with gr.Blocks(
     map = Folium(value=starting_map, height=600, label="Chosen locations")
     button.click(run_display, inputs=[text], outputs=[map, display_rationale])
 
-    gr.Markdown("### Examples")
+    gr.Markdown("### Other examples")
     clickable_examples = gr.DataFrame(value=df_examples, height=200)
     clickable_examples.select(
         select_example, clickable_examples, outputs=[text, map, display_rationale]