modulo geocoding

app.py

@@ -1,93 +1,17 @@
-import os
-
 import folium
 import matplotlib.pyplot as plt
-import requests
 import streamlit as st
-from dotenv import load_dotenv
 from sklearn.cluster import KMeans
 from streamlit_folium import folium_static
-
-load_dotenv()
+import geocoding
 
 st.set_page_config(layout='wide')  # Para usar todo el ancho de la página
 
-API_KEY = os.getenv('API_KEY')
-# BASE_URL = os.getenv('BASE_URL')
-BASE_URL = 'https://maps.googleapis.com/maps/api/geocode/json'
-
-def geocode_address(address):
-    api_key = API_KEY
-    base_url = BASE_URL
-    params = {
-        'address': address,
-        'key': api_key
-        }
-
-    try:
-        response = requests.get(base_url, params=params)
-        data = response.json()
-
-        if data['status'] == 'OK' and len(data['results']) > 0:
-            location = data['results'][0]['geometry']['location']
-            latitude = location['lat']
-            longitude = location['lng']
-
-            return latitude, longitude
-        else:
-            st.error(
-                    'No se encontraron resultados para la dirección especificada.'
-                    )
-    except requests.exceptions.RequestException as e:
-        st.error('Error en la solicitud:', e)
-
-
-direcciones = [
-    'San Martín y Garibaldi',
-    'Avenida Emilio Civit s/n',
-    'Parque General San Martín',
-    'Calle Sarmiento, entre las calles Patricias Mendocinas y Garibaldi',
-    'Calle Belgrano y España',
-    'Calle Las Heras 50',
-    'Plaza Independencia',
-    'Avenida España y Costanera',
-    'Calle 9 de Julio 1228',
-    'Calle Chile 1754',
-    'Avenida Arístides Villanueva',
-    'Avenida Emilio Civit y España',
-    'Calle Chile y Avenida Colón',
-    'Calle Emilio Civit y Avenida San Martín',
-    'Acceso Este y Avenida San Francisco de Asís',
-    'Calle San Martín y Avellaneda',
-    'Plaza Pedro del Castillo',
-    'Calle San Martín y Avenida España',
-    'Calle Emilio Civit y Avenida San Martín',
-    'Avenida España',
-    'Calle Avellaneda y Patricias Mendocinas',
-    'Parque General San Martín',
-    'Ruta Nacional 7',
-    'Avenida Costanera y calle Peltier',
-    'Calle Montecaseros 2625',
-    'Calle Francisco Delgado 1220',
-    'Ruta Provincial 86, s/n',
-    'Ruta 15, km 23',
-    'Calle San Martín 2044',
-    'Ruta 7 y Acceso Sur'
-    ]
-
-coordenadas = []
-for direccion in direcciones:
-    resultado = geocode_address(direccion + ', Capital, Mendoza, Argentina')
-    if resultado:
-        coordenadas.append(resultado)
-    else:
-        coordenadas.append((None, None))
-
-max_k = len(coordenadas)
+max_k = len(geocoding.coordenadas)
 inertias = []
 for k in range(2, max_k + 1):
     modelo = KMeans(n_clusters=k, random_state=42)
-    modelo.fit(coordenadas)
+    modelo.fit(geocoding.coordenadas)
     inertias.append(modelo.inertia_)
 
 # Titulo
@@ -111,7 +35,7 @@ k_optimo = st.number_input(
         )
 
 modelo = KMeans(n_clusters=k_optimo, random_state=42)
-modelo.fit(coordenadas)
+modelo.fit(geocoding.coordenadas)
 
 etiquetas = modelo.labels_
 num_zonas = len(set(etiquetas))
@@ -120,10 +44,10 @@ zonas = {}
 for i, etiqueta in enumerate(etiquetas):
     if etiqueta not in zonas:
         zonas[etiqueta] = []
-    zonas[etiqueta].append(direcciones[i])
+    zonas[etiqueta].append(geocoding.direcciones[i])
 
 # Crear el mapa centrado en la primera coordenada
-primer_coordenada = coordenadas[0]
+primer_coordenada = geocoding.coordenadas[0]
 mapa = folium.Map(location=primer_coordenada, zoom_start=15)
 
 colors = ['red', 'blue', 'green', 'purple', 'orange', 'darkred', 'lightred',
@@ -131,7 +55,7 @@ colors = ['red', 'blue', 'green', 'purple', 'orange', 'darkred', 'lightred',
           'darkpurple', 'white', 'pink', 'lightblue', 'lightgreen', 'gray',
           'black', 'lightgray']  # Colores disponibles en Folium
 
-for i, coordenada in enumerate(coordenadas):
+for i, coordenada in enumerate(geocoding.coordenadas):
     zona = etiquetas[i] + 1
     color = colors[zona % len(colors)]
     if coordenada != (None, None):

geocoding.py

@@ -0,0 +1,80 @@
+import os
+
+import requests
+import streamlit as st
+from dotenv import load_dotenv
+
+load_dotenv()
+
+API_KEY = os.getenv('API_KEY')
+# BASE_URL = os.getenv('BASE_URL')
+BASE_URL = 'https://maps.googleapis.com/maps/api/geocode/json'
+
+
+def geocode_address(address):
+    api_key = API_KEY
+    base_url = BASE_URL
+    params = {
+        'address': address,
+        'key': api_key
+        }
+
+    try:
+        response = requests.get(base_url, params=params)
+        data = response.json()
+
+        if data['status'] == 'OK' and len(data['results']) > 0:
+            location = data['results'][0]['geometry']['location']
+            latitude = location['lat']
+            longitude = location['lng']
+
+            return latitude, longitude
+        else:
+            st.error(
+                    'No se encontraron resultados para la dirección '
+                    'especificada. Cargaste la API_KEY: https://console.cloud.google.com/google/maps-apis/credentials'
+                    )
+    except requests.exceptions.RequestException as e:
+        st.error('Error en la solicitud:', e)
+
+
+direcciones = [
+    'San Martín y Garibaldi',
+    'Avenida Emilio Civit s/n',
+    'Parque General San Martín',
+    'Calle Sarmiento, entre las calles Patricias Mendocinas y Garibaldi',
+    'Calle Belgrano y España',
+    'Calle Las Heras 50',
+    'Plaza Independencia',
+    'Avenida España y Costanera',
+    'Calle 9 de Julio 1228',
+    'Calle Chile 1754',
+    'Avenida Arístides Villanueva',
+    'Avenida Emilio Civit y España',
+    'Calle Chile y Avenida Colón',
+    'Calle Emilio Civit y Avenida San Martín',
+    'Acceso Este y Avenida San Francisco de Asís',
+    'Calle San Martín y Avellaneda',
+    'Plaza Pedro del Castillo',
+    'Calle San Martín y Avenida España',
+    'Calle Emilio Civit y Avenida San Martín',
+    'Avenida España',
+    'Calle Avellaneda y Patricias Mendocinas',
+    'Parque General San Martín',
+    'Ruta Nacional 7',
+    'Avenida Costanera y calle Peltier',
+    'Calle Montecaseros 2625',
+    'Calle Francisco Delgado 1220',
+    'Ruta Provincial 86, s/n',
+    'Ruta 15, km 23',
+    'Calle San Martín 2044',
+    'Ruta 7 y Acceso Sur'
+    ]
+
+coordenadas = []
+for direccion in direcciones:
+    resultado = geocode_address(direccion + ', Capital, Mendoza, Argentina')
+    if resultado:
+        coordenadas.append(resultado)
+    else:
+        coordenadas.append((None, None))