initial commit

video_stabilizer.py

@@ -8,20 +8,20 @@ class VideoStabilizationTool(Tool):
     Inputs are input_path, output_path, smoothing, zoom.
     Output is the output_path.
     """
-    inputs = ["text", "text", "integer", "integer", "integer"]
+    inputs = ["text", "text", "text", "text", "text"]
     outputs = ["text"]
 
     def __call__(
         self,
         input_path: str,
         output_path: str,
-        smoothing: int = 10,
-        zoom: int = 0,
-        shakiness: int = 5,
+        smoothing: str = "10",
+        zoom: str = "0",
+        shakiness: str = "5",
     ):
         (
             ffmpeg.input(input_path)
-            .output("null", vf="vidstabdetect=shakiness={}".format(shakiness), f="null")
+            .output("null", vf="vidstabdetect=shakiness={}".format(int(shakiness)), f="null")
             .overwrite_output()
             .run(capture_stdout=True, capture_stderr=True)
         )
@@ -30,7 +30,7 @@ class VideoStabilizationTool(Tool):
             .output(
                 output_path,
                 vf="vidstabtransform=smoothing={}:zoom={}:input={}".format(
-                    smoothing, zoom, "transforms.trf"
+                    int(smoothing), int(zoom), "transforms.trf"
                 ),
             )
             .overwrite_output()