Create app.py

app.py

@@ -0,0 +1,65 @@
+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
+                )