Files
viralfactory/src/engines/__init__.py
Paillat 9f88e6d069 feat(GenerationContext.py): add new file GenerationContext.py to handle the context of generation engines
feat(OpenaiLLMEngine.py): add orjson library for JSON serialization and deserialization, and implement the generate method to make API call to OpenAI chat completions endpoint

feat(__init__.py): import OpenaiLLMEngine in LLMEngine package

feat(BaseScriptEngine.py): add time_script method to the BaseScriptEngine class

feat(CustomScriptEngine.py): add new file CustomScriptEngine.py to handle custom script generation, implement generate method to return the provided script, and add get_options method to provide a textbox for the prompt input

feat(__init__.py): import CustomScriptEngine in ScriptEngine package

feat(__init__.py): import LLMEngine package and add OpenaiLLMEngine to the ENGINES dictionary

refactor(gradio_ui.py): change equal_height attribute of Row to False to allow different heights for input blocks
2024-02-15 11:23:36 +01:00

11 lines
343 B
Python

from . import TTSEngine
from .BaseEngine import BaseEngine
from . import ScriptEngine
from . import LLMEngine
ENGINES = {
"LLMEngine": [LLMEngine.OpenaiLLMEngine],
"TTSEngine": [TTSEngine.CoquiTTSEngine, TTSEngine.ElevenLabsTTSEngine],
"ScriptEngine": [ScriptEngine.ShowerThoughtsScriptEngine, ScriptEngine.CustomScriptEngine],
}