Simple polygon conversion

app.py

@@ -10,6 +10,7 @@ from staticmap import StaticMap, CircleMarker, Polygon
 from diffusers import ControlNetModel, StableDiffusionControlNetInpaintPipeline
 import spaces
 import logging
+import math
 from typing import List, Union
 
 # Set up logging
@@ -151,6 +152,36 @@ def generate_geojson(response):
 
     return geojson_data
 
+# Sort coordinates for a simple polygon (Reduce intersection points)
+def sort_coordinates_for_simple_polygon(geojson):
+    # Extract coordinates from the GeoJSON
+    coordinates = geojson['features'][0]['geometry']['coordinates'][0]
+
+    # Remove the last point if it duplicates the first (GeoJSON convention for polygons)
+    if coordinates[0] == coordinates[-1]:
+        coordinates = coordinates[:-1]
+
+    # Calculate the centroid of the points
+    centroid_x = sum(point[0] for point in coordinates) / len(coordinates)
+    centroid_y = sum(point[1] for point in coordinates) / len(coordinates)
+
+    # Define a function to calculate the angle relative to the centroid
+    def angle_from_centroid(point):
+        dx = point[0] - centroid_x
+        dy = point[1] - centroid_y
+        return math.atan2(dy, dx)
+
+    # Sort points by their angle from the centroid
+    sorted_coordinates = sorted(coordinates, key=angle_from_centroid)
+
+    # Close the polygon by appending the first point to the end
+    sorted_coordinates.append(sorted_coordinates[0])
+
+    # Update the GeoJSON with sorted coordinates
+    geojson['features'][0]['geometry']['coordinates'][0] = sorted_coordinates
+
+    return geojson
+
 # Generate static map image
 @spaces.GPU
 def generate_static_map(geojson_data, invisible=False):
@@ -167,6 +198,10 @@ def generate_static_map(geojson_data, invisible=False):
             for coord in coords:
                 m.add_marker(CircleMarker((coord[0], coord[1]), '#1C00ff00' if invisible else '#42445A85', 100))
         elif geom_type in ["Polygon", "MultiPolygon"]:
+            # Sort coordinates for simple polygons
+            if geom_type == "Polygon":
+                feature["geometry"]["coordinates"] = sort_coordinates_for_simple_polygon(geojson_data)
+                coords = feature["geometry"]["coordinates"]
             for polygon in coords:
                 m.add_polygon(Polygon([(c[0], c[1]) for c in polygon], '#1C00ff00' if invisible else '#42445A85', 3))