adding video_crop_tool

README.md

@@ -1,8 +1,8 @@
 ---
 title: Video Crop Tool
-emoji: 🔥
+emoji: ✂️
 colorFrom: green
-colorTo: yellow
+colorTo: orange
 sdk: gradio
 sdk_version: 3.33.1
 app_file: app.py

app.py

@@ -0,0 +1,4 @@
+from transformers.tools.base import launch_gradio_demo
+from video_crop import VideoCropTool 
+
+launch_gradio_demo(VideoCropTool)

requirements.txt

@@ -0,0 +1,3 @@
+ffmpeg-python
+huggingface_hub
+transformers

tool_config.json

@@ -0,0 +1,6 @@
+{
+  "description": "This tool crops a video with inputs: input_path, output_path, top_x, top_y, bottom_x, bottom_y. Output is the output_path.",
+  "name": "video_crop_tool",
+  "tool_class": "video_crop.VideoCropTool"
+}
+

video_crop.py

@@ -0,0 +1,30 @@
+import json
+import ffmpeg
+from transformers import Tool
+
+class VideoCropTool(Tool):
+    name = "video_crop_tool"
+    description = """
+    This tool crops a video with inputs: 
+    input_path, output_path, 
+    top_x, top_y, 
+    bottom_x, bottom_y.
+    Output is the output_path.
+    """                                                                  
+    inputs = ["text", "text", "text", "text", "text", "text"]            
+    outputs = ["text"]
+                 
+    def __call__(
+        self,           
+        input_path: str, 
+        output_path: str,
+        top_x: str,
+        top_y: str,   
+        bottom_x: str,
+        bottom_y: str,
+    ):                                   
+        stream = ffmpeg.input(input_path)                                             
+        stream = ffmpeg.crop(stream, int(top_y), int(top_x), int(bottom_y) - int(top_y), int(bottom_x) - int(top_x))
+        stream = ffmpeg.output(stream, output_path)
+        ffmpeg.run(stream)
+        return output_path