Update app.py

app.py

@@ -2,6 +2,11 @@ import os
 import subprocess
 import gradio as gr
 
+def check_video_size(video_path, max_size_mb=1000):  # Allow up to 1 GB
+    size_mb = os.path.getsize(video_path) / (1024 * 1024)
+    if size_mb > max_size_mb:
+        raise gr.Error(f"Video is too large (max {max_size_mb} MB allowed).")
+
 def split_video_into_chunks(video_path, chunk_duration=6, output_dir="video_chunks"):
     # Create output directory
     os.makedirs(output_dir, exist_ok=True)
@@ -27,8 +32,8 @@ def split_video_into_chunks(video_path, chunk_duration=6, output_dir="video_chun
             "ffmpeg", "-i", video_path,
             "-ss", str(start_time),
             "-t", str(chunk_duration),
-            "-c:v", "libx264",  # Re-encode video stream
-            "-c:a", "aac",      # Re-encode audio stream
+            "-c:v", "libx264", "-preset", "fast",  # Faster encoding
+            "-c:a", "aac",
             output_file
         ]
         subprocess.run(command, check=True)
@@ -37,24 +42,30 @@ def split_video_into_chunks(video_path, chunk_duration=6, output_dir="video_chun
 
     return chunk_files
 
-def process_video(video):
-    # Split the video into 6-second chunks
-    chunk_files = split_video_into_chunks(video)
+def process_video(video, chunk_duration):
+    # Check video size (up to 1 GB allowed)
+    check_video_size(video, max_size_mb=1000)
+
+    # Split the video into chunks with the specified duration
+    chunk_files = split_video_into_chunks(video, chunk_duration=chunk_duration)
 
     # Return the list of chunk files for download
     return chunk_files
 
 # Gradio Interface
-def gradio_interface(video):
-    return process_video(video)
+def gradio_interface(video, chunk_duration):
+    return process_video(video, chunk_duration)
 
 # Define Gradio app
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.Video(label="Upload Video"),
+    inputs=[
+        gr.Video(label="Upload Video"),
+        gr.Slider(minimum=1, maximum=60, value=6, step=1, label="Chunk Duration (seconds)")
+    ],
     outputs=gr.Files(label="Download Chunks"),
-    title="Video Splitter into 6-Second Chunks",
-    description="Upload a video, and the app will split it into 6-second chunks for download."
+    title="Video Splitter with Custom Chunk Duration",
+    description="Upload a video (max 1 GB) and choose the chunk duration in seconds. The app will split the video into chunks for download."
 )
 
 # Launch the app