chore: move video utils to util functions

.gitignore

@@ -2,4 +2,5 @@
 *.MP4 
 *.mov
 *.MOV
-testing_data
+testing_data
+__pycache__/

app.py

@@ -1,13 +1,10 @@
 import gradio as gr 
 from run_on_video.run import MomentDETRPredictor
-# from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 import torch 
 from lbhd.infer import lbhd_predict
 import os
 import subprocess
-from moviepy.config import get_setting
-from moviepy.tools import subprocess_call
-import datetime
+from utils.export_utils import trim_video
 
 DESCRIPTION = """
 _This Space demonstrates model [QVHighlights: Detecting Moments and Highlights in Videos via Natural Language Queries](https://arxiv.org/abs/2107.09609), NeurIPS 2021, by [Jie Lei](http://www.cs.unc.edu/~jielei/), [Tamara L. Berg](http://tamaraberg.com/), [Mohit Bansal](http://www.cs.unc.edu/~mbansal/)_
@@ -24,28 +21,6 @@ moment_detr_predictor = MomentDETRPredictor(
     device=device
 )
 
-def ffmpeg_extract_subclip(filename, t1, t2, targetname=None):
-    """ Makes a new video file playing video file ``filename`` between
-        the times ``t1`` and ``t2``. """
-    name, ext = os.path.splitext(filename)
-    if not targetname:
-        T1, T2 = [int(1000*t) for t in [t1, t2]]
-        targetname = "%sSUB%d_%d.%s" % (name, T1, T2, ext)
-
-    cmd = [get_setting("FFMPEG_BINARY"),"-y",
-           "-i", filename,
-           "-ss", "%0.4f"%t1,
-           "-t", "%0.4f"%(t2-t1),
-           "-vf",
-           "scale=320:-2",
-           targetname
-        ]
-    print(f'== cmd: {cmd}')
-    subprocess_call(cmd)
-
-def trim_video(video_path, start, end, output_file='result.mp4'):
-    ffmpeg_extract_subclip(video_path, start, end, targetname=output_file)
-    return output_file
 
 def display_prediction(result):
     return f'### Start time:  {result[0]:.2f},   End time:  {result[1]:.2f},  Score:  {result[2]:.2f}'

utils/export_utils.py

@@ -0,0 +1,27 @@
+from moviepy.config import get_setting
+from moviepy.tools import subprocess_call
+import os
+
+
+def ffmpeg_extract_subclip(filename, t1, t2, targetname=None):
+    """ Makes a new video file playing video file ``filename`` between
+        the times ``t1`` and ``t2``. """
+    name, ext = os.path.splitext(filename)
+    if not targetname:
+        T1, T2 = [int(1000*t) for t in [t1, t2]]
+        targetname = "%sSUB%d_%d.%s" % (name, T1, T2, ext)
+
+    cmd = [get_setting("FFMPEG_BINARY"),"-y",
+           "-i", filename,
+           "-ss", "%0.4f"%t1,
+           "-t", "%0.4f"%(t2-t1),
+           "-vf",
+           "scale=320:-2",
+           targetname
+        ]
+    print(f'== cmd: {cmd}')
+    subprocess_call(cmd)
+
+def trim_video(video_path, start, end, output_file='result.mp4'):
+    ffmpeg_extract_subclip(video_path, start, end, targetname=output_file)
+    return output_file