Update app.py

app.py

@@ -409,6 +409,8 @@ elif shape_type.lower() == "polygon":
 #                     m.to_streamlit()
 #                     st.session_state.map_data = m
 
+# ... (Previous code remains unchanged until the file upload section) ...
+
 # Ask user to upload a file based on shape type
 file_upload = st.file_uploader(f"Upload your {shape_type} data (CSV, GeoJSON, KML)", type=["csv", "geojson", "kml"])
 
@@ -428,10 +430,14 @@ if file_upload is not None:
                 points = []
                 for placemark in root.findall('.//kml:Placemark', ns):
                     name = placemark.findtext('kml:name', default=f"Point_{len(points)}", namespaces=ns)
-                    coords = placemark.find('.//kml:Point/kml:coordinates', ns)
-                    if coords is not None:
-                        lon, lat, *_ = coords.text.strip().split(',')
-                        points.append({'name': name, 'geometry': f"POINT ({lon} {lat})"})
+                    coords_elem = placemark.find('.//kml:Point/kml:coordinates', ns)
+                    if coords_elem is not None:
+                        # Split coordinates and handle potential whitespace
+                        coords_text = coords_elem.text.strip()
+                        coords = [c.strip() for c in coords_text.split(',')]
+                        if len(coords) >= 2:  # Ensure at least lon, lat
+                            lon, lat = float(coords[0]), float(coords[1])
+                            points.append({'name': name, 'geometry': f"POINT ({lon} {lat})"})
                 if not points:
                     st.error("No valid Point data found in the KML file.")
                     locations_df = pd.DataFrame()
@@ -485,11 +491,14 @@ if file_upload is not None:
                 polygons = []
                 for placemark in root.findall('.//kml:Placemark', ns):
                     name = placemark.findtext('kml:name', default=f"Polygon_{len(polygons)}", namespaces=ns)
-                    coords = placemark.find('.//kml:Polygon//kml:coordinates', ns)
-                    if coords is not None:
-                        coord_list = coords.text.strip().split()
-                        coords_str = " ".join([f"{lon} {lat}" for lon, lat, *_ in [c.split(',') for c in coord_list]])
-                        polygons.append({'name': name, 'geometry': f"POLYGON (({coords_str}))"})
+                    coords_elem = placemark.find('.//kml:Polygon//kml:coordinates', ns)
+                    if coords_elem is not None:
+                        # Process coordinates string
+                        coords_text = coords_elem.text.strip()
+                        coord_pairs = [pair.split(',')[:2] for pair in coords_text.split() if pair]  # Take only lon, lat, ignore altitude
+                        if len(coord_pairs) >= 4:  # Minimum 4 points for a closed polygon
+                            coords_str = " ".join([f"{float(lon)} {float(lat)}" for lon, lat in coord_pairs])
+                            polygons.append({'name': name, 'geometry': f"POLYGON (({coords_str}))"})
                 if not polygons:
                     st.error("No valid Polygon data found in the KML file.")
                     locations_df = pd.DataFrame()