adding ffprobe tool

README.md

@@ -1,8 +1,8 @@
 ---
 title: Ffprobe Tool
-emoji: 🏆
-colorFrom: gray
-colorTo: gray
+emoji: 🧐
+colorFrom: blue
+colorTo: purple
 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 ffprobe import FFProbeTool
+
+launch_gradio_demo(FFProbeTool)

ffprobe.py

@@ -0,0 +1,19 @@
+import json
+import ffmpeg
+from transformers import Tool
+
+class FFProbeTool(Tool):
+    name = "ffprobe_tool"
+    description = """
+    This tool extracts metadata from input video using ffmpeg/ffprobe
+    Input is input_path. Output is a text string.
+    """
+    inputs = ["text"]
+    outputs = ["text"]
+
+    def __call__(self, input_path: str):
+        probe = ffmpeg.probe(input_path)
+        return json.dumps(next(
+            (stream for stream in probe["streams"] if stream["codec_type"] == "video"),
+            None,
+        ), indent=2)

requirements.txt

@@ -0,0 +1,4 @@
+json
+ffmpeg-python
+huggingface_hub
+transformers

tool_config.json

@@ -0,0 +1,6 @@
+{
+  "description": "This tool extracts metadata from input video using ffmpeg/ffprobe. Input is input_path. Output is a text string.",
+  "name": "ffprobe_tool",
+  "tool_class": "ffproble.FFProbeTool"
+}
+