import streamlit as st
import folium
from folium.plugins import HeatMap
import pandas as pd

# Título de la aplicación
st.title("Heatmap de Población en México")

# Cargar datos
data = pd.DataFrame({
    'Entidad Federativa': ['Aguascalientes', 'Baja California', 'Baja California Sur', 'Campeche', 'Coahuila de Zaragoza',
                           'Colima', 'Chiapas', 'Chihuahua', 'Ciudad de México', 'Durango', 'Guanajuato', 'Guerrero',
                           'Hidalgo', 'Jalisco', 'México', 'Michoacán de Ocampo', 'Morelos', 'Nayarit', 'Nuevo León',
                           'Oaxaca', 'Puebla', 'Querétaro', 'Quintana Roo', 'San Luis Potosí', 'Sinaloa', 'Sonora',
                           'Tabasco', 'Tamaulipas', 'Tlaxcala', 'Veracruz de Ignacio de la Llave', 'Yucatán', 'Zacatecas'],
    'Poblacion': [728924, 1868431, 392568, 471424, 1583102, 370769, 2837881, 1888047, 4805017, 927784, 3170480,
                  1840073, 1601462, 4249696, 8741123, 2442505, 1020673, 623178, 2893492, 2157305, 3423163,
                  1211647, 921206, 1449804, 1532128, 1472643, 1228927, 1791595, 693083, 4190805, 1180619, 83108],
    'Latitud': [21.8818, 30.8406, 25.6212, 19.8301, 27.0587, 19.2452, 16.7573, 28.632, 19.4326, 24.0277, 21.019,
                17.6753, 20.0911, 20.6595, 19.4326, 19.5665, 18.6813, 21.7514, 25.5922, 17.0732, 19.0414, 20.5888,
                19.1817, 22.1565, 25.1721, 29.2972, 17.8409, 24.2669, 19.3149, 19.1738, 20.7099, 22.7709],
    'Longitud': [-102.2916, -115.2838, -111.0571, -90.5349, -101.7068, -103.7244, -93.1292, -106.0691, -99.1332,
                 -104.6532, -101.2574, -99.5516, -98.7624, -103.3494, -99.1332, -101.7068, -99.1013, -104.8455,
                 -99.9962, -96.7266, -98.2063, -100.3899, -88.4791, -100.9855, -107.4795, -110.33, -92.6189, -98.8363,
                 -98.2405, -96.1342, -89.0943, -102.5832]
})

# Mapa de México
mexico_map = folium.Map(location=[23.6345, -102.5528], zoom_start=5)

# Crear el heatmap
heat_data = data[['Latitud', 'Longitud', 'Poblacion']].values.tolist()
HeatMap(heat_data, radius=15).add_to(mexico_map)

# Mostrar el mapa
folium_static(mexico_map)