Update app.py

app.py

@@ -3,7 +3,7 @@ from gradio_folium import Folium
 from folium import Map
 import pandas as pd
 import pathlib
-import os # for file search of csv files
+import os
 
 # Function to list all CSV files in the current directory
 def list_csv_files():
@@ -16,9 +16,9 @@ def update_map(csv_file):
     return df, Folium(value=Map(location=[df.iloc[0]['Latitude'], df.iloc[0]['Longitude']], zoom_start=10), height=400)
 
 # Function to update location on map based on selected data row
-def select(data: gr.Dataframe):
-    row = data.iloc[0, :]
-    return Map(location=[row['Latitude'], row['Longitude']], zoom_start=10)
+def select(row_data):
+    lat, lon = row_data['Latitude'], row_data['Longitude']
+    return Folium(value=Map(location=[lat, lon], zoom_start=10), height=400)
 
 # Gradio Blocks
 with gr.Blocks() as demo:
@@ -38,7 +38,7 @@ with gr.Blocks() as demo:
 
     # Interaction logic
     csv_selector.change(update_map, inputs=csv_selector, outputs=[data, map_component])
-    data.select(select, inputs=data, outputs=map_component)
+    data.select(select, inputs=data.row, outputs=map_component)  # Updated line
     reload_button.click(update_map, inputs=csv_selector, outputs=[data, map_component])
 
 # Launch the app