import gradio as gr
from image_model import generate_image

def create_image(prompt):
    return generate_image(prompt)

with gr.Blocks() as demo:
    gr.Markdown(
        """
        # 🎨 AI Image Generator  
        **Powered by Stable Diffusion**  
        Enter a text prompt and generate an image
        """
    )

    prompt = gr.Textbox(
        label="Image Prompt",
        placeholder="e.g. A futuristic city at sunset, digital art",
        lines=2
    )

    generate_btn = gr.Button("Generate Image", variant="primary")

    output_image = gr.Image(label="Generated Image")

    generate_btn.click(
        fn=create_image,
        inputs=prompt,
        outputs=output_image
    )

demo.launch()