Formatting & improving imports

.gitignore

@@ -152,7 +152,7 @@ cython_debug/
 #.idea/
 
 output/
-local/*
+local/
 local/presets.json
 cookies.txt
 

src/__init__.py

@@ -1,3 +1,3 @@
+from . import chore
 from . import engines
 from . import utils
-from . import chore

src/chore/GenerationContext.py

@@ -1,12 +1,10 @@
-import moviepy.editor as mp
-import time
 import os
-import gradio as gr
-
+import time
 from datetime import datetime
 
+import moviepy.editor as mp
+
 from .. import engines
-from ..utils.prompting import get_prompt
 from ..models import Video, SessionLocal
 
 

src/engines/AssetsEngine/AssetsEngineSelector.py

@@ -1,7 +1,7 @@
 import json
 
-from ...utils.prompting import get_prompt
 from ...chore import GenerationContext
+from ...utils.prompting import get_prompt
 
 
 class AssetsEngineSelector:

src/engines/AssetsEngine/BaseAssetsEngine.py

@@ -1,7 +1,6 @@
-from abc import ABC, abstractmethod
+from abc import abstractmethod
+
 from ..BaseEngine import BaseEngine
-from typing import TypedDict
-from moviepy.editor import ImageClip, VideoFileClip
 
 
 class BaseAssetsEngine(BaseEngine):

src/engines/AssetsEngine/DallEAssetsEngine.py

@@ -1,15 +1,12 @@
-import gradio as gr
-import openai
-import moviepy.editor as mp
-import io
-import base64
-import time
-import requests
 import os
-
-from moviepy.video.fx.resize import resize
 from typing import Literal, TypedDict
 
+import gradio as gr
+import moviepy.editor as mp
+import openai
+import requests
+from moviepy.video.fx.resize import resize
+
 from . import BaseAssetsEngine
 
 

src/engines/AssetsEngine/GoogleAssetsEngine.py

@@ -1,15 +1,12 @@
+import os
+import os
+import shutil
+from typing import TypedDict
+
 import gradio as gr
 import moviepy.editor as mp
-import io
-import base64
-import time
-import requests
-import shutil
-import os
-
 from google_images_search import GoogleImagesSearch
 from moviepy.video.fx.resize import resize
-from typing import Literal, TypedDict
 
 from . import BaseAssetsEngine
 

src/engines/AssetsEngine/__init__.py

@@ -1,4 +1,4 @@
+from .AssetsEngineSelector import AssetsEngineSelector
 from .BaseAssetsEngine import BaseAssetsEngine
 from .DallEAssetsEngine import DallEAssetsEngine
-from .AssetsEngineSelector import AssetsEngineSelector
 from .GoogleAssetsEngine import GoogleAssetsEngine

src/engines/BackgroundEngine/BaseBackgroundEngine.py

@@ -1,7 +1,6 @@
-from abc import ABC, abstractmethod
-from ..BaseEngine import BaseEngine
+from abc import abstractmethod
 
-from moviepy.editor import VideoClip
+from ..BaseEngine import BaseEngine
 
 
 class BaseBackgroundEngine(BaseEngine):

src/engines/BackgroundEngine/VideoBackgroundEngine.py

@@ -1,12 +1,12 @@
 import os
-import shutil
 import random
+import shutil
 import time
+
 import gradio as gr
 import moviepy.editor as mp
-
-from moviepy.video.fx.resize import resize
 from moviepy.video.fx.crop import crop
+
 from . import BaseBackgroundEngine
 
 

src/engines/BaseEngine.py

@@ -1,7 +1,6 @@
-import gradio as gr
-import moviepy.editor as mp
-
 from abc import ABC, abstractmethod
+
+import moviepy.editor as mp
 from sqlalchemy.future import select
 
 from ..chore import GenerationContext

src/engines/CaptioningEngine/BaseCaptioningEngine.py

@@ -1,7 +1,6 @@
-from abc import ABC, abstractmethod
-from ..BaseEngine import BaseEngine
+from abc import abstractmethod
 
-from moviepy.editor import TextClip
+from ..BaseEngine import BaseEngine
 
 
 class BaseCaptioningEngine(BaseEngine):

src/engines/CaptioningEngine/SimpleCaptioningEngine.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from moviepy.editor import TextClip
-from PIL import ImageFont
+
 from . import BaseCaptioningEngine
 
 

src/engines/LLMEngine/BaseLLMEngine.py

@@ -1,7 +1,6 @@
-from abc import ABC, abstractmethod
-from ..BaseEngine import BaseEngine
+from abc import abstractmethod
 
-import openai
+from ..BaseEngine import BaseEngine
 
 
 class BaseLLMEngine(BaseEngine):

src/engines/LLMEngine/OpenaiLLMEngine.py

@@ -1,9 +1,7 @@
-import openai
 import gradio as gr
+import openai
 import orjson
 
-from abc import ABC, abstractmethod
-
 from .BaseLLMEngine import BaseLLMEngine
 
 OPENAI_POSSIBLE_MODELS = [  # Theese shall be the openai models supporting force_json

src/engines/LLMEngine/__init__.py

@@ -1,3 +1,3 @@
+from .AnthropicLLMEngine import AnthropicLLMEngine
 from .BaseLLMEngine import BaseLLMEngine
 from .OpenaiLLMEngine import OpenaiLLMEngine
-from .AnthropicLLMEngine import AnthropicLLMEngine

src/engines/MetadataEngine/BaseMetadataEngine.py

@@ -1,5 +1,4 @@
 from abc import abstractmethod
-from typing import TypedDict
 
 from .. import BaseEngine
 

src/engines/ScriptEngine/BaseScriptEngine.py

@@ -1,4 +1,5 @@
-from abc import ABC, abstractmethod
+from abc import abstractmethod
+
 from ..BaseEngine import BaseEngine
 
 

src/engines/ScriptEngine/CustomScriptEngine.py

@@ -1,6 +1,7 @@
-from .BaseScriptEngine import BaseScriptEngine
 import gradio as gr
 
+from .BaseScriptEngine import BaseScriptEngine
+
 
 class CustomScriptEngine(BaseScriptEngine):
     name = "Custom Script Engine"

src/engines/ScriptEngine/ShowerThoughtsScriptEngine.py

@@ -1,6 +1,7 @@
-import gradio as gr
 import os
 
+import gradio as gr
+
 from .BaseScriptEngine import BaseScriptEngine
 from ...utils.prompting import get_prompt
 

src/engines/ScriptEngine/__init__.py

@@ -1,3 +1,3 @@
 from .BaseScriptEngine import BaseScriptEngine
-from .ShowerThoughtsScriptEngine import ShowerThoughtsScriptEngine
 from .CustomScriptEngine import CustomScriptEngine
+from .ShowerThoughtsScriptEngine import ShowerThoughtsScriptEngine

src/engines/SettingsEngine/SettingsEngine.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from abc import ABC, abstractmethod
+
 from ..BaseEngine import BaseEngine
 
 

src/engines/TTSEngine/BaseTTSEngine.py

@@ -1,9 +1,9 @@
+from abc import abstractmethod
+from typing import TypedDict
+
 import moviepy.editor as mp
 import whisper_timestamped as wt
-
-from typing import TypedDict
 from torch.cuda import is_available
-from abc import ABC, abstractmethod
 
 from ..BaseEngine import BaseEngine
 

src/engines/TTSEngine/CoquiTTSEngine.py

@@ -1,13 +1,10 @@
-import gradio as gr
-
-from TTS.api import TTS
 import os
 
+import gradio as gr
 import torch
+from TTS.api import TTS
 
-from .BaseTTSEngine import BaseTTSEngine, Word
-
-from ...utils.prompting import get_prompt
+from .BaseTTSEngine import BaseTTSEngine
 
 
 class CoquiTTSEngine(BaseTTSEngine):

src/engines/UploadEngine/TikTokUploadEngine.py

@@ -1,5 +1,4 @@
 import gradio as gr
-
 from tiktok_uploader.upload import upload_video
 
 from .BaseUploadEngine import BaseUploadEngine

src/engines/UploadEngine/YouTubeUploadEngine.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import orjson
-
 from google_auth_oauthlib.flow import InstalledAppFlow
 
 from . import BaseUploadEngine
 from ...utils import youtube_uploading
 
+
 class YouTubeUploadEngine(BaseUploadEngine):
     name = "YouTube"
     description = "Upload videos to YouTube"
@@ -81,6 +81,7 @@ class YouTubeUploadEngine(BaseUploadEngine):
                     label="Client Secret File", file_types=["json"], type="binary"
                 )
                 submit_button = gr.Button("Save")
+
                 def save(binary, clien_secret_name):
                     current_client_secrets = cls.retrieve_setting(type="youtube_client_secrets") or {}
                     client_secret_json = orjson.loads(binary)
@@ -99,6 +100,7 @@ class YouTubeUploadEngine(BaseUploadEngine):
                 choosen_client_secret = gr.Dropdown(label="Login secret", choices=possible_client_secrets)
                 name = gr.Textbox(label="Name", max_lines=1)
                 login_button = gr.Button("Login", variant="primary")
+
                 def login(choosen_client_secret, name):
                     choosen_secret_data = cls.retrieve_setting(type="youtube_client_secrets")[choosen_client_secret]
                     new_oauth_entry = cls.__oauth(choosen_secret_data)
@@ -112,4 +114,5 @@ class YouTubeUploadEngine(BaseUploadEngine):
                         data=current_oauths,
                     )
                     gr.Info(f"{name} saved successfully !")
+
                 login_button.click(login, inputs=[choosen_client_secret, name])

src/engines/__init__.py

@@ -1,15 +1,16 @@
 from typing import TypedDict
+
+from . import AssetsEngine
+from . import BackgroundEngine
+from . import CaptioningEngine
+from . import LLMEngine
+from . import MetadataEngine
+from . import ScriptEngine
+from . import SettingsEngine
+from . import TTSEngine
+from . import UploadEngine
 from .BaseEngine import BaseEngine
 from .NoneEngine import NoneEngine
-from . import TTSEngine
-from . import ScriptEngine
-from . import LLMEngine
-from . import CaptioningEngine
-from . import AssetsEngine
-from . import SettingsEngine
-from . import BackgroundEngine
-from . import MetadataEngine
-from . import UploadEngine
 
 
 class EngineDict(TypedDict):

src/models/DatabaseManager.py

@@ -1,7 +1,5 @@
-import os
-
 from sqlalchemy import create_engine
-from sqlalchemy.orm import Session, sessionmaker
+from sqlalchemy.orm import sessionmaker
 
 from . import Base
 

src/models/File.py

@@ -1,8 +1,8 @@
-from . import Base
-from typing import Optional
 from sqlalchemy import String, Column, JSON, Integer
 from sqlalchemy.ext.mutable import MutableDict
 
+from . import Base
+
 
 class File(Base):
     __tablename__ = "files"

src/models/Setting.py

@@ -1,8 +1,8 @@
-from . import Base
-from typing import Optional
 from sqlalchemy import String, Column, JSON, Integer
 from sqlalchemy.ext.mutable import MutableDict
 
+from . import Base
+
 
 class Setting(Base):
     __tablename__ = "Settings"

src/models/Video.py

@@ -1,8 +1,9 @@
-from . import Base
-from typing import Optional
+from datetime import datetime
+
 from sqlalchemy import String, Column, JSON, Integer, DateTime
 from sqlalchemy.ext.mutable import MutableList
-from datetime import datetime
+
+from . import Base
 
 
 class Video(Base):
@@ -12,6 +13,6 @@ class Video(Base):
     title: str = Column(String, nullable=False)
     description: str = Column(String, nullable=False)
     script: str = Column(String, nullable=False)
-    timed_script: dict = Column(MutableList.as_mutable(JSON), nullable=False)
+    timed_script: dict = Column(MutableList.as_mutable(JSON), nullable=False)  # type: ignore
     timestamp: datetime = Column(DateTime, nullable=False, default=datetime.now())
     path: str = Column(String, nullable=False)

src/utils/prompting.py

@@ -1,7 +1,8 @@
-import yaml
 import os
 from typing import TypedDict
 
+import yaml
+
 
 class Prompt(TypedDict):
     system: str

src/utils/youtube_uploading.py

@@ -1,9 +1,9 @@
-from http import client
-import httplib2
 import random
 import time
+from http import client
 
 import google.oauth2.credentials
+import httplib2
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
 from googleapiclient.http import MediaFileUpload
@@ -34,6 +34,7 @@ API_VERSION = "v3"
 
 VALID_PRIVACY_STATUSES = ("public", "private", "unlisted")
 
+
 def get_youtube(oauth_credentials: dict):
     oauth_credentials = google.oauth2.credentials.Credentials(
         token=oauth_credentials["token"],
@@ -45,6 +46,7 @@ def get_youtube(oauth_credentials: dict):
     )
     return build(API_SERVICE_NAME, API_VERSION, credentials=oauth_credentials)
 
+
 def upload(oauth_credentials, options):
     youtube = get_youtube(oauth_credentials)
     tags = None

ui/gradio_ui.py

@@ -1,10 +1,12 @@
 import os
-import gradio as gr
-import orjson
 import sys
 
-from src.engines import ENGINES, BaseEngine
+import gradio as gr
+import orjson
+
 from src.chore import GenerationContext
+from src.engines import ENGINES, BaseEngine
+
 
 class GenerateUI:
     def __init__(self):
@@ -12,6 +14,7 @@ class GenerateUI:
                 font-size: 5rem !important
             }
         """
+
     def get_presets(self):
         with open("local/presets.json", "r") as f:
             return orjson.loads(f.read())
@@ -53,6 +56,7 @@ class GenerateUI:
     def get_settings_interface(self) -> gr.Blocks:
         with gr.Blocks() as interface:
             reload_ui = gr.Button("Reload UI", variant="primary")
+
             def reload():
                 self.ui.close()
                 sys.exit("Reload")
@@ -115,6 +119,7 @@ class GenerateUI:
                             value=None
                         )
                         preset_button = gr.Button("Load")
+
                         def load_preset(preset_name, *inputs) -> list[gr.update]:
                             with open("local/presets.json", "r") as f:
                                 presets = orjson.loads(f.read())
@@ -128,7 +133,8 @@ class GenerateUI:
                                     for engine in engines:
                                         if engine.name in preset.get(engine_type, {}).keys():
                                             values[0].append(engine.name)
-                                            values.extend(gr.update(value=value) for value in preset[engine_type][engine.name])
+                                            values.extend(
+                                                gr.update(value=value) for value in preset[engine_type][engine.name])
                                         else:
                                             values.extend(gr.update() for _ in range(engine.num_options))
                                     returnable.extend(values)
@@ -153,7 +159,8 @@ class GenerateUI:
                                     presets[preset_name] = new_preset
                                     f.write(orjson.dumps(presets))
                             return [gr.update(value=presets.keys()), *returnable]
-                    preset_button.click(load_preset, inputs=[preset_dropdown, *inputs], outputs=[preset_dropdown,*inputs])
+                    preset_button.click(load_preset, inputs=[preset_dropdown, *inputs],
+                                        outputs=[preset_dropdown, *inputs])
                     output_gallery = gr.Markdown("aaa", render=False)
                     button.click(
                         self.run_generate_interface,

ui/launcher.py

@@ -1,4 +1,6 @@
 from . import GenerateUI
+
+
 def launch():
     ui_generator = GenerateUI()
     ui_generator.launch_ui()