Spaces:
Paused
Paused
File size: 3,073 Bytes
2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f a83f70a 2b5095f |
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 |
from flask import Flask, render_template, request, send_file, jsonify
import os
import tempfile
import subprocess
from zipfile import ZipFile
# Initialize the Flask app, setting the template folder to current directory
app = Flask(__name__, template_folder='.')
# Ensure FFmpeg is accessible and executable
ffmpeg_path = "ffmpeg" # Adjust if ffmpeg is not on PATH
@app.route('/')
def home():
return render_template('html.html')
@app.route('/process', methods=['POST'])
def process_video():
try:
video_file = request.files['video']
action = request.form['action']
format_option = request.form.get('format')
copy_streams = 'copy_streams' in request.form
# Save the uploaded video to a temporary file
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
video_file.save(temp_file.name)
output_path = tempfile.mktemp(suffix=f".{format_option or 'mp4'}")
# Prepare ffmpeg command based on action
command = [ffmpeg_path, '-y', '-i', temp_file.name]
if action == 'Convert Format' and format_option:
command.extend(['-c', 'copy' if copy_streams else 'libx264', output_path])
elif action == 'Trim Video':
start_time = request.form.get('start_time')
duration = request.form.get('duration')
command.extend(['-ss', start_time, '-t', duration, '-c', 'copy' if copy_streams else 'libx264', output_path])
elif action == 'Resize Video':
width = request.form.get('width')
height = request.form.get('height')
command.extend(['-vf', f'scale={width}:{height}', '-c:a', 'copy', output_path])
elif action == 'Extract Audio':
output_path = tempfile.mktemp(suffix=".mp3")
command.extend(['-vn', '-acodec', 'mp3', output_path])
elif action == 'Extract Frames':
frame_output_dir = tempfile.mkdtemp()
command.extend(['-vf', 'fps=1', f'{frame_output_dir}/frame_%03d.png'])
# Run the ffmpeg command
result = subprocess.run(command, capture_output=True, text=True)
if result.returncode != 0:
return jsonify({"error": f"Processing failed: {result.stderr}"}), 500
# Handle response
if action == 'Extract Frames':
zip_path = os.path.join(tempfile.gettempdir(), "frames.zip")
with ZipFile(zip_path, 'w') as zipf:
for frame_file in os.listdir(frame_output_dir):
zipf.write(os.path.join(frame_output_dir, frame_file), frame_file)
return send_file(zip_path, as_attachment=True, download_name="frames.zip")
else:
return send_file(output_path, as_attachment=True)
finally:
os.remove(temp_file.name)
if os.path.exists(output_path):
os.remove(output_path)
# Run the app using waitress (adjust port as needed for Hugging Face Spaces)
if __name__ == '__main__':
from waitress import serve
serve(app, host='0.0.0.0', port=7860) |