Added prompt to output

app.py

@@ -178,7 +178,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
+    return map_image, satellite_image, response
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -188,8 +188,8 @@ with gr.Blocks() as demo:
     with gr.Row():
         map_output = gr.Image(label="Map Visualization")
         satellite_output = gr.Image(label="Generated Satellite Image")
-
-    submit_btn.click(handle_query, inputs=[query_input], outputs=[map_output, satellite_output])
+        image_prompt = gr.Textbox(label="Image Prompt Used")
+    submit_btn.click(handle_query, inputs=[query_input], outputs=[map_output, satellite_output, image_prompt])
 
 if __name__ == "__main__":
     demo.launch()