import gradio as gr
from inference_code import generate_images

def generate_image_predictions(prompt):
    images = generate_images(prompt)
    return images

iface = gr.Interface(
    fn=generate_image_predictions,
    inputs=gr.inputs.Textbox("Enter a text prompt here"),
    outputs=[gr.outputs.Image() for i in range(4)],
    title="Map Diffuser",
    description="Generates four images from a given text prompt.",
    examples=[["Satellite image of amsterdam with industrial area and highways"], ["Satellite image with forests and residential, no water"], ["A person playing guitar on a stage"]]
)

iface.launch()