Formatting and stuff

src/engines/BackgroundEngine/VideoBackgroundEngine.py

@@ -39,7 +39,7 @@ class VideoBackgroundEngine(BaseBackgroundEngine):
             )
         ]
 
-    def get_background(self) -> mp.VideoClip:
+    def get_background(self):
         background = mp.VideoFileClip(f"{self.background_video}", audio=False)
         background_max_start = background.duration - self.ctx.duration
         if background_max_start < 0:

src/engines/BaseEngine.py

@@ -28,6 +28,7 @@ class BaseEngine(ABC):
     def get_audio_duration(self, path: str) -> float:
         return mp.AudioFileClip(path).duration
 
+    # noinspection PyShadowingBuiltins
     @classmethod
     def get_assets(cls, *, type: str = None, by_id: int = None) -> list[File] | File | None:
         with SessionLocal() as db:
@@ -58,6 +59,7 @@ class BaseEngine(ABC):
                     .all()
                 )
 
+    # noinspection PyShadowingBuiltins
     @classmethod
     def add_asset(cls, *, path: str, metadata: dict, type: str = None):
         with SessionLocal() as db:
@@ -70,6 +72,7 @@ class BaseEngine(ABC):
             db.execute(select(File).filter(File.path == path)).delete()
             db.commit()
 
+    # noinspection PyShadowingBuiltins
     @classmethod
     def store_setting(cls, *, type: str = None, data: dict):
         with SessionLocal() as db:
@@ -92,6 +95,7 @@ class BaseEngine(ABC):
         """
         return cls.retrieve_setting(*args, **kwargs)
 
+    # noinspection PyShadowingBuiltins
     @classmethod
     def retrieve_setting(cls, *, identifier: str = None, type: str = None) -> dict | list[dict] | None:
         """
@@ -127,6 +131,7 @@ class BaseEngine(ABC):
                     .all()
                 ]
 
+    # noinspection PyShadowingBuiltins
     @classmethod
     def remove_setting(cls, *, identifier: str = None, type: str = None):
         """

src/engines/MetadataEngine/BaseMetadataEngine.py

@@ -1,11 +1,11 @@
 from abc import abstractmethod
-from typing import TypedDict
 
-from .. import BaseEngine
+from ..BaseEngine import BaseEngine
 
 
 class BaseMetadataEngine(BaseEngine):
     def __init__(self, **kwargs) -> None:
+        super().__init__()
         ...
 
     @abstractmethod

src/engines/MetadataEngine/ShortsMetadataEngine.py

@@ -10,6 +10,7 @@ class ShortsMetadataEngine(BaseMetadataEngine):
     num_options = 0
 
     def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
         ...
 
     def get_metadata(self):

src/engines/NoneEngine.py

@@ -7,7 +7,7 @@ class NoneEngine(BaseEngine):
     description = "No engine selected"
 
     def __init__(self):
-        pass
+        super().__init__()
 
     @classmethod
     def get_options(cls):

src/engines/ScriptEngine/ShowerThoughtsScriptEngine.py

@@ -15,7 +15,7 @@ class ShowerThoughtsScriptEngine(BaseScriptEngine):
         self.n_sentences = options[0]
         super().__init__()
 
-    def generate(self) -> str:
+    def generate(self):
         sys_prompt, chat_prompt = get_prompt(
             "shower_thoughts",
             location=os.path.join(

src/engines/TTSEngine/BaseTTSEngine.py

@@ -22,17 +22,6 @@ class BaseTTSEngine(BaseEngine):
     def remove_punctuation(self, text: str) -> str:
         return text.translate(str.maketrans("", "", ".,!?;:"))
 
-    def fix_captions(self, script: str, captions: list[Word]) -> list[Word]:
-        script = script.split(" ")
-        new_captions = []
-        for i, word in enumerate(script):
-            original_word = self.remove_punctuation(word.lower())
-            stt_word = self.remove_punctuation(word.lower())
-            if stt_word in original_word:
-                captions[i]["text"] = word
-                new_captions.append(captions[i])
-            # elif there is a word more in the stt than in the original, we
-
     def time_with_whisper(self, path: str) -> list[Word]:
         """
         Transcribes the audio file at the given path using a pre-trained model and returns a list of words.

src/engines/UploadEngine/BaseUploadEngine.py

@@ -1,10 +1,11 @@
 from abc import abstractmethod
 
-from .. import BaseEngine
+from ..BaseEngine import BaseEngine
 
 
 class BaseUploadEngine(BaseEngine):
     def __init__(self, **kwargs) -> None:
+        super().__init__()
         ...
 
     @abstractmethod

src/engines/UploadEngine/TikTokUploadEngine.py

@@ -16,7 +16,7 @@ class TikTokUploadEngine(BaseUploadEngine):
 
     def upload(self):
         cookies = self.get_setting(type="cookies")["cookies"]
-        if cookies == None:
+        if cookies is None:
             gr.Warning(
                 "Skipping upload to TikTok because no cookies were provided. Please provide cookies in the settings."
             )

src/engines/UploadEngine/YouTubeUploadEngine.py

@@ -12,7 +12,8 @@ class YouTubeUploadEngine(BaseUploadEngine):
 
     num_options = 2
 
-    def __init__(self, options: list):
+    def __init__(self, options: list, **kwargs):
+        super().__init__(**kwargs)
         self.oauth_name = options[0]
         self.oauth = self.retrieve_setting(type="oauth_credentials")[self.oauth_name]
         self.credentials = self.retrieve_setting(type="youtube_client_secrets")[self.oauth["client_secret"]]

src/engines/__init__.py

@@ -12,13 +12,7 @@ from . import UploadEngine
 from .BaseEngine import BaseEngine
 from .NoneEngine import NoneEngine
 
-
-class EngineDict(TypedDict):
-    classes: list[BaseEngine]
-    multiple: bool
-
-
-ENGINES: dict[str, EngineDict] = {
+ENGINES: dict[str, dict[str, bool | list[BaseEngine]]] = {
     "SettingsEngine": {
         "classes": [SettingsEngine.SettingsEngine],
         "multiple": False,