Formatting

src/chore/GenerationContext.py

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

src/engines/AssetsEngine/BaseAssetsEngine.py

@@ -1,4 +1,7 @@
-from abc import abstractmethod
+from abc import ABC, abstractmethod
+from typing import TypedDict
+
+from moviepy.editor import ImageClip, VideoFileClip
 
 from ..BaseEngine import BaseEngine
 

src/engines/AssetsEngine/DallEAssetsEngine.py

@@ -1,4 +1,7 @@
+import base64
+import io
 import os
+import time
 from typing import Literal, TypedDict
 
 import gradio as gr

src/engines/AssetsEngine/GoogleAssetsEngine.py

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

src/engines/BackgroundEngine/BaseBackgroundEngine.py

@@ -1,4 +1,6 @@
-from abc import abstractmethod
+from abc import ABC, abstractmethod
+
+from moviepy.editor import VideoClip
 
 from ..BaseEngine import BaseEngine
 

src/engines/BackgroundEngine/VideoBackgroundEngine.py

@@ -6,6 +6,7 @@ import time
 import gradio as gr
 import moviepy.editor as mp
 from moviepy.video.fx.crop import crop
+from moviepy.video.fx.resize import resize
 
 from . import BaseBackgroundEngine
 

src/engines/BaseEngine.py

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

src/engines/CaptioningEngine/BaseCaptioningEngine.py

@@ -1,4 +1,6 @@
-from abc import abstractmethod
+from abc import ABC, abstractmethod
+
+from moviepy.editor import TextClip
 
 from ..BaseEngine import BaseEngine
 

src/engines/CaptioningEngine/SimpleCaptioningEngine.py

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

src/engines/LLMEngine/BaseLLMEngine.py

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

src/engines/LLMEngine/OpenaiLLMEngine.py

@@ -1,3 +1,5 @@
+from abc import ABC, abstractmethod
+
 import gradio as gr
 import openai
 import orjson

src/engines/MetadataEngine/BaseMetadataEngine.py

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

src/engines/ScriptEngine/BaseScriptEngine.py

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

src/engines/SettingsEngine/SettingsEngine.py

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

src/engines/TTSEngine/BaseTTSEngine.py

@@ -1,4 +1,4 @@
-from abc import abstractmethod
+from abc import ABC, abstractmethod
 from typing import TypedDict
 
 import moviepy.editor as mp

src/engines/TTSEngine/CoquiTTSEngine.py

@@ -4,7 +4,8 @@ import gradio as gr
 import torch
 from TTS.api import TTS
 
-from .BaseTTSEngine import BaseTTSEngine
+from .BaseTTSEngine import BaseTTSEngine, Word
+from ...utils.prompting import get_prompt
 
 
 class CoquiTTSEngine(BaseTTSEngine):

src/engines/engines.json

@@ -1,6 +1,6 @@
 {
-  "TTSEngine": [
-    "CoquiTTSEngine",
-    "ElevenLabsTTSEngine"
-  ]
+    "TTSEngine": [
+        "CoquiTTSEngine",
+        "ElevenLabsTTSEngine"
+    ]
 }

src/models/DatabaseManager.py

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

src/models/File.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from sqlalchemy import String, Column, JSON, Integer
 from sqlalchemy.ext.mutable import MutableDict
 

src/models/Setting.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from sqlalchemy import String, Column, JSON, Integer
 from sqlalchemy.ext.mutable import MutableDict
 

src/models/Video.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+from typing import Optional
 
 from sqlalchemy import String, Column, JSON, Integer, DateTime
 from sqlalchemy.ext.mutable import MutableList

ui/gradio_ui.py

@@ -1,12 +1,10 @@
 import os
-import sys
-
 import gradio as gr
 import orjson
+import sys
 
-from src.chore import GenerationContext
 from src.engines import ENGINES, BaseEngine
-
+from src.chore import GenerationContext
 
 class GenerateUI:
     def __init__(self):
@@ -14,7 +12,6 @@ class GenerateUI:
                 font-size: 5rem !important
             }
         """
-
     def get_presets(self):
         with open("local/presets.json", "r") as f:
             return orjson.loads(f.read())
@@ -56,7 +53,6 @@ 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")
@@ -119,13 +115,12 @@ 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())
                             returnable = []
                             if preset_name in presets.keys():
-                                # If the preset exists
+                            # If the preset exists
                                 preset = presets[preset_name]
                                 for engine_type, engines in ENGINES.items():
                                     engines = engines["classes"]
@@ -133,8 +128,7 @@ 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)
@@ -159,8 +153,7 @@ 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,
@@ -204,8 +197,8 @@ class GenerateUI:
                     options[engine_type].append(
                         engine(options=args[: engine.num_options])
                     )
-                    args = args[engine.num_options:]
+                    args = args[engine.num_options :]
                 else:
                     # we don't care about this, it's not the selected engine, we throw it away
-                    args = args[engine.num_options:]
+                    args = args[engine.num_options :]
         return options