import gradio as gr
from PIL import Image
from utils import generate_gradio_images

# Create a Gradio interface
iface = gr.Interface(
    fn=generate_gradio_images,  # Set to None since we won't use a function
    inputs=[
        gr.Textbox("Enter Prompt")
    ],
    outputs=gr.Image()  # Gradio Image component for displaying images
)

# Launch the Gradio interface
iface.launch()