Spaces:
Paused
Paused
| from flask import Flask, render_template, request, send_file | |
| import os | |
| import tempfile | |
| import subprocess | |
| import zipfile | |
| from waitress import serve | |
| # Initialize the Flask app, tell it to look in the current directory for templates | |
| app = Flask(__name__, template_folder='.') | |
| def home(): | |
| return render_template('html.html') # Flask will look in the current directory | |
| def process_video(): | |
| video_file = request.files['video'] | |
| action = request.form['action'] | |
| copy_streams = 'copy_streams' in request.form # Check if the user selected "copy streams" | |
| # Save the uploaded video to a temporary file | |
| temp_file = tempfile.NamedTemporaryFile(delete=False) | |
| video_file.save(temp_file.name) | |
| output_path = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name # Default output format | |
| # Process video based on action | |
| if action == 'Convert Format': | |
| selected_format = request.form['format'] | |
| output_path = tempfile.NamedTemporaryFile(delete=False, suffix=f'.{selected_format}').name # Adjust output format based on user selection | |
| if copy_streams: | |
| # If "copy streams" is selected, try to copy streams without re-encoding | |
| result = subprocess.run(['ffmpeg', '-i', temp_file.name, '-c:v', 'copy', '-c:a', 'copy', '-y', output_path], capture_output=True) | |
| if result.returncode != 0: | |
| # If copying the streams failed, re-encode the video | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-y', output_path]) | |
| else: | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-y', output_path]) | |
| elif action == 'Trim Video': | |
| start_time = request.form['start_time'] | |
| duration = request.form['duration'] | |
| output_path = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name # Example output format | |
| if copy_streams: | |
| # Try to copy streams during trim operation (if selected) | |
| result = subprocess.run(['ffmpeg', '-i', temp_file.name, '-ss', start_time, '-t', duration, '-c:v', 'copy', '-c:a', 'copy', '-y', output_path], capture_output=True) | |
| if result.returncode != 0: | |
| # If copying the streams failed, re-encode the video | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-ss', start_time, '-t', duration, '-y', output_path]) | |
| else: | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-ss', start_time, '-t', duration, '-y', output_path]) | |
| elif action == 'Resize Video': | |
| width = request.form['width'] | |
| height = request.form['height'] | |
| output_path = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name # Example output format | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-vf', f'scale={width}:{height}', '-y', output_path]) | |
| elif action == 'Extract Audio': | |
| output_path = tempfile.NamedTemporaryFile(delete=False, suffix='.mp3').name # Output audio in MP3 format | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-vn', '-acodec', 'mp3', '-y', output_path]) | |
| elif action == 'Extract Frames': | |
| # Extract frames and save as .jpg files | |
| frame_dir = tempfile.mkdtemp() # Create a temporary directory to store frames | |
| output_zip = tempfile.NamedTemporaryFile(delete=False, suffix='.zip').name # Output ZIP file | |
| # Extract frames from the video (saving as JPG files) | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-an', f'{frame_dir}/frame_%04d.jpg', '-y']) | |
| # Create a ZIP file containing the frames | |
| with zipfile.ZipFile(output_zip, 'w', zipfile.ZIP_DEFLATED) as zipf: | |
| for frame in os.listdir(frame_dir): | |
| if frame.endswith('.jpg'): | |
| zipf.write(os.path.join(frame_dir, frame), frame) # Write each frame into the ZIP | |
| # Clean up the extracted frames directory | |
| for frame in os.listdir(frame_dir): | |
| os.remove(os.path.join(frame_dir, frame)) | |
| os.rmdir(frame_dir) | |
| # Send the ZIP file containing the frames | |
| return send_file(output_zip, as_attachment=True, mimetype='application/zip') | |
| elif action == 'Change Video Speed': | |
| speed_factor = request.form['speed_factor'] | |
| output_path = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name # Example output format | |
| subprocess.run(['ffmpeg', '-i', temp_file.name, '-filter:v', f'setpts={1/speed_factor}*PTS', '-y', output_path]) | |
| # Send the processed file back as a download (if applicable) | |
| if os.path.exists(output_path): | |
| return send_file(output_path, as_attachment=True) | |
| return 'Error processing video', 500 | |
| if __name__ == '__main__': | |
| # Run the app using waitress to serve the HTML and video processing functionality | |
| serve(app, host='0.0.0.0', port=8080) |