Files
viralfactory/src/engines/__init__.py
Paillat bd00a49063 🐛 fix(GenerationContext.py): fix indentation issue in process() method
 feat(GenerationContext.py): add support for z-index of moviepy clips to improve video rendering
The indentation issue in the process() method has been fixed. The z-index of moviepy clips has been added to improve the rendering of the video. This allows the clips to be rendered in different layers based on their index, resulting in a more visually appealing video.
2024-02-21 09:06:36 +01:00

61 lines
1.6 KiB
Python

from typing import TypedDict
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
class EngineDict(TypedDict):
classes: list[BaseEngine]
multiple: bool
ENGINES: dict[str, EngineDict] = {
"SettingsEngine": {
"classes": [SettingsEngine.SettingsEngine],
"multiple": False,
"show_dropdown": False,
},
"SimpleLLMEngine": {
"classes": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine],
"multiple": False,
},
"PowerfulLLMEngine": {
"classes": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine],
"multiple": False,
},
"ScriptEngine": {
"classes": [
ScriptEngine.ShowerThoughtsScriptEngine,
ScriptEngine.CustomScriptEngine,
],
"multiple": False,
},
"TTSEngine": {
"classes": [TTSEngine.CoquiTTSEngine, TTSEngine.ElevenLabsTTSEngine],
"multiple": False,
},
"CaptioningEngine": {
"classes": [CaptioningEngine.SimpleCaptioningEngine, NoneEngine],
"multiple": False,
},
"AssetsEngine": {
"classes": [AssetsEngine.DallEAssetsEngine, NoneEngine],
"multiple": True,
},
"BackgroundEngine": {
"classes": [BackgroundEngine.VideoBackgroundEngine, NoneEngine],
"multiple": False,
},
"MetadataEngine": {
"classes": [MetadataEngine.ShortsMetadataEngine],
"multiple": False,
},
}