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.components.Textbox(label="Enter a text prompt here"),
    outputs=gr.components.Image(label="Output Image"),
    title="Map Diffuser",
    description="""
        🌏 Generates four images from a given text prompt.
        Examples:
        - "Satellite image of amsterdam with industrial area and highways"
        - Watercolor style map of Amsterdam with residential area and highways
        - "Toner style map of Amsterdam with residential area and highways"
        - "Satellite image with forests and residential, no water"
    """
    # examples=[["Satellite image of amsterdam with industrial area and highways"], [
    #     "Satellite image with forests and residential, no water"]]
)

iface.launch()