kitt/core/utils.py

import json
import re
from typing import List, Optional, Tuple, Union


def plot_route(points, vehicle: Union[tuple[float, float], None] = None):
    import plotly.express as px

    lats = []
    lons = []

    for point in points:
        lats.append(point["latitude"])
        lons.append(point["longitude"])
    # fig = px.line_geo(lat=lats, lon=lons)
    # fig.update_geos(fitbounds="locations")

    fig = px.line_mapbox(lat=lats, lon=lons, color_discrete_sequence=["red"], zoom=6)

    if vehicle:
        fig.add_trace(
            px.scatter_mapbox(
                lat=[vehicle[0]],
                lon=[vehicle[1]],
                color_discrete_sequence=["blue"],
            ).data[0]
        )

    fig.update_layout(
        mapbox_style="open-street-map",
        # mapbox_zoom=12,
    )
    fig.update_geos(fitbounds="locations")
    fig.update_layout(height=600, margin={"r": 20, "t": 20, "l": 20, "b": 20})
    return fig


def extract_json_from_markdown(text):
    """
    Extracts the JSON string from the given text using a regular expression pattern.

    Args:
        text (str): The input text containing the JSON string.

    Returns:
        dict: The JSON data loaded from the extracted string, or None if the JSON string is not found.
    """
    json_pattern = r"```json\r?\n(.*?)\r?\n```"
    match = re.search(json_pattern, text, re.DOTALL)
    if match:
        json_string = match.group(1)
        try:
            data = json.loads(json_string)
            return data
        except json.JSONDecodeError as e:
            print(f"Error decoding JSON string: {e}")
    else:
        print("JSON string not found in the text.")
    return None