import gradio as gr
import imageio
import numpy as np

def image_to_video(image):
    video_filename = "output.mp4"
    writer = imageio.get_writer(video_filename, fps=10)

    # Generate a simple video with repeated frames
    for _ in range(30):  # 30 frames (1 second at 30fps)
        writer.append_data(image)
    writer.close()

    return video_filename

demo = gr.Interface(
    fn=image_to_video,
    inputs=gr.Image(type="numpy", label="Upload Image"),
    outputs=gr.Video(),
    title="Image to Video Converter",
    description="Upload an image to generate a short looping video."
)

demo.launch()