File size: 1,109 Bytes
5f685fd |
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 |
import os
import platform
import sys
import subprocess
import subprocess
import tempfile
def search_program(program_name):
try:
search_cmd = "where" if platform.system() == "Windows" else "which"
return subprocess.check_output([search_cmd, program_name]).decode().strip()
except subprocess.CalledProcessError:
return None
def get_program_path(program_name):
program_path = search_program(program_name)
return program_path
magick_path = get_program_path("magick")
if magick_path:
os.environ['IMAGEMAGICK_BINARY'] = magick_path
import os
def is_running_in_colab():
return 'COLAB_GPU' in os.environ
def handle_path(path, extension = ".mp4"):
if 'https' in path:
if is_running_in_colab():
temp_file = tempfile.NamedTemporaryFile(suffix= extension, delete=False)
# The '-y' option overwrites the output file if it already exists.
command = ['ffmpeg', '-y', '-i', path, temp_file.name]
subprocess.run(command, check=True)
temp_file.close()
return temp_file.name
return path |