Spaces:
Sleeping
Sleeping
File size: 2,239 Bytes
7d582a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import streamlit as st
import ffmpeg
import math
import os
from pathlib import Path
def split_video(input_file_path, output_dir, duration=20):
"""
Splits the input video file into segments of specified duration.
Parameters:
- input_file_path (str): Path to the input video file.
- output_dir (str): Directory to save output segments.
- duration (int): Duration of each segment in seconds.
Returns:
- segment_files (list): List of file paths for each video segment.
"""
probe = ffmpeg.probe(input_file_path)
total_duration = float(probe['format']['duration'])
num_segments = math.ceil(total_duration / duration)
segment_files = []
for i in range(num_segments):
start_time = i * duration
output_file = os.path.join(output_dir, f"segment_{i + 1}.mp4")
ffmpeg.input(input_file_path, ss=start_time, t=duration).output(output_file).run()
segment_files.append(output_file)
return segment_files
# Streamlit UI
st.title("Video Splitter")
st.write("Upload a video file, specify the segment duration, and download the split video segments.")
# File uploader
uploaded_file = st.file_uploader("Choose a video file", type=["mp4", "mov", "avi"])
# Duration input
duration = st.number_input("Segment duration (seconds)", min_value=1, max_value=300, value=20)
if uploaded_file is not None:
# Save the uploaded file to a temporary path
input_file_path = Path("uploaded_video.mp4")
with open(input_file_path, "wb") as f:
f.write(uploaded_file.getbuffer())
# Output directory for segments
output_dir = Path("segments")
output_dir.mkdir(exist_ok=True)
if st.button("Split Video"):
# Split video into segments
segment_files = split_video(str(input_file_path), str(output_dir), duration)
st.success(f"Video split into {len(segment_files)} segments.")
# Provide download links for each segment
for segment in segment_files:
with open(segment, "rb") as file:
btn = st.download_button(
label=f"Download {Path(segment).name}",
data=file,
file_name=Path(segment).name
)
|