import os
import prodiapy
import gradio as gr

api_key = os.getenv("PRODIA_API_KEY")
prodia = prodiapy.Prodia(api_key=api_key)

def generate_image(prompt):
    job = prodia.sd.generate(prompt=prompt)
    result = prodia.wait(job)
    return result.image_url

demo = gr.Interface(fn=generate_image, inputs="text", outputs="image", title="Simple Stable Diffusion", description="Generate images using a prompt.")
demo.launch()