import gradio as gr
import os
from prodia import createProdia

def generate_image(prompt):
    api_key = os.getenv("YOUR_PRODIA_KEY")
    if not api_key:
        raise ValueError("API key not found in environment variables")
    
    prodia = createProdia({"apiKey": api_key})
    
    job = prodia.generate({"prompt": prompt})
    result = prodia.wait(job)
    return result["imageUrl"]

with gr.Blocks() as demo:
    gr.Markdown("## Prodia Stable Diffusion API")
    prompt = gr.Textbox(placeholder="Enter prompt here...")
    output = gr.Image()
    btn = gr.Button("Generate")
    btn.click(fn=generate_image, inputs=prompt, outputs=output)

demo.launch()