storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-29 09:18:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

gradually move to argparse, pass output video name

Browse Source
This commit is contained in:
Brandon Thomas
2024-07-09 19:55:37 -04:00
parent 1a5e0b6558
commit b650e956df
1 changed files with 25 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflows/comfy/format_video.py
+25 -6
View File
@@ -1,3 +1,4 @@
import argparse
import os import os
import subprocess import subprocess
import json import json
@@ -50,7 +51,7 @@ def format_timestamp(ms):
seconds, ms = divmod(ms, 1000) seconds, ms = divmod(ms, 1000)
return f"{int(hours):02d}:{int(minutes):02d}:{int(seconds):02d}.{int(ms):03d}" return f"{int(hours):02d}:{int(minutes):02d}:{int(seconds):02d}.{int(ms):03d}"
def process_video(input_video, max_resolution, start_ms, end_ms, framerate): def process_video(input_video, max_resolution, start_ms, end_ms, framerate, output_video = None):
width, height = get_video_resolution(input_video) width, height = get_video_resolution(input_video)
print(f"Original resolution: {width}x{height}") print(f"Original resolution: {width}x{height}")
@@ -61,10 +62,10 @@ def process_video(input_video, max_resolution, start_ms, end_ms, framerate):
duration_ms = end_ms - start_ms duration_ms = end_ms - start_ms
duration = format_timestamp(duration_ms) duration = format_timestamp(duration_ms)
# get input video directory if not output_video:
input_vid_dir = Path(input_video).absolute().parent # NB: This is salt's old implicit output video naming
output_video = input_vid_dir.joinpath(f"input.mp4") input_vid_dir = Path(input_video).absolute().parent
# os.makedirs(os.path.dirname(output_video), exist_ok=True) output_video = input_vid_dir.joinpath(f"input.mp4")
command = [ command = [
'ffmpeg', 'ffmpeg',
@@ -90,11 +91,29 @@ if __name__ == "__main__":
print("Usage: python script.py <input_video> <start_ms> <end_ms> <framerate>") print("Usage: python script.py <input_video> <start_ms> <end_ms> <framerate>")
sys.exit(1) sys.exit(1)
# Old Salt Args (replace this mess)
input_video = sys.argv[1] input_video = sys.argv[1]
output_video = None
# max_resolution = sys.argv[2] # max_resolution = sys.argv[2]
max_resolution = None max_resolution = None
start_ms = int(sys.argv[2]) start_ms = int(sys.argv[2])
end_ms = int(sys.argv[3]) end_ms = int(sys.argv[3])
framerate = float(sys.argv[4]) framerate = float(sys.argv[4])
process_video(input_video, max_resolution, start_ms, end_ms, framerate) remaining_args = sys.argv[5:]
print('remaining args', remaining_args)
parser = argparse.ArgumentParser(description='Format Video')
# TODO: Output and input are optional for now until we migrate away from salt's arguments.
parser.add_argument('--input', type=str, help='Input Video Filename', required=False)
parser.add_argument('--output', type=str, help='Output Video Filename', required=False)
args = parser.parse_args(remaining_args)
if args.input:
input_video = args.input
if args.output:
output_video = args.output
process_video(input_video, max_resolution, start_ms, end_ms, framerate, output_video=output_video)