removing bounds, adding duplicates

app.py

@@ -113,31 +113,9 @@ def generate_geojson(response):
 
 # Generate static map image
 @spaces.GPU
-def generate_static_map(geojson_data, bounds=None):
+def generate_static_map(geojson_data, invisible=False):
     # Create a static map object with specified dimensions
     m = StaticMap(600, 600)
-    
-    if bounds:
-        #log
-        logging.info(f"Bounds received: {bounds}")
-        # If bounds are provided, set the center of the map
-        center_lat = (bounds[0][0] + bounds[1][0]) / 2
-        center_lng = (bounds[0][1] + bounds[1][1]) / 2
-        #zoom = 10  # Adjust zoom level as needed
-        #m.set_center(center_lat, center_lng) #zoom
-    else:
-        # Default center and zoom level
-        center_lat, center_lng = 0, 0
-        #zoom = 1
-    
-    # Check if there are no features to avoid an empty map
-    if not geojson_data["features"]:
-        # Add a small invisible marker to prevent rendering error
-        #log marker
-        logging.info(f"Adding invisible marker to {bounds[0], bounds[1]}")
-        m.add_marker(CircleMarker(tuple(bounds[0]), 'blue', 10))  # White marker with size 0 #FFFFFF
-        m.add_marker(CircleMarker(tuple(bounds[1]), 'blue', 10))  # White marker with size 0
-        return m.render()
 
     # Process each feature in the GeoJSON
     for feature in geojson_data["features"]:
@@ -145,13 +123,13 @@ def generate_static_map(geojson_data, bounds=None):
         coords = feature["geometry"]["coordinates"]
 
         if geom_type == "Point":
-            m.add_marker(CircleMarker((coords[0], coords[1]), 'blue', 10))
+            m.add_marker(CircleMarker((coords[0], coords[1]), '#1C00ff00' if invisible == True else 'blue', 10))
         elif geom_type in ["MultiPoint", "LineString"]:
             for coord in coords:
-                m.add_marker(CircleMarker((coord[0], coord[1]), 'blue', 10))
+                m.add_marker(CircleMarker((coord[0], coord[1]), '#1C00ff00' if invisible == True else 'blue', 10))
         elif geom_type in ["Polygon", "MultiPolygon"]:
             for polygon in coords:
-                m.add_polygon(Polygon([(c[0], c[1]) for c in polygon], 'blue', 3))
+                m.add_polygon(Polygon([(c[0], c[1]) for c in polygon], '#1C00ff00' if invisible == True else 'blue', 3))
 
     return m.render() #zoom=10
 
@@ -188,27 +166,6 @@ def generate_satellite_image(init_image, mask_image, prompt):
         )
     return result.images[0]
 
-def get_bounds(geojson):
-    coordinates = []
-    for feature in geojson["features"]:
-        geom_type = feature["geometry"]["type"]
-        coords = feature["geometry"]["coordinates"]
-        if geom_type == "Point":
-            coordinates.append(coords)
-        elif geom_type in ["MultiPoint", "LineString"]:
-            coordinates.extend(coords)
-        elif geom_type in ["MultiLineString", "Polygon"]:
-            for part in coords:
-                coordinates.extend(part)
-        elif geom_type == "MultiPolygon":
-            for polygon in coords:
-                for part in polygon:
-                    coordinates.extend(part)
-    lats = [coord[1] for coord in coordinates]
-    lngs = [coord[0] for coord in coordinates]
-    logging.info(f"Bounds: {min(lats), min(lngs)}, {max(lats), max(lngs)}")
-    return [[min(lats), min(lngs)], [max(lats), max(lngs)]]
-
 # Gradio UI
 @spaces.GPU
 def handle_query(query):
@@ -219,13 +176,7 @@ def handle_query(query):
     # Generate the main map image
     map_image = generate_static_map(geojson_data)
 
-    # Generate the empty map using the same bounds
-    bounds = get_bounds(geojson_data)
-    empty_geojson = {
-        "type": "FeatureCollection",
-        "features": []  # Empty map contains no features
-    }
-    empty_map_image = generate_static_map(empty_geojson, bounds=bounds)  # Empty map with the same bounds
+    empty_map_image = generate_static_map(geojson_data, invisible=True)  # Empty map with the same bounds
 
     # Create the mask
     difference = np.abs(np.array(map_image.convert("RGB")) - np.array(empty_map_image.convert("RGB")))