import folium
import geopandas as gpd
import gradio as gr

# Cargar datos geoespaciales de los barrios (por ejemplo, un archivo GeoJSON)
barrios_gdf = gpd.read_file("barrios_y_veredas.geojson")

# Crear un mapa centrado en Medellín
medellin_map = folium.Map(location=[6.2442, -75.5812], zoom_start=12.3)

# Definir una función para personalizar el estilo de las líneas
def style_function(feature):
    return {
        'color': 'black',  # Color de las líneas
        'weight': 0.5  # Grosor de las líneas (ajusta este valor)
    }

# Agregar los polígonos del archivo GeoJSON al mapa con estilo personalizado
folium.GeoJson(barrios_gdf, style_function=style_function).add_to(medellin_map)

# Crear una función para mostrar el mapa en Gradio
def mostrar_mapa():
    return medellin_map._repr_html_()

# Crear la interfaz de Gradio y mostrar el mapa al inicio
app = gr.Interface(fn=mostrar_mapa, inputs=None, outputs="html")
app.launch()