fix: SceneExtractionType

ankit-v2-3 · ankit-v2-3 · commit cc0cbefbc1d8 · 2024-06-14T15:51:32.000+05:30
diff --git a/videodb/__init__.py b/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",
 ]
 
 
diff --git a/videodb/_constants.py b/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:
diff --git a/videodb/video.py b/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] = [],
