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"),
        gr.Slider("Time Stamps", value=1, step=1, maximum=50),
        gr.Checkbox(label = "Use custom loss",value=False)
    ],
    outputs=[gr.Image(type="pil", height=128, width = 128),
             gr.Image(type="pil", height=128, width = 128), 
             gr.Image(type="pil", height=128, width = 128), 
             gr.Image(type="pil", height=128, width = 128),
             gr.Image(type="pil", height=128, width = 128)]  # Gradio Image component for displaying images
)

# Launch the Gradio interface
iface.launch()