Create app.py

app.py

@@ -0,0 +1,154 @@
+import gradio as gr
+import subprocess
+import tempfile
+import os
+from pathlib import Path
+import shutil
+
+def compress_video(input_video, resolution, quality):
+    if input_video is None:
+        return None, "Please upload a video file first."
+    
+    try:
+        # Create temporary files
+        with tempfile.NamedTemporaryFile(delete=False, suffix=Path(input_video).name) as tmp_input:
+            shutil.copy2(input_video, tmp_input.name)
+            input_path = tmp_input.name
+        
+        output_path = input_path.replace(Path(input_video).suffix, "_compressed.mp4")
+        
+        # Map quality to CRF (Constant Rate Factor)
+        crf_map = {"Low": 28, "Medium": 23, "High": 18}
+        crf = crf_map[quality]
+        
+        # Map resolution to scale filter
+        scale_map = {
+            "Original": "",
+            "480p": "scale=-1:480",
+            "720p": "scale=-1:720",
+            "1080p": "scale=-1:1080"
+        }
+        scale = scale_map[resolution]
+        
+        # Build FFmpeg command
+        cmd = ["ffmpeg", "-i", input_path]
+        
+        if scale:
+            cmd.extend(["-vf", scale])
+        
+        cmd.extend([
+            "-c:v", "libx264",
+            "-crf", str(crf),
+            "-preset", "slow",
+            "-c:a", "aac",
+            "-b:a", "128k",
+            "-movflags", "+faststart",
+            "-y",  # Overwrite output file if exists
+            output_path
+        ])
+        
+        # Run FFmpeg
+        result = subprocess.run(cmd, check=True, capture_output=True, text=True)
+        
+        # Calculate file sizes
+        original_size = os.path.getsize(input_path) / (1024 * 1024)
+        compressed_size = os.path.getsize(output_path) / (1024 * 1024)
+        compression_ratio = (1 - compressed_size / original_size) * 100
+        
+        info_text = f"""
+        Compression Complete!
+        
+        Original size: {original_size:.2f} MB
+        Compressed size: {compressed_size:.2f} MB
+        Compression ratio: {compression_ratio:.1f}%
+        
+        Settings used:
+        - Resolution: {resolution}
+        - Quality: {quality}
+        """
+        
+        return output_path, info_text
+        
+    except subprocess.CalledProcessError as e:
+        return None, f"Compression failed: {e.stderr}"
+    except Exception as e:
+        return None, f"An error occurred: {str(e)}"
+    finally:
+        # Clean up input file
+        if 'input_path' in locals() and os.path.exists(input_path):
+            os.unlink(input_path)
+
+# Create the Gradio interface
+with gr.Blocks(title="Video Compressor", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎬 Video Compressor")
+    gr.Markdown("Upload a video file, choose compression settings, and download the compressed version.")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            # Input section
+            input_video = gr.File(
+                label="Upload Video",
+                file_types=["video"],
+                file_count="single"
+            )
+            
+            # Settings section
+            with gr.Group():
+                gr.Markdown("### Compression Settings")
+                
+                resolution = gr.Dropdown(
+                    choices=["Original", "480p", "720p", "1080p"],
+                    value="Original",
+                    label="Resolution",
+                    info="Lower resolution = smaller file size"
+                )
+                
+                quality = gr.Dropdown(
+                    choices=["Low", "Medium", "High"],
+                    value="Medium",
+                    label="Quality",
+                    info="Lower quality = smaller file size"
+                )
+            
+            # Compress button
+            compress_btn = gr.Button("Compress Video", variant="primary", size="lg")
+        
+        with gr.Column(scale=1):
+            # Output section
+            output_video = gr.File(
+                label="Compressed Video",
+                visible=False
+            )
+            
+            output_info = gr.Markdown(
+                label="Compression Info",
+                value="Upload a video and click 'Compress Video' to begin."
+            )
+    
+    # Examples section
+    gr.Examples(
+        examples=[
+            ["sample.mp4", "720p", "Medium"],
+            ["sample.mp4", "480p", "Low"],
+        ],
+        inputs=[input_video, resolution, quality],
+        cache_examples=False,
+        label="Example Settings (you'll need to provide your own video file)"
+    )
+    
+    # Set up the compression function
+    compress_btn.click(
+        fn=compress_video,
+        inputs=[input_video, resolution, quality],
+        outputs=[output_video, output_info]
+    )
+    
+    # Auto-compress when settings change (optional)
+    input_video.change(
+        fn=lambda x: gr.update(visible=False) if x is None else gr.update(),
+        inputs=[input_video],
+        outputs=[output_video]
+    )
+
+if __name__ == "__main__":
+    demo.launch()