Update app.py

app.py

@@ -49,17 +49,17 @@ def parse_text_file(filepath):
 def parse_html_file(filepath):
     """
     Parseo para archivos HTML (por ejemplo, de Wikipedia).
-    Extrae el título desde la etiqueta <title> y las coordenadas desde el bloque "wgCoordinates".
-    Si la coordenada es '-' se asigna None.
+    Extrae el título, las coordenadas desde el bloque "wgCoordinates" y, si es posible,
+    la sección histórica (buscando un encabezado que contenga "History" o "Historia").
     """
     with open(filepath, 'r', encoding='utf-8') as f:
         content = f.read()
     
-    # Extraer el título usando BeautifulSoup
+    # Parsear con BeautifulSoup
     soup = BeautifulSoup(content, "html.parser")
     title = soup.title.string.strip() if soup.title else os.path.splitext(os.path.basename(filepath))[0]
     
-    # Buscar las coordenadas en el contenido HTML (ej. en el bloque "wgCoordinates")
+    # Extraer coordenadas (pueden venir en el bloque "wgCoordinates")
     coord_match = re.search(r'"wgCoordinates":\s*{\s*"lat":\s*([0-9\.-]+),\s*"lon":\s*([0-9\.-]+)', content)
     if coord_match:
         lat_str = coord_match.group(1)
@@ -75,11 +75,24 @@ def parse_html_file(filepath):
     else:
         lat, lon = None, None
 
+    # Intentar extraer datos históricos: se busca un encabezado que contenga "History" o "Historia"
+    history = ""
+    history_header = soup.find(lambda tag: tag.name in ["h2", "h3"] and ("History" in tag.get_text() or "Historia" in tag.get_text()))
+    if history_header:
+        history_parts = []
+        # Recorrer los hermanos hasta llegar a otro encabezado del mismo nivel (o superior)
+        for sibling in history_header.find_next_siblings():
+            if sibling.name and sibling.name.startswith("h"):
+                break
+            history_parts.append(sibling.get_text(strip=True))
+        history = "\n".join(history_parts)
+         
     return {
         "name": title,
         "lat": lat,
         "lon": lon,
-        "content": content
+        "content": content,
+        "history": history
     }
 
 def load_data_from_directory(directory, parser_func):
@@ -106,7 +119,7 @@ news_dir = os.path.join("data", "news")
 landmarks = load_data_from_directory(landmarks_dir, parse_text_file)
 news_data = load_data_from_directory(news_dir, parse_text_file)
 
-# Para municipalities, se usa el parseo de HTML
+# Para municipalities, se usa el parseo de HTML (que ahora incluye datos históricos)
 municipalities = load_data_from_directory(municipalities_dir, parse_html_file)
 
 # =============================================================================
@@ -126,12 +139,15 @@ for landmark in landmarks:
             icon=folium.Icon(color='blue', icon='info-sign')
         ).add_to(m)
 
-# Agregar marcadores para Municipios
+# Agregar marcadores para Municipios, incluyendo datos históricos si están disponibles
 for municipality in municipalities:
     if municipality["lat"] is not None and municipality["lon"] is not None:
+        history_text = municipality.get("history", "")
+        history_excerpt = history_text[:300] + "..." if len(history_text) > 300 else history_text
+        popup_content = f"<b>{municipality['name']}</b><br><b>Historia:</b><br>{history_excerpt}"
         folium.Marker(
             location=[municipality["lat"], municipality["lon"]],
-            popup=folium.Popup(f"<b>{municipality['name']}</b><br>{municipality['content'][:200]}...", max_width=300),
+            popup=folium.Popup(popup_content, max_width=300),
             icon=folium.Icon(color='green', icon='home')
         ).add_to(m)
 
@@ -162,18 +178,19 @@ if st.button("Enviar consulta"):
     if user_input:
         # Configura tu clave API de OpenAI (defínela en st.secrets)
         openai.api_key = st.secrets["OPENAI_API_KEY"]
-        try:
-            response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo",
-                messages=[
-                    {"role": "system", "content": "Eres un asistente experto en recomendaciones turísticas y culturales de Puerto Rico. Ayudas a los usuarios a descubrir lugares de interés y actividades en la isla."},
-                    {"role": "user", "content": user_input}
-                ]
-            )
-            answer = response.choices[0].message.content.strip()
-            st.markdown("**Respuesta:**")
-            st.write(answer)
-        except Exception as e:
-            st.error(f"Error al conectarse con la API de OpenAI: {e}")
+        with st.spinner("Consultando a OpenAI..."):
+            try:
+                response = openai.ChatCompletion.create(
+                    model="gpt-3.5-turbo",
+                    messages=[
+                        {"role": "system", "content": "Eres un asistente experto en recomendaciones turísticas y culturales de Puerto Rico. Ayudas a los usuarios a descubrir lugares de interés y actividades en la isla."},
+                        {"role": "user", "content": user_input}
+                    ]
+                )
+                answer = response.choices[0].message.content.strip()
+                st.markdown("**Respuesta:**")
+                st.write(answer)
+            except Exception as e:
+                st.error(f"Error al conectarse con la API de OpenAI: {e}")
     else:
         st.warning("Por favor, ingresa una consulta.")