Difference with actual map

app.py

@@ -164,6 +164,54 @@ 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]
+    return [[min(lats), min(lngs)], [max(lats), max(lngs)]]
+
+@spaces.GPU
+def generate_static_map(geojson_data, bounds=None):
+    # Create a static map object with specified dimensions
+    m = StaticMap(600, 600)
+    
+    if bounds:
+        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)
+    
+    # Process each feature in the GeoJSON
+    for feature in geojson_data["features"]:
+        geom_type = feature["geometry"]["type"]
+        coords = feature["geometry"]["coordinates"]
+
+        if geom_type == "Point":
+            m.add_marker(CircleMarker((coords[0], coords[1]), 'blue', 10))
+        elif geom_type in ["MultiPoint", "LineString"]:
+            for coord in coords:
+                m.add_marker(CircleMarker((coord[0], coord[1]), '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))
+
+    return m.render(zoom=10)
+
+
 # Gradio UI
 @spaces.GPU
 def handle_query(query):
@@ -171,22 +219,33 @@ def handle_query(query):
     response = process_openai_response(query)
     geojson_data = generate_geojson(response)
 
-    # Generate map image
+    # Generate the main map image
     map_image = generate_static_map(geojson_data)
 
-    # Generate mask for ControlNet
-    empty_map = Image.new("RGB", map_image.size, "white")
-    difference = np.array(map_image) - np.array(empty_map)
-    mask = np.any(difference != 0, axis=-1).astype(np.uint8) * 255
+    # 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)  # 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")))
+    threshold = 10  # Tolerance for difference
+    mask = (np.sum(difference, axis=-1) > threshold).astype(np.uint8) * 255
 
-    # Convert mask to PIL Image
-    mask_image = Image.fromarray(mask)
+    # Convert the mask to a PIL image
+    mask_image = Image.fromarray(mask, mode="L")
 
-    # Generate satellite image
-    satellite_image = generate_satellite_image(map_image, mask_image, response['output']['feature_representation']['properties']['description'])
+    # Generate the satellite image
+    satellite_image = generate_satellite_image(
+        map_image, mask_image, response['output']['feature_representation']['properties']['description']
+    )
 
     return map_image, satellite_image, mask_image, response
 
+
 # Gradio interface
 with gr.Blocks() as demo:
     with gr.Row():