mirror of
https://github.com/Paillat-dev/viralfactory.git
synced 2026-01-02 17:24:54 +00:00
51 lines
1.3 KiB
Python
51 lines
1.3 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
|
|
|
|
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,
|
|
},
|
|
}
|