Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import geopandas as gpd
 import pandas as pd
 import os
+import matplotlib.pyplot as plt
 from shapely.geometry import shape
 from shapely.ops import unary_union
 #from datasets import load_dataset
@@ -64,6 +65,28 @@ def process_buildings(input_gdf, sensitive_sites_gdf, default_building_height_m,
     
     return buffers, intersected_sites, intersection_desc
 
+def get_max_extent(*gdfs):
+    minx = min(gdf.total_bounds[0] for gdf in gdfs)
+    miny = min(gdf.total_bounds[1] for gdf in gdfs)
+    maxx = max(gdf.total_bounds[2] for gdf in gdfs)
+    maxy = max(gdf.total_bounds[3] for gdf in gdfs)
+    
+    return minx, miny, maxx, maxy
+
+def create_plot(filename, extent, *gdfs):
+    fig, ax = plt.subplots(1, 1)
+    
+    colors = ['blue', 'red', 'green', 'purple', 'orange', 'yellow'] # Extend this list as needed
+
+    for idx, gdf in enumerate(gdfs):
+        gdf.plot(ax=ax, color=colors[idx % len(colors)]) # Cycle through colors
+
+    ax.set_xlim(extent[0], extent[2])
+    ax.set_ylim(extent[1], extent[3])
+
+    plt.savefig(filename)
+
+
 
 def ss_intersect(geojson1, ss_geoselect, multiplier_factor, default_building_height): 
     # Read the GeoJSON files
@@ -106,9 +129,16 @@ def ss_intersect(geojson1, ss_geoselect, multiplier_factor, default_building_hei
         # Save the union GeoDataFrame as a GeoJSON file
         buffer_union_gdf.to_file("buffer_union.geojson", driver='GeoJSON')
 
+
+    # Calculate the maximum extent
+    extent = get_max_extent(input_gdf, buffers_gdf)
+
+    # Create and save the plot
+    create_plot('output_image.png', extent, input_gdf, intersected_sites_gdf, buffer_union_gdf)
+
             
     # Return the image
-    return "building_buffers.geojson", "buffer_union.geojson", intersection_desc
+    return "output_image.png", "building_buffers.geojson", "buffer_union.geojson", intersection_desc
 
 
 iface = gr.Interface(
@@ -116,11 +146,11 @@ iface = gr.Interface(
     inputs=[
         gr.inputs.File(label="Building Footprints GeoJSON"),
         gr.Radio(["Parks Properties", "Park Zones"], label="Which Sensitive Sites?", info="From NYC DPR", type="index"),
-        #gr.inputs.File(label="Sensitive Sites GeoJSON"), #Replaced by radio button above
         gr.inputs.Slider(minimum=0.0, maximum=10.0, default=4.3, label="Building Height Multiplier"),
         gr.inputs.Number(default=200, label="Default Building Height"), #Can I make this optional?
     ], 
     outputs=[
+        gr.outputs.Image(label="Result Image"),
         gr.outputs.File(label="Building Buffers"),
         gr.outputs.File(label="Union of Building Buffers"),
         gr.outputs.Textbox(label="Building and Sensitive Site Vicinities"),