import gradio as gr
from distort import apply_wave_effect

def distort_image(image):
    return apply_wave_effect(image)

demo = gr.Interface(
    fn=distort_image,
    inputs=gr.Image(type="pil"),
    outputs="image",
    title="Paper Wave Distortion",
    description="Upload an image and apply wave-like paper distortion."
)

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