Upload 2 files

app.py

@@ -0,0 +1,84 @@
+import gradio as gr
+import ffmpeg
+import os
+import tempfile
+
+def compress_video(input_path, target_bitrate, target_width):
+    # Probe the original video to get resolution
+    probe = ffmpeg.probe(input_path)
+    video_streams = [stream for stream in probe['streams'] if stream['codec_type'] == 'video']
+    if not video_streams:
+        raise ValueError("No video stream found in the uploaded file.")
+
+    orig_width = int(video_streams[0]['width'])
+    orig_height = int(video_streams[0]['height'])
+
+    # Calculate new height to maintain aspect ratio
+    new_width = int(target_width)
+    new_height = int(orig_height * new_width / orig_width)
+    # Ensure height is even (ffmpeg requirement)
+    if new_height % 2 != 0:
+        new_height += 1
+
+    # Create a temporary output file
+    out_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+    output_path = out_file.name
+    out_file.close()
+
+    # Run ffmpeg to compress and resize
+    (
+        ffmpeg
+        .input(input_path)
+        .output(
+            output_path,
+            video_bitrate=f'{target_bitrate}k',
+            vf=f'scale={new_width}:{new_height}',
+            audio_bitrate='128k',
+            preset='medium'
+        )
+        .run(overwrite_output=True)
+    )
+
+    return output_path
+
+
+def process_video(input_path, target_bitrate, target_width):
+    # Compress the video and return the path for download
+    compressed_path = compress_video(input_path, target_bitrate, target_width)
+    return compressed_path
+
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Video Compressor App")
+
+        with gr.Row():
+            input_video = gr.Video(label="Upload Video")
+            target_bitrate = gr.Slider(
+                minimum=100,
+                maximum=5000,
+                step=100,
+                value=1000,
+                label="Target Video Bitrate (kbps)"
+            )
+            target_width = gr.Slider(
+                minimum=240,
+                maximum=1920,
+                step=16,
+                value=640,
+                label="Target Width (px)"
+            )
+
+        compress_button = gr.Button("Compress")
+        output_video = gr.File(label="Download Compressed Video")
+
+        compress_button.click(
+            fn=process_video,
+            inputs=[input_video, target_bitrate, target_width],
+            outputs=output_video
+        )
+
+    demo.launch()
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+ffmpeg-python