import gradio as gr
import moviepy.editor as mp
from math import ceil

def split_video(input_video, duration):
    video_clip = mp.VideoFileClip(input_video)
    video_duration = video_clip.duration
    num_segments = ceil(video_duration / duration)

    segments = []
    for i in range(num_segments):
        start_time = i * duration
        end_time = min((i + 1) * duration, video_duration)
        output_video = f"segment_{i+1}.mp4"

        stream = mp.VideoFileClip(input_video)
        stream = stream.subclip(start_time, end_time)
        stream.write_videofile(output_video, codec='libx264', threads=4)

        segments.append(output_video)

    return segments

interface = gr.Interface(
    split_video,
    inputs=[
        gr.File(label="Upload Video"),
        gr.Number(label="Segment Duration (seconds)"),
    ],
    outputs=gr.Files(label="Video Segments"),
    title="Video Splitter",
    description="Split a video into equal-duration segments.",
)

interface.launch()