apply only in polygon

app.py

@@ -180,9 +180,6 @@ def sort_coordinates_for_simple_polygon(geojson):
     # Update the GeoJSON with sorted coordinates
     geojson['features'][0]['geometry']['coordinates'][0] = sorted_coordinates
 
-    #log geojson
-    logger.info(f"Sorted GeoJSON: {geojson}")
-
     return geojson
 
 # Generate static map image
@@ -201,14 +198,7 @@ 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":
-                oredered_geojson = sort_coordinates_for_simple_polygon(geojson_data)
-                feature["geometry"]["coordinates"] = oredered_geojson["features"][0]["geometry"]["coordinates"]
-                coords = feature["geometry"]["coordinates"]
-                logger.info(f"coords: {coords}")
             for polygon in coords:
-                logger.info(f"polygon: {polygon}")
                 m.add_polygon(Polygon([(c[0], c[1]) for c in polygon], '#1C00ff00' if invisible else '#42445A85', 3))
 
     return m.render()
@@ -251,7 +241,11 @@ def handle_query(query):
     response = process_openai_response(query)
     geojson_data = generate_geojson(response)
 
-    map_image = generate_static_map(geojson_data)
+    if geojson_data["features"][0]["geometry"]["type"] == 'Polygon':
+        geojson_data_coords = sort_coordinates_for_simple_polygon(geojson_data)
+        map_image = generate_static_map(geojson_data_coords)
+    else:
+        map_image = generate_static_map(geojson_data)
     empty_map_image = generate_static_map(geojson_data, invisible=True)
 
     difference = np.abs(np.array(map_image.convert("RGB")) - np.array(empty_map_image.convert("RGB")))