Function addition, requirements update

app.py

@@ -71,6 +71,28 @@ def generate_geojson(response):
         }]
     }
 
+
+# Function to compute bounds from GeoJSON
+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)]]
+
 # Generate map image
 def save_map_image(geojson_data):
     m = Map()
@@ -90,6 +112,16 @@ pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
 )
 pipeline.enable_model_cpu_offload()
 
+def make_inpaint_condition(init_image, mask_image):
+    init_image = np.array(init_image.convert("RGB")).astype(np.float32) / 255.0
+    mask_image = np.array(mask_image.convert("L")).astype(np.float32) / 255.0
+
+    assert init_image.shape[0:1] == mask_image.shape[0:1], "image and image_mask must have the same image size"
+    init_image[mask_image > 0.5] = -1.0  # set as masked pixel
+    init_image = np.expand_dims(init_image, 0).transpose(0, 3, 1, 2)
+    init_image = torch.from_numpy(init_image)
+    return init_image
+
 def generate_satellite_image(init_image_path, mask_image_path, prompt):
     init_image = Image.open(init_image_path)
     mask_image = Image.open(mask_image_path)

requirements.txt

@@ -1,11 +1,10 @@
-torch
 openai       # For interacting with OpenAI API
 gradio       # For creating the Gradio UI
 gradio-folium   # For embedding Folium maps into Gradio
 folium        # For creating maps
 geopy       # For fetching geolocation data      # For PyTorch (used by Diffusers and ControlNet)
-diffusers     # For the Stable Diffusion inpainting pipeline
-opencv-python-headless # For image processing with OpenCV
-Pillow # For working with images
 numpy  # For numerical operations
-safetensors
+diffusers
+transformers
+torchvision
+torch