import gradio as gr
from utils import generate_sticker

def predict(image, prompt):
    result_img = generate_sticker(image, prompt)
    return result_img

with gr.Blocks() as demo:
    gr.Markdown("# 🦄 AI Sticker Generator (Stable Diffusion + IP-Adapter)")
    with gr.Row():
        image_input = gr.Image(type="pil", label="Upload your photo")
        prompt_input = gr.Textbox(
            label="Prompt (style or mood for emoji)",
            value="cartoon emoji, white outline, clean background",
        )
    output_image = gr.Image(label="Sticker Output")
    run_btn = gr.Button("Generate Sticker")
    run_btn.click(predict, inputs=[image_input, prompt_input], outputs=output_image)

if __name__ == "__main__":
    demo.launch()