Add points KML export to FN4B parser

apps/fnb_parser.py

@@ -7,10 +7,12 @@ upload one or many Word documents and instantly visualise / download the results
 import io
 import os
 import re
+from datetime import datetime
 from typing import List
 
 import pandas as pd
 import plotly.express as px
+import simplekml
 import streamlit as st
 from docx import Document
 from streamlit.commands.execution_control import rerun
@@ -237,6 +239,61 @@ def process_files_to_dataframe(uploaded_files) -> pd.DataFrame:
     return df
 
 
+def generate_points_kml(df: pd.DataFrame) -> io.BytesIO | None:
+    """Build a point-only KML file from extracted site coordinates."""
+    required_columns = {"X_decimal", "Y_decimal"}
+    if not required_columns.issubset(df.columns):
+        return None
+
+    metadata_columns = [
+        "Code",
+        "Site Name",
+        "Localité",
+        "Adresse",
+        "File",
+        "X_decimal",
+        "Y_decimal",
+    ]
+    available_columns = [col for col in metadata_columns if col in df.columns]
+    points_df = df[available_columns].dropna(subset=["X_decimal", "Y_decimal"])
+    if points_df.empty:
+        return None
+
+    dedupe_keys = [col for col in ["Code", "X_decimal", "Y_decimal"] if col in points_df.columns]
+    if dedupe_keys:
+        points_df = points_df.drop_duplicates(subset=dedupe_keys)
+
+    kml = simplekml.Kml()
+    for _, row in points_df.iterrows():
+        point_name = row.get("Code") or row.get("Site Name") or "Site"
+        point = kml.newpoint(
+            name=str(point_name),
+            coords=[(float(row["Y_decimal"]), float(row["X_decimal"]))],
+        )
+        point.style.iconstyle.icon.href = (
+            "http://maps.google.com/mapfiles/kml/pushpin/ylw-pushpin.png"
+        )
+        point.style.iconstyle.scale = 1.1
+
+        description_parts = []
+        for label, column in [
+            ("Code", "Code"),
+            ("Site Name", "Site Name"),
+            ("Localité", "Localité"),
+            ("Adresse", "Adresse"),
+            ("Source File", "File"),
+        ]:
+            value = row.get(column)
+            if pd.notna(value) and str(value).strip():
+                description_parts.append(f"{label}: {value}")
+        point.description = "<br>".join(description_parts)
+
+    kml_data = io.BytesIO()
+    kml_data.write(kml.kml().encode("utf-8"))
+    kml_data.seek(0)
+    return kml_data
+
+
 ###############################################################################
 # -----------------------------  Streamlit UI  ------------------------------ #
 ###############################################################################
@@ -287,14 +344,30 @@ def main() -> None:
         buffer = io.BytesIO()
         with pd.ExcelWriter(buffer, engine="xlsxwriter") as writer:
             df.to_excel(writer, index=False, sheet_name="Extract")
-        st.download_button(
-            label="Download Excel",
-            data=buffer.getvalue(),
-            file_name="extracted_fnb.xlsx",
-            mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-            on_click="ignore",
-            type="primary",
-        )
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        excel_col, kml_col = st.columns(2)
+        with excel_col:
+            st.download_button(
+                label="Download Excel",
+                data=buffer.getvalue(),
+                file_name=f"extracted_fnb_{timestamp}.xlsx",
+                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+                on_click="ignore",
+                type="primary",
+                use_container_width=True,
+            )
+
+        points_kml_data = generate_points_kml(df)
+        if points_kml_data is not None:
+            with kml_col:
+                st.download_button(
+                    label="Download Points KML",
+                    data=points_kml_data.getvalue(),
+                    file_name=f"fnb_points_{timestamp}.kml",
+                    mime="application/vnd.google-earth.kml+xml",
+                    on_click="ignore",
+                    use_container_width=True,
+                )
 
         st.markdown("---")