Merge pull request #278 from edenai/add-duration-to-file-info

appelstroop · web-flow · commit 04cfa46ee5c7 · 2024-10-23T10:57:29.000+02:00
Add optional duration to FileInfo
diff --git a/edenai_apis/utils/files.py b/edenai_apis/utils/files.py
@@ -11,13 +11,15 @@ def __init__(
         self.file_extension = file_extension
         if args:
             self.file_frame_rate, self.file_channels = args
-
+        self.file_duration = kwargs.get("duration", None)
     file_size: int
     file_media_type: str
     supported_extensions: List[str]
     # for audio and video files
     file_frame_rate: Optional[str]
     file_channels: Optional[str]
+    file_duration: Optional[float]
+
 
 
 class FileWrapper:
