Update google_places.py

google_places.py

@@ -1,154 +1,154 @@
-from dotenv import load_dotenv
-import os
-import googlemaps
-import re
-import gradio as gr
-from datetime import datetime
-
-# Load environment variables and initialize Google Maps client
-dotenv_path = os.getenv('DOTENV_PATH', None)
-if dotenv_path:
-    load_dotenv(dotenv_path)
-else:
-    load_dotenv()
-
-api_key = os.getenv("GOOGLE_API_KEY")
-if not api_key:
-    raise EnvironmentError("GOOGLE_API_KEY not found in environment variables.")
-
-gmaps = googlemaps.Client(key=api_key)
-
-
-def get_places(trip_type: str, location: str) -> list[str]:
-    """
-    Fetches up to 5 popular tourist attractions matching the trip_type in the given location.
-    """
-    query = f"{trip_type} places in {location}"
-    results = gmaps.places(query=query, type="tourist_attraction")
-    places = results.get("results", [])[:5]
-    return [place.get("name") for place in places]
-
-
-def get_restaurants(location: str, prefs: str) -> list[str]:
-    """
-    Fetches up to 3 restaurants near the given location matching the user preferences.
-    """
-    query = f"{prefs} restaurants near {location}"
-    results = gmaps.places(query=query, type="restaurant")
-    restaurants = results.get("results", [])[:3]
-    return [r.get("name") for r in restaurants]
-
-
-def get_directions(
-    source: str,
-    destination: str,
-    mode: str,
-    departure_time: str,
-    waypoints: str = ""
-) -> str:
-    """
-    Returns a Markdown-formatted string with step-by-step directions and a static map URL.
-
-    Args:
-        source: Starting address or place name.
-        destination: Ending address or place name.
-        mode: Transport mode: driving, walking, bicycling, or transit.
-        departure_time: Trip start time in "YYYY-MM-DD HH:MM" format.
-        waypoints: Optional comma-separated list of waypoints.
-
-    Returns:
-        A single Markdown string summarizing the route or an error message.
-    """
-    # Parse departure time
-    try:
-        dt = datetime.strptime(departure_time.strip(), "%Y-%m-%d %H:%M")
-    except ValueError:
-        return (
-            "**⚠️ Invalid time format.** Please enter departure time as YYYY-MM-DD HH:MM."
-        )
-
-    # Build waypoints list
-    wp_list = [w.strip() for w in waypoints.split(",")] if waypoints else None
-
-    # Call Directions API
-    directions = gmaps.directions(
-        origin=source,
-        destination=destination,
-        mode=mode.lower(),
-        departure_time=dt,
-        waypoints=wp_list,
-        optimize_waypoints=False,
-    )
-
-    if not directions:
-        return "**No route found.**"
-
-    # Use first route & leg
-    route = directions[0]
-    leg = route["legs"][0]
-
-    # Build summary
-    summary = f"**Trip from '{leg['start_address']}' to '{leg['end_address']}'**\n"
-    summary += f"- Total distance: {leg['distance']['text']}\n"
-    summary += f"- Estimated duration: {leg['duration']['text']}\n\n"
-    summary += "### Step-by-Step Directions:\n"
-
-    for i, step in enumerate(leg["steps"], start=1):
-        travel_mode = step.get("travel_mode", "").upper()
-        if travel_mode == "TRANSIT":
-            details = step.get("transit_details", {})
-            line = details.get("line", {})
-            vehicle = line.get("vehicle", {}).get("type", "Transit")
-            line_name = line.get("short_name") or line.get("name", "")
-            dep = details.get("departure_stop", {}).get("name", "")
-            arr = details.get("arrival_stop", {}).get("name", "")
-            stops = details.get("num_stops", "")
-            instr = f"Take {vehicle} {line_name} from {dep} to {arr} ({stops} stops)"
-            dist = step.get("distance", {}).get("text", "")
-        else:
-            raw = step.get("html_instructions", "")
-            instr = re.sub(r"<[^>]+>", "", raw)
-            dist = step.get("distance", {}).get("text", "")
-
-        summary += f"{i}. {instr} ({dist})\n"
-
-    # Static map snapshot
-    poly = route.get("overview_polyline", {}).get("points")
-    if poly:
-        static_map_url = (
-            "https://maps.googleapis.com/maps/api/staticmap"
-            f"?size=600x400&path=enc:{poly}&key={api_key}"
-        )
-        summary += "\n---\n"
-        summary += ("Here’s a **static map snapshot** of this route:\n" + static_map_url)
-
-    return summary
-
-
-# Gradio UI
-demo = gr.Interface(
-    fn=get_directions,
-    inputs=[
-        gr.Textbox(label="Source Location", placeholder="e.g. New York City"),
-        gr.Textbox(label="Destination Location", placeholder="e.g. Los Angeles"),
-        gr.Radio(
-            ["driving", "walking", "bicycling", "transit"],
-            label="Travel Mode",
-            value="driving"
-        ),
-        gr.Textbox(
-            label="Departure Time (YYYY-MM-DD HH:MM)",
-            placeholder="e.g. 2025-06-08 14:30"
-        ),
-        gr.Textbox(
-            label="Waypoints (optional, comma-separated)",
-            placeholder="e.g. museum, coffee shop"
-        ),
-    ],
-    outputs=gr.Markdown(label="Directions Summary"),
-    title="Google Maps Directions",
-    description="Get step-by-step directions between two locations using the Google Maps API."
-)
-
-if __name__ == "__main__":
-    demo.launch(mcp_server=True, share=True)
+from dotenv import load_dotenv
+import os
+import googlemaps
+import re
+import gradio as gr
+from datetime import datetime
+
+# Load environment variables and initialize Google Maps client
+dotenv_path = os.getenv('DOTENV_PATH', None)
+if dotenv_path:
+    load_dotenv(dotenv_path)
+else:
+    load_dotenv()
+
+api_key = os.getenv("GOOGLE_API_KEY")
+if not api_key:
+    raise EnvironmentError("GOOGLE_API_KEY not found in environment variables.")
+
+gmaps = googlemaps.Client(key=api_key)
+
+
+def get_places(trip_type: str, location: str) -> list[str]:
+    """
+    Fetches up to 5 popular tourist attractions matching the trip_type in the given location.
+    """
+    query = f"{trip_type} places in {location}"
+    results = gmaps.places(query=query, type="tourist_attraction")
+    places = results.get("results", [])[:5]
+    return [place.get("name") for place in places]
+
+
+def get_restaurants(location: str, prefs: str) -> list[str]:
+    """
+    Fetches up to 3 restaurants near the given location matching the user preferences.
+    """
+    query = f"{prefs} restaurants near {location}"
+    results = gmaps.places(query=query, type="restaurant")
+    restaurants = results.get("results", [])[:3]
+    return [r.get("name") for r in restaurants]
+
+
+def get_directions(
+    source: str,
+    destination: str,
+    mode: str,
+    departure_time: str,
+    waypoints: str = ""
+) -> str:
+    """
+    Returns a Markdown-formatted string with step-by-step directions and a static map URL.
+
+    Args:
+        source: Starting address or place name.
+        destination: Ending address or place name.
+        mode: Transport mode: driving, walking, bicycling, or transit.
+        departure_time: Trip start time in "YYYY-MM-DD HH:MM" format.
+        waypoints: Optional comma-separated list of waypoints.
+
+    Returns:
+        A single Markdown string summarizing the route or an error message.
+    """
+    # Parse departure time
+    try:
+        dt = datetime.strptime(departure_time.strip(), "%Y-%m-%d %H:%M")
+    except ValueError:
+        return (
+            "**⚠️ Invalid time format.** Please enter departure time as YYYY-MM-DD HH:MM."
+        )
+
+    # Build waypoints list
+    wp_list = [w.strip() for w in waypoints.split(",")] if waypoints else None
+
+    # Call Directions API
+    directions = gmaps.directions(
+        origin=source,
+        destination=destination,
+        mode=mode.lower(),
+        departure_time=dt,
+        waypoints=wp_list,
+        optimize_waypoints=False,
+    )
+
+    if not directions:
+        return "**No route found.**"
+
+    # Use first route & leg
+    route = directions[0]
+    leg = route["legs"][0]
+
+    # Build summary
+    summary = f"**Trip from '{leg['start_address']}' to '{leg['end_address']}'**\n"
+    summary += f"- Total distance: {leg['distance']['text']}\n"
+    summary += f"- Estimated duration: {leg['duration']['text']}\n\n"
+    summary += "### Step-by-Step Directions:\n"
+
+    for i, step in enumerate(leg["steps"], start=1):
+        travel_mode = step.get("travel_mode", "").upper()
+        if travel_mode == "TRANSIT":
+            details = step.get("transit_details", {})
+            line = details.get("line", {})
+            vehicle = line.get("vehicle", {}).get("type", "Transit")
+            line_name = line.get("short_name") or line.get("name", "")
+            dep = details.get("departure_stop", {}).get("name", "")
+            arr = details.get("arrival_stop", {}).get("name", "")
+            stops = details.get("num_stops", "")
+            instr = f"Take {vehicle} {line_name} from {dep} to {arr} ({stops} stops)"
+            dist = step.get("distance", {}).get("text", "")
+        else:
+            raw = step.get("html_instructions", "")
+            instr = re.sub(r"<[^>]+>", "", raw)
+            dist = step.get("distance", {}).get("text", "")
+
+        summary += f"{i}. {instr} ({dist})\n"
+
+    # Static map snapshot
+    poly = route.get("overview_polyline", {}).get("points")
+    if poly:
+        static_map_url = (
+            "https://maps.googleapis.com/maps/api/staticmap"
+            f"?size=600x400&path=enc:{poly}&key={api_key}"
+        )
+        summary += "\n---\n"
+        summary += ("Here’s a **static map snapshot** of this route:\n" + static_map_url)
+
+    return summary
+
+
+# Gradio UI
+demo = gr.Interface(
+    fn=get_directions,
+    inputs=[
+        gr.Textbox(label="Source Location", placeholder="e.g. New York City"),
+        gr.Textbox(label="Destination Location", placeholder="e.g. Los Angeles"),
+        gr.Radio(
+            ["driving", "walking", "bicycling", "transit"],
+            label="Travel Mode",
+            value="driving"
+        ),
+        gr.Textbox(
+            label="Departure Time (YYYY-MM-DD HH:MM)",
+            placeholder="e.g. 2025-06-08 14:30"
+        ),
+        gr.Textbox(
+            label="Waypoints (optional, comma-separated)",
+            placeholder="e.g. museum, coffee shop"
+        ),
+    ],
+    outputs=gr.Markdown(label="Directions Summary"),
+    title="Google Maps Directions",
+    description="Get step-by-step directions between two locations using the Google Maps API."
+)
+
+
+demo.launch(mcp_server=True, share=True)