mirror of
https://github.com/Paillat-dev/viralfactory.git
synced 2026-01-02 09:16:19 +00:00
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
This commit is contained in:
17
src/chore/GenerationContext.py
Normal file
17
src/chore/GenerationContext.py
Normal file
@@ -0,0 +1,17 @@
|
||||
import moviepy
|
||||
|
||||
from .. import engines
|
||||
class GenerationContext:
|
||||
|
||||
def __init__(self, llmengine: engines.LLMEngine.BaseLLMEngine, scriptengine: engines.ScriptEngine.BaseScriptEngine, ttsengine: engines.TTSEngine.BaseTTSEngine) -> None:
|
||||
self.llmengine = llmengine
|
||||
self.llmengine.ctx = self
|
||||
|
||||
self.scriptengine = scriptengine
|
||||
self.scriptengine.ctx = self
|
||||
|
||||
self.ttsengine = ttsengine
|
||||
self.ttsengine.ctx = self
|
||||
|
||||
def process(self):
|
||||
timed_script = self.scriptengine.generate()
|
||||
Reference in New Issue
Block a user