Spaces:
Paused
Paused
File size: 4,771 Bytes
a83f70a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
from flask import Flask, render_template, request, send_file
import os
import tempfile
import subprocess
import zipfile
# Initialize the Flask app, tell it to look in the current directory for templates
app = Flask(__name__, template_folder='.')
@app.route('/')
def home():
return render_template('html.html') # Flask will look in the current directory
@app.route('/process', methods=['POST'])
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__':
app.run(debug=True) |