fix: SceneExtractionType

Author: ankit-v2-3

videodb/__init__.py

@@ -7,6 +7,7 @@
 from videodb._utils._video import play_stream
 from videodb._constants import (
     VIDEO_DB_API,
+    SceneExtractionType,
     MediaType,
     SearchType,
     SubtitleAlignment,
@@ -39,6 +40,7 @@
     "SubtitleBorderStyle",
     "SubtitleStyle",
     "TextStyle",
+    "SceneExtractionType",
 ]
 
 

videodb/_constants.py

@@ -23,9 +23,8 @@ class IndexType:
 
 
 class SceneExtractionType:
-    scene = "scene"
-    time_based = "time_based"
-    compression_based = "compression_based"
+    scene_based = "scene"
+    time_based = "time"
 
 
 class Workflows:

videodb/video.py

@@ -208,7 +208,7 @@ def _format_scene_collection(self, collection_data: dict) -> SceneCollection:
 
     def extract_scenes(
         self,
-        extraction_type: SceneExtractionType = SceneExtractionType.scene,
+        extraction_type: SceneExtractionType = SceneExtractionType.scene_based,
         extraction_config: dict = {},
         force: bool = False,
         callback_url: str = None,
@@ -243,7 +243,7 @@ def delete_scene_collection(self, collection_id: str) -> None:
 
     def index_scenes(
         self,
-        extraction_type: SceneExtractionType = SceneExtractionType.scene,
+        extraction_type: SceneExtractionType = SceneExtractionType.scene_based,
         extraction_config: dict = {},
         prompt: str = None,
         scenes: List[Scene] = [],