Update app.py

app.py

@@ -86,8 +86,6 @@ def create_plot(filename, extent, *gdfs):
 
     plt.savefig(filename)
 
-
-
 def ss_intersect(geojson1, ss_geoselect, multiplier_factor, default_building_height): 
     # Read the GeoJSON files
     input_gdf = gpd.read_file(geojson1.name)
@@ -134,11 +132,10 @@ def ss_intersect(geojson1, ss_geoselect, multiplier_factor, default_building_hei
     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)
-
+    fig = create_plot('output_image.png', extent, input_gdf, intersected_sites_gdf, buffer_union_gdf)
             
     # Return the image
-    return "output_image.png", "building_buffers.geojson", "buffer_union.geojson", intersection_desc
+    return fig, "building_buffers.geojson", "buffer_union.geojson", intersection_desc
 
 
 iface = gr.Interface(
@@ -150,7 +147,7 @@ iface = gr.Interface(
         gr.inputs.Number(default=200, label="Default Building Height"), #Can I make this optional?
     ], 
     outputs=[
-        gr.outputs.Image(label="Result Image"),
+        gr.outputs.Image(type="pil", 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"),