import gradio as gr
from diffusers import DiffusionPipeline

# Load the diffusion pipeline
pipe = DiffusionPipeline.from_pretrained("ostris/OpenFLUX.1")

def generate_image(prompt):
    image = pipe(prompt).images[0]
    return image

# Set up the Gradio interface
iface = gr.Interface(
    fn=generate_image,
    inputs=gr.Textbox(label="Prompt", placeholder="Enter a prompt..."),
    outputs=gr.Image(label="Generated Image"),
    title="Diffusion Model Image Generator",
    description="Generate images using a diffusion model based on your text prompts."
)

# Launch the app
if __name__ == "__main__":
    iface.launch()