Spaces:
Running
Running
File size: 5,364 Bytes
7f4e4c3 4217047 7f4e4c3 1a9f164 7f4e4c3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 |
import gradio as gr
import numpy as np
import pandas as pd
from gradio_folium import Folium
from smolagents.gradio_ui import pull_messages_from_step
from smolagents.types import handle_agent_output_types, AgentText
from folium import Map, TileLayer, Marker, Icon, Popup
from folium.plugins import Fullscreen
FINAL_MESSAGE_HEADER = "**Final answer/ Réponse finale** \n 🤖⛷️💭"
MAP_URL = "https://{s}.tile.openstreetmap.fr/osmfr/{z}/{x}/{y}.png"
def toggle_visibility(show):
return gr.Textbox(visible=show)
def create_map_from_markers(dataframe: pd.DataFrame) -> Map:
"""
Create a Folium map with markers for each location in the dataframe.
Args:
dataframe (pd.DataFrame): Dataframe containing the locations.
Returns:
Map: Folium map with markers.
"""
dataframe["Latitude"] = dataframe["Latitude"].astype(float, errors='raise')
dataframe["Longitude"] = dataframe["Longitude"].astype(float, errors='raise')
f_map = Map(
location=[dataframe["Latitude"].mean(), dataframe["Longitude"].mean()],
zoom_start=10,
tiles=
TileLayer(
tiles=MAP_URL,
attr="Google",
name="Google Maps",
overlay=True,
control=True,
),
)
for _, row in dataframe.iterrows():
if np.isnan(row["Latitude"]) or np.isnan(row["Longitude"]):
continue
#popup_message = f"<h4 style='color: #d53e2a;'>{row['name'].split(',')[0]}</h4><p style='font-weight:500'>{row['description']}</p>"
#popup_message += f"<a href='https://www.google.com/search?q={row['name']}' target='_blank'><b>Learn more about {row['name'].split(',')[0]}</b></a>"
marker = Marker(
location=[float(row["Latitude"]), float(row["Longitude"])],
icon=Icon(color="blue", icon="fa-map-marker", prefix='fa'),
popup = Popup(f"Infos: <a href='{row['Route Link']}'>{row['Name']}</a>", max_width=300)
)
marker.add_to(f_map)
Fullscreen(position='topright', title='Expand me', title_cancel='Exit me', force_separate_button=True).add_to(f_map)
#bounds = [[float(row["Latitude"]), float(row["Longitude"])] for _, row in dataframe.iterrows()]
#f_map.fit_bounds(bounds, padding=(100, 100))
return f_map
def update_map_on_selection(row: pd.Series, df_routes: gr.State) -> Map:
"""
Update the map with a marker at the selected location.
Args:
row (pd.Series): Selected row from the dataframe.
Returns:
Map: Updated Folium map.
"""
row = df_routes.loc[df_routes['Name'] == row['Name']]
f_map = Map(
location=[row["Latitude"][0], row["Longitude"][0]],
tiles=TileLayer(
tiles=MAP_URL,
attr="Google",
name="Google Maps",
overlay=True,
control=True,
),
)
popup = Popup(f"Infos: <a href='{row['Route Link'][0]}'>{row['Name'][0]}</a>", max_width=300)
Marker(
location=[row["Latitude"][0], row["Longitude"][0]],
icon=Icon(color="blue", icon="fa-map-marker", prefix='fa'),
popup=popup
).add_to(f_map)
Fullscreen(position='topright', title='Expand', title_cancel='Exit', force_separate_button=True).add_to(f_map)
return f_map
# Simplified interaction function
def interact_with_agent(agent, prompt, messages, df_routes, additional_args):
messages.append(gr.ChatMessage(role="user", content=prompt))
yield (messages, df_routes, gr.Textbox(value=FINAL_MESSAGE_HEADER, container=True))
for msg, _df_routes, final_message in stream_to_gradio(
agent,
df_routes=df_routes,
task=prompt,
reset_agent_memory=True,
additional_args=additional_args,
):
messages.append(msg)
yield (messages, _df_routes, final_message)
yield (messages, _df_routes, final_message)
def stream_to_gradio(
agent,
df_routes,
task: str,
test_mode: bool = False,
reset_agent_memory: bool = False,
**kwargs,
):
"""Runs an agent with the given task and streams the messages from the agent as gradio ChatMessages."""
for step_log in agent.run(task, stream=True, reset=reset_agent_memory, **kwargs):
for message in pull_messages_from_step(step_log, test_mode=test_mode):
yield (message, df_routes, gr.Markdown(value=FINAL_MESSAGE_HEADER , container=True))
final_answer = step_log # Last log is the run's final_answer
final_answer = handle_agent_output_types(final_answer)
if isinstance(final_answer, dict):
final_message = final_answer.get("message")
itineraries = final_answer.get("itineraries")
if itineraries:
df_routes = pd.DataFrame(itineraries)
df_routes.columns = ["id", "Name", "Latitude", "Longitude", "Route Link"]
else:
final_message = final_answer
text_output = gr.Markdown(value=FINAL_MESSAGE_HEADER + f": {str(final_message)}", container=True)
if isinstance(final_answer, AgentText):
yield (gr.ChatMessage(
role="assistant",
content=f"**Final answer:**\n{str(final_message)}\n",
), df_routes, text_output)
else:
yield (gr.ChatMessage(role="assistant", content=str(final_message)), df_routes, text_output)
|