fix: remove scenemodels

Author: ankit-v2-3

videodb/__init__.py

@@ -8,7 +8,6 @@
 from videodb._constants import (
     VIDEO_DB_API,
     MediaType,
-    SceneModels,
     SearchType,
     SubtitleAlignment,
     SubtitleBorderStyle,
@@ -38,7 +37,6 @@
     "SubtitleAlignment",
     "SubtitleBorderStyle",
     "SubtitleStyle",
-    "SceneModels",
 ]
 
 

videodb/_constants.py

@@ -22,12 +22,6 @@ class IndexType:
     scene = "scene"
 
 
-class SceneModels:
-    gemini_vision: str = "gemini-vision"
-    gpt4_vision: str = "gpt4-v"
-    all: str = "all"
-
-
 class Workflows:
     add_subtitles = "add_subtitles"
 

videodb/_utils/_http_client.py

@@ -133,7 +133,7 @@ def _get_output(self, url: str):
             response_json.get("status") == Status.in_progress
             or response_json.get("status") == Status.processing
         ):
-            percentage = response_json.get("data").get("percentage")
+            percentage = response_json.get("data", {}).get("percentage")
             if percentage and self.show_progress and self.progress_bar:
                 self.progress_bar.n = int(percentage)
                 self.progress_bar.update(0)
@@ -169,7 +169,7 @@ def _parse_response(self, response: requests.Response):
                         bar_format="{l_bar}{bar:100}{r_bar}{bar:-100b}",
                     )
                 response_json = self._get_output(
-                    response_json.get("data").get("output_url")
+                    response_json.get("data", {}).get("output_url")
                 )
                 if response_json.get("success"):
                     return response_json.get("data")

videodb/search.py

@@ -186,7 +186,6 @@ def search_inside_video(
         self,
         video_id: str,
         query: str,
-        scene_model: Optional[str] = None,
         result_threshold: Optional[int] = None,
         score_threshold: Optional[int] = None,
         dynamic_score_percentage: Optional[int] = None,
@@ -197,7 +196,6 @@ def search_inside_video(
             data={
                 "index_type": SearchType.scene,
                 "query": query,
-                "model_name": scene_model,
                 "score_threshold": score_threshold,
                 "result_threshold": result_threshold,
             },

videodb/video.py

@@ -3,7 +3,6 @@
 from videodb._constants import (
     ApiPath,
     IndexType,
-    SceneModels,
     SearchType,
     SubtitleStyle,
     Workflows,
@@ -47,7 +46,6 @@ def search(
         self,
         query: str,
         search_type: Optional[str] = SearchType.semantic,
-        scene_model: Optional[str] = SceneModels.gemini_vision,
         result_threshold: Optional[int] = None,
         score_threshold: Optional[int] = None,
         dynamic_score_percentage: Optional[int] = None,
@@ -59,7 +57,6 @@ def search(
             result_threshold=result_threshold,
             score_threshold=score_threshold,
             dynamic_score_percentage=dynamic_score_percentage,
-            scene_model=scene_model,
         )
 
     def delete(self) -> None:
@@ -136,7 +133,6 @@ def index_spoken_words(self) -> None:
 
     def index_scenes(
         self,
-        scene_model: str = SceneModels.gemini_vision,
         force: bool = False,
         prompt: str = None,
         callback_url: str = None,
@@ -145,34 +141,29 @@ def index_scenes(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.index}",
             data={
                 "index_type": IndexType.scene,
-                "model_name": scene_model,
                 "force": force,
                 "prompt": prompt,
                 "callback_url": callback_url,
             },
         )
 
-    def get_scenes(
-        self, scene_model: str = SceneModels.gemini_vision
-    ) -> Union[list, None]:
+    def get_scenes(self) -> Union[list, None]:
         if self.scenes:
             return self.scenes
         scene_data = self._connection.get(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.index}",
             params={
                 "index_type": IndexType.scene,
-                "model_name": scene_model,
             },
         )
         self.scenes = scene_data
         return scene_data if scene_data else None
 
-    def delete_scene_index(self, scene_model: str = SceneModels.gemini_vision) -> None:
+    def delete_scene_index(self) -> None:
         self._connection.post(
             path=f"{ApiPath.video}/{self.id}/{ApiPath.index}/{ApiPath.delete}",
             data={
                 "index_type": IndexType.scene,
-                "model_name": scene_model,
             },
         )
         self.scenes = None