chore: Update vehicle destination in calculate_route function
Browse files- kitt.py +53 -98
- skills/common.py +1 -1
- skills/routing.py +43 -18
kitt.py
CHANGED
@@ -1,42 +1,20 @@
|
|
1 |
import gradio as gr
|
2 |
-
import plotly.express as px
|
3 |
import requests
|
4 |
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
)
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
) # "sr=16000" ensures that the sampling rate is as required
|
17 |
-
|
18 |
-
# process the audio array
|
19 |
-
input_features = processor(
|
20 |
-
audio_array, sampling_rate, return_tensors="pt"
|
21 |
-
).input_features
|
22 |
-
predicted_ids = modelw.generate(input_features)
|
23 |
-
transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)
|
24 |
-
quest_processing = FnAnswer(
|
25 |
-
general_context, transcription, place, time, delete_history, state
|
26 |
-
)
|
27 |
-
state = quest_processing[2]
|
28 |
-
print("langue " + quest_processing[3])
|
29 |
-
|
30 |
-
tts.tts_to_file(
|
31 |
-
text=str(quest_processing[0]),
|
32 |
-
file_path="output.wav",
|
33 |
-
speaker_wav=f"Audio_Files/{voice}.wav",
|
34 |
-
language=quest_processing[3],
|
35 |
-
emotion="angry",
|
36 |
-
)
|
37 |
|
38 |
-
|
39 |
-
return
|
40 |
|
41 |
|
42 |
# to be able to use the microphone on chrome, you will have to go to chrome://flags/#unsafely-treat-insecure-origin-as-secure and enter http://10.186.115.21:7860/
|
@@ -47,42 +25,6 @@ def transcript(
|
|
47 |
# What's the closest restaurant from here?
|
48 |
|
49 |
|
50 |
-
import gradio as gr
|
51 |
-
|
52 |
-
shortcut_js = """
|
53 |
-
<script>
|
54 |
-
function shortcuts(e) {
|
55 |
-
var event = document.all ? window.event : e;
|
56 |
-
switch (e.target.tagName.toLowerCase()) {
|
57 |
-
case "input":
|
58 |
-
case "textarea":
|
59 |
-
break;
|
60 |
-
default:
|
61 |
-
if (e.key.toLowerCase() == "r" && e.ctrlKey) {
|
62 |
-
console.log("recording")
|
63 |
-
document.getElementById("recorder").start_recording();
|
64 |
-
}
|
65 |
-
if (e.key.toLowerCase() == "s" && e.ctrlKey) {
|
66 |
-
console.log("stopping")
|
67 |
-
document.getElementById("recorder").stop_recording();
|
68 |
-
}
|
69 |
-
}
|
70 |
-
}
|
71 |
-
document.addEventListener('keypress', shortcuts, false);
|
72 |
-
</script>
|
73 |
-
"""
|
74 |
-
|
75 |
-
# with gr.Blocks(head=shortcut_js) as demo:
|
76 |
-
# action_button = gr.Button(value="Name", elem_id="recorder")
|
77 |
-
# textbox = gr.Textbox()
|
78 |
-
# action_button.click(lambda : "button pressed", None, textbox)
|
79 |
-
|
80 |
-
# demo.launch()
|
81 |
-
|
82 |
-
|
83 |
-
# Generate options for hours (00-23)
|
84 |
-
hour_options = [f"{i:02d}:00:00" for i in range(24)]
|
85 |
-
|
86 |
model_answer = ""
|
87 |
general_context = ""
|
88 |
# Define the initial state with some initial context.
|
@@ -93,25 +35,29 @@ initial_context = initial_state["context"]
|
|
93 |
|
94 |
|
95 |
with gr.Blocks(theme=gr.themes.Default()) as demo:
|
|
|
|
|
|
|
96 |
|
97 |
with gr.Row():
|
98 |
with gr.Column(scale=1, min_width=300):
|
99 |
time_picker = gr.Dropdown(
|
100 |
-
choices=hour_options, label="What time is it?", value="08:00
|
101 |
)
|
102 |
history = gr.Radio(
|
103 |
-
["Yes", "No"], label="Maintain the conversation history?", value="No"
|
104 |
)
|
105 |
voice_character = gr.Radio(
|
106 |
choices=[
|
107 |
-
"
|
108 |
"Eddie Murphy",
|
109 |
"David Attenborough",
|
110 |
-
"
|
111 |
],
|
112 |
label="Choose a voice",
|
113 |
-
value="
|
114 |
show_label=True,
|
|
|
115 |
)
|
116 |
emotion = gr.Radio(
|
117 |
choices=["Cheerful", "Grumpy"],
|
@@ -119,36 +65,36 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
|
|
119 |
value="Cheerful",
|
120 |
show_label=True,
|
121 |
)
|
122 |
-
# place = gr.Radio(
|
123 |
-
# choices=[
|
124 |
-
# "Luxembourg Gare, Luxembourg",
|
125 |
-
# "Kirchberg Campus, Kirchberg",
|
126 |
-
# "Belval Campus, Belval",
|
127 |
-
# "Eiffel Tower, Paris",
|
128 |
-
# "Thionville, France",
|
129 |
-
# ],
|
130 |
-
# label="Choose a location for your car",
|
131 |
-
# value="Kirchberg Campus, Kirchberg",
|
132 |
-
# show_label=True,
|
133 |
-
# )
|
134 |
origin = gr.Textbox(
|
135 |
value="Luxembourg Gare, Luxembourg", label="Origin", interactive=True
|
136 |
)
|
137 |
destination = gr.Textbox(
|
138 |
-
value="Kirchberg Campus,
|
139 |
label="Destination",
|
140 |
interactive=True,
|
141 |
)
|
142 |
-
|
143 |
-
type="filepath", label="input audio", elem_id="recorder"
|
144 |
-
)
|
145 |
with gr.Column(scale=2, min_width=600):
|
146 |
map_plot = gr.Plot()
|
147 |
-
|
148 |
-
destination.submit(fn=calculate_route, outputs=map_plot)
|
149 |
-
output_audio = gr.Audio(label="output audio")
|
150 |
# map_if = gr.Interface(fn=plot_map, inputs=year_input, outputs=map_plot)
|
151 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
152 |
# iface = gr.Interface(
|
153 |
# fn=transcript,
|
154 |
# inputs=[
|
@@ -165,12 +111,21 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
|
|
165 |
# head=shortcut_js,
|
166 |
# )
|
167 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
168 |
# close all interfaces open to make the port available
|
169 |
gr.close_all()
|
170 |
# Launch the interface.
|
171 |
|
172 |
-
|
173 |
-
|
174 |
-
|
|
|
175 |
|
176 |
# iface.launch(debug=True, share=False, server_name="0.0.0.0", server_port=7860, ssl_verify=False)
|
|
|
1 |
import gradio as gr
|
|
|
2 |
import requests
|
3 |
|
4 |
+
import skills
|
5 |
+
from skills.common import config, vehicle
|
6 |
+
from skills.routing import calculate_route
|
7 |
+
|
8 |
+
|
9 |
+
# Generate options for hours (00-23)
|
10 |
+
hour_options = [f"{i:02d}:00" for i in range(24)]
|
11 |
+
|
12 |
+
def set_time(time_picker):
|
13 |
+
vehicle.time = time_picker
|
14 |
+
return vehicle.model_dump_json()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
|
16 |
+
def get_vehicle_status(state):
|
17 |
+
return state.value["vehicle"].model_dump_json()
|
18 |
|
19 |
|
20 |
# to be able to use the microphone on chrome, you will have to go to chrome://flags/#unsafely-treat-insecure-origin-as-secure and enter http://10.186.115.21:7860/
|
|
|
25 |
# What's the closest restaurant from here?
|
26 |
|
27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
28 |
model_answer = ""
|
29 |
general_context = ""
|
30 |
# Define the initial state with some initial context.
|
|
|
35 |
|
36 |
|
37 |
with gr.Blocks(theme=gr.themes.Default()) as demo:
|
38 |
+
state = gr.State(
|
39 |
+
value={"context": initial_context, "query": "", "vehicle": vehicle, "route_points": []}
|
40 |
+
)
|
41 |
|
42 |
with gr.Row():
|
43 |
with gr.Column(scale=1, min_width=300):
|
44 |
time_picker = gr.Dropdown(
|
45 |
+
choices=hour_options, label="What time is it? (HH:MM)", value="08:00", interactive=True
|
46 |
)
|
47 |
history = gr.Radio(
|
48 |
+
["Yes", "No"], label="Maintain the conversation history?", value="No", interactive=True
|
49 |
)
|
50 |
voice_character = gr.Radio(
|
51 |
choices=[
|
52 |
+
"Morgan Freeman",
|
53 |
"Eddie Murphy",
|
54 |
"David Attenborough",
|
55 |
+
"Rick Sanches",
|
56 |
],
|
57 |
label="Choose a voice",
|
58 |
+
value="Morgan Freeman",
|
59 |
show_label=True,
|
60 |
+
interactive=True,
|
61 |
)
|
62 |
emotion = gr.Radio(
|
63 |
choices=["Cheerful", "Grumpy"],
|
|
|
65 |
value="Cheerful",
|
66 |
show_label=True,
|
67 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
68 |
origin = gr.Textbox(
|
69 |
value="Luxembourg Gare, Luxembourg", label="Origin", interactive=True
|
70 |
)
|
71 |
destination = gr.Textbox(
|
72 |
+
value="Kirchberg Campus, Luxembourg",
|
73 |
label="Destination",
|
74 |
interactive=True,
|
75 |
)
|
76 |
+
|
|
|
|
|
77 |
with gr.Column(scale=2, min_width=600):
|
78 |
map_plot = gr.Plot()
|
79 |
+
|
|
|
|
|
80 |
# map_if = gr.Interface(fn=plot_map, inputs=year_input, outputs=map_plot)
|
81 |
|
82 |
+
with gr.Row():
|
83 |
+
with gr.Column():
|
84 |
+
recorder = gr.Audio(
|
85 |
+
type="filepath", label="Input audio", elem_id="recorder"
|
86 |
+
)
|
87 |
+
input_text = gr.Textbox(
|
88 |
+
value="How is the weather?", label="Input text", interactive=True
|
89 |
+
)
|
90 |
+
vehicle_status = gr.Textbox(
|
91 |
+
value=get_vehicle_status(state), label="Vehicle status", interactive=False
|
92 |
+
)
|
93 |
+
with gr.Column():
|
94 |
+
output_audio = gr.Audio(label="output audio")
|
95 |
+
output_text = gr.TextArea(
|
96 |
+
value="", label="Output text", interactive=False
|
97 |
+
)
|
98 |
# iface = gr.Interface(
|
99 |
# fn=transcript,
|
100 |
# inputs=[
|
|
|
111 |
# head=shortcut_js,
|
112 |
# )
|
113 |
|
114 |
+
# Update plot based on the origin and destination
|
115 |
+
# Sets the current location and destination
|
116 |
+
origin.submit(fn=calculate_route, inputs=[origin, destination], outputs=[map_plot, vehicle_status])
|
117 |
+
destination.submit(fn=calculate_route, inputs=[origin, destination], outputs=[map_plot, vehicle_status])
|
118 |
+
|
119 |
+
# Update time based on the time picker
|
120 |
+
time_picker.select(fn=set_time, inputs=[time_picker], outputs=[vehicle_status])
|
121 |
+
|
122 |
# close all interfaces open to make the port available
|
123 |
gr.close_all()
|
124 |
# Launch the interface.
|
125 |
|
126 |
+
if __name__ == "__main__":
|
127 |
+
demo.launch(
|
128 |
+
debug=True, server_name="0.0.0.0", server_port=7860, ssl_verify=False
|
129 |
+
)
|
130 |
|
131 |
# iface.launch(debug=True, share=False, server_name="0.0.0.0", server_port=7860, ssl_verify=False)
|
skills/common.py
CHANGED
@@ -58,5 +58,5 @@ vehicle = VehicleStatus(
|
|
58 |
location_coordinates=(49.6002, 6.1296),
|
59 |
date="2025-03-29",
|
60 |
time="08:00:20",
|
61 |
-
destination="Kirchberg Campus,
|
62 |
)
|
|
|
58 |
location_coordinates=(49.6002, 6.1296),
|
59 |
date="2025-03-29",
|
60 |
time="08:00:20",
|
61 |
+
destination="Kirchberg Campus, Luxembourg"
|
62 |
)
|
skills/routing.py
CHANGED
@@ -17,36 +17,56 @@ def find_coordinates(address):
|
|
17 |
return lat, lon
|
18 |
|
19 |
|
20 |
-
def
|
21 |
-
|
22 |
-
destination = "49.586745,6.140002"
|
23 |
-
|
24 |
-
url = f"https://api.tomtom.com/routing/1/calculateRoute/{origin}:{destination}/json?key={config.TOMTOM_API_KEY}"
|
25 |
-
response = requests.get(url)
|
26 |
-
data = response.json()
|
27 |
|
28 |
lats = []
|
29 |
lons = []
|
30 |
|
31 |
-
for point in
|
32 |
lats.append(point["latitude"])
|
33 |
lons.append(point["longitude"])
|
34 |
# fig = px.line_geo(lat=lats, lon=lons)
|
35 |
# fig.update_geos(fitbounds="locations")
|
36 |
|
37 |
-
fig = px.line_mapbox(
|
|
|
|
|
38 |
|
39 |
fig.update_layout(
|
40 |
mapbox_style="open-street-map",
|
41 |
-
mapbox_zoom=12,
|
42 |
-
mapbox_center_lat=lats[0],
|
43 |
-
mapbox_center_lon=lons[0],
|
44 |
)
|
45 |
-
fig.
|
46 |
-
|
47 |
return fig
|
48 |
|
49 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
50 |
def find_route_tomtom(
|
51 |
lat_depart="0",
|
52 |
lon_depart="0",
|
@@ -75,8 +95,6 @@ def find_route_tomtom(
|
|
75 |
|
76 |
result = response["routes"][0]["summary"]
|
77 |
|
78 |
-
|
79 |
-
|
80 |
distance_m = result["lengthInMeters"]
|
81 |
duration_s = result["travelTimeInSeconds"]
|
82 |
arrival_time = result["arrivalTime"]
|
@@ -112,7 +130,11 @@ def find_route(destination=""):
|
|
112 |
lat_depart, lon_depart, lat_dest, lon_dest, departure_time
|
113 |
)
|
114 |
|
115 |
-
distance, duration, arrival_time =
|
|
|
|
|
|
|
|
|
116 |
|
117 |
# Calculate distance in kilometers (1 meter = 0.001 kilometers)
|
118 |
distance_km = distance * 0.001
|
@@ -131,4 +153,7 @@ def find_route(destination=""):
|
|
131 |
arrival_hour_display = arrival_time.strftime("%H:%M")
|
132 |
|
133 |
# return the distance and time
|
134 |
-
return
|
|
|
|
|
|
|
|
17 |
return lat, lon
|
18 |
|
19 |
|
20 |
+
def plot_route(points):
|
21 |
+
import plotly.express as px
|
|
|
|
|
|
|
|
|
|
|
22 |
|
23 |
lats = []
|
24 |
lons = []
|
25 |
|
26 |
+
for point in points:
|
27 |
lats.append(point["latitude"])
|
28 |
lons.append(point["longitude"])
|
29 |
# fig = px.line_geo(lat=lats, lon=lons)
|
30 |
# fig.update_geos(fitbounds="locations")
|
31 |
|
32 |
+
fig = px.line_mapbox(
|
33 |
+
lat=lats, lon=lons, zoom=12, height=600, color_discrete_sequence=["red"]
|
34 |
+
)
|
35 |
|
36 |
fig.update_layout(
|
37 |
mapbox_style="open-street-map",
|
38 |
+
# mapbox_zoom=12,
|
|
|
|
|
39 |
)
|
40 |
+
fig.update_geos(fitbounds="locations")
|
41 |
+
fig.update_layout(margin={"r": 20, "t": 20, "l": 20, "b": 20})
|
42 |
return fig
|
43 |
|
44 |
|
45 |
+
def calculate_route(origin, destination):
|
46 |
+
"""This function is called when the origin or destination is updated in the GUI. It calculates the route between the origin and destination."""
|
47 |
+
print(f"calculate_route(origin: {origin}, destination: {destination})")
|
48 |
+
origin_coords = find_coordinates(origin)
|
49 |
+
destination_coords = find_coordinates(destination)
|
50 |
+
|
51 |
+
orig_coords_str = ",".join(map(str, origin_coords))
|
52 |
+
dest_coords_str = ",".join(map(str, destination_coords))
|
53 |
+
print(f"origin_coords: {origin_coords}, destination_coords: {destination_coords}")
|
54 |
+
|
55 |
+
vehicle.destination = destination
|
56 |
+
vehicle.location_coordinates = origin_coords
|
57 |
+
vehicle.location = origin
|
58 |
+
|
59 |
+
# origin = "49.631997,6.171029"
|
60 |
+
# destination = "49.586745,6.140002"
|
61 |
+
|
62 |
+
url = f"https://api.tomtom.com/routing/1/calculateRoute/{orig_coords_str}:{dest_coords_str}/json?key={config.TOMTOM_API_KEY}"
|
63 |
+
response = requests.get(url)
|
64 |
+
data = response.json()
|
65 |
+
points = data["routes"][0]["legs"][0]["points"]
|
66 |
+
|
67 |
+
return plot_route(points), vehicle.model_dump_json()
|
68 |
+
|
69 |
+
|
70 |
def find_route_tomtom(
|
71 |
lat_depart="0",
|
72 |
lon_depart="0",
|
|
|
95 |
|
96 |
result = response["routes"][0]["summary"]
|
97 |
|
|
|
|
|
98 |
distance_m = result["lengthInMeters"]
|
99 |
duration_s = result["travelTimeInSeconds"]
|
100 |
arrival_time = result["arrivalTime"]
|
|
|
130 |
lat_depart, lon_depart, lat_dest, lon_dest, departure_time
|
131 |
)
|
132 |
|
133 |
+
distance, duration, arrival_time = (
|
134 |
+
trip_info["distance_m"],
|
135 |
+
trip_info["duration_s"],
|
136 |
+
trip_info["arrival_time"],
|
137 |
+
)
|
138 |
|
139 |
# Calculate distance in kilometers (1 meter = 0.001 kilometers)
|
140 |
distance_km = distance * 0.001
|
|
|
153 |
arrival_hour_display = arrival_time.strftime("%H:%M")
|
154 |
|
155 |
# return the distance and time
|
156 |
+
return (
|
157 |
+
f"The route to {destination} is {distance_km:.2f} km and {time_display}. Leaving now, the arrival time is estimated at {arrival_hour_display}.",
|
158 |
+
raw_response["routes"][0]["legs"][0]["points"],
|
159 |
+
)
|