Index Interface Updates

ashish-spext · ashish-spext · commit 7036ece22672 · 2024-06-14T15:15:34.000+05:30
- Deleted old scene_index and replaced with renaming new create_scene_index
  - Added missing prompt
- Synced name of scene_collection[s] with server
- Replaced delete_scene_index with renaming new delete_index
diff --git a/videodb/video.py b/videodb/video.py
@@ -161,22 +161,6 @@ def index_spoken_words(
             show_progress=True,
         )
 
-    def index_scenes(
-        self,
-        force: bool = False,
-        prompt: str = None,
-        callback_url: str = None,
-    ) -> None:
-        self._connection.post(
-            path=f"{ApiPath.video}/{self.id}/{ApiPath.index}",
-            data={
-                "index_type": IndexType.scene,
-                "force": force,
-                "prompt": prompt,
-                "callback_url": callback_url,
-            },
-        )
-
     def get_scenes(self) -> Union[list, None]:
         if self.scenes:
             return self.scenes
@@ -216,7 +200,7 @@ def _format_scene_collection(self, collection_data: dict) -> SceneCollection:
 
         return SceneCollection(
             self._connection,
-            collection_data.get("scenes_collection_id"),
+            collection_data.get("scene_collection_id"),
             self.id,
             collection_data.get("config", {}),
             scenes,
@@ -232,36 +216,36 @@ def extract_scenes(
         scenes_data = self._connection.post(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.scenes}",
             data={
-                "index_type": IndexType.scene,
                 "extraction_type": extraction_type,
                 "extraction_config": extraction_config,
                 "force": force,
                 "callback_url": callback_url,
             },
         )
-        return self._format_scene_collection(scenes_data.get("scenes_collection"))
+        return self._format_scene_collection(scenes_data.get("scene_collection"))
 
     def get_scene_collection(self, collection_id: str):
         scenes_data = self._connection.get(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.scenes}/{collection_id}"
         )
-        return self._format_scene_collection(scenes_data.get("scenes_collection"))
+        return self._format_scene_collection(scenes_data.get("scene_collection"))
 
     def get_scene_collections(self):
         scene_collections_data = self._connection.get(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.scenes}"
         )
-        return scene_collections_data.get("scenes_collections", [])
+        return scene_collections_data.get("scene_collections", [])
 
     def delete_scene_collection(self, collection_id: str) -> None:
         self._connection.delete(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.scenes}/{collection_id}"
         )
 
-    def create_scene_index(
+    def index_scenes(
         self,
         extraction_type: SceneExtractionType = SceneExtractionType.scene,
         extraction_config: dict = {},
+        prompt: str = None,
         scenes: List[Scene] = [],
         force: bool = False,
         callback_url: str = None,
@@ -272,6 +256,7 @@ def create_scene_index(
                 "scenes": [scene.to_json() for scene in scenes],
                 "extraction_type": extraction_type,
                 "extraction_config": extraction_config,
+                "prompt": prompt,
                 "force": force,
                 "callback_url": callback_url,
             },
@@ -291,20 +276,11 @@ def get_scene_index(self, scene_index_id: str) -> Scene:
         )
         return index_data.get("scene_index_records", [])
 
-    def delete_index(self, scene_index_id: str) -> None:
+    def delete_scene_index(self, scene_index_id: str) -> None:
         self._connection.delete(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.index}/{ApiPath.scene}/{scene_index_id}"
         )
 
-    def delete_scene_index(self) -> None:
-        self._connection.post(
-            path=f"{ApiPath.video}/{self.id}/{ApiPath.index}/{ApiPath.delete}",
-            data={
-                "index_type": IndexType.scene,
-            },
-        )
-        self.scenes = None
-
     def add_subtitle(self, style: SubtitleStyle = SubtitleStyle()) -> str:
         if not isinstance(style, SubtitleStyle):
             raise ValueError("style must be of type SubtitleStyle")
