Added mask to outputs

app.py

@@ -185,7 +185,7 @@ def handle_query(query):
     # Generate satellite image
     satellite_image = generate_satellite_image(map_image, mask_image, response['output']['feature_representation']['properties']['description'])
 
-    return map_image, satellite_image, response
+    return map_image, satellite_image, mask, response
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -195,6 +195,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         map_output = gr.Image(label="Map Visualization")
         satellite_output = gr.Image(label="Generated Satellite Image")
+        mask_output = gr.Image(label="Mask")
         image_prompt = gr.Textbox(label="Image Prompt Used")
     submit_btn.click(handle_query, inputs=[query_input], outputs=[map_output, satellite_output, image_prompt])