geocode update

app.py

@@ -131,13 +131,24 @@ def generate_geojson(response):
     properties = response['output']['feature_representation']['properties']
 
     coordinates = []
-    for city in city_names:
-        coord = get_geo_coordinates(city)
-        if coord:
-            coordinates.append(coord)
 
-    if feature_type == "Polygon" and len(coordinates) > 2:
-        coordinates.append(coordinates[0])  # Close the polygon
+    # Fetch coordinates for cities
+    for city in city_names:
+        try:
+            coord = get_geo_coordinates(city)  # Function to fetch city coordinates
+            if coord:
+                coordinates.append(coord)
+            else:
+                logger.warning(f"Coordinates not found for city: {city}")
+        except Exception as e:
+            logger.error(f"Error fetching coordinates for {city}: {e}")
+
+    if feature_type == "Polygon":
+        if len(coordinates) < 3:
+            raise ValueError("Polygon requires at least 3 coordinates.")
+        # Close the polygon by appending the first point at the end
+        coordinates.append(coordinates[0])
+        coordinates = [coordinates]  # Nest coordinates for Polygon
 
     # Create the GeoJSON object
     geojson_data = {
@@ -148,10 +159,10 @@ def generate_geojson(response):
                 "properties": properties,
                 "geometry": {
                     "type": feature_type,
-                    "coordinates": coordinates if feature_type != "Polygon" else [coordinates],
-                }
+                    "coordinates": coordinates,
+                },
             }
-        ]
+        ],
     }
 
     # Validate the GeoJSON