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

src/chore/GenerationContext.py

@@ -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()

src/engines/LLMEngine/OpenaiLLMEngine.py

@@ -1,5 +1,6 @@
 import openai
 import gradio as gr
+import orjson
 
 from abc import ABC, abstractmethod
 
@@ -15,10 +16,29 @@ class OpenaiLLMEngine(BaseLLMEngine):
     name = "OpenAI"
     description = "OpenAI language model engine."
 
-    def generate(self, system_prompt: str, chat_prompt: str, max_tokens: int = 512, temperature: float = 1.0, json_mode: bool= False, top_p: float = 1, frequency_penalty: float = 0, presence_penalty: float = 0) -> str:
-        ... # TODO: Implement this method
+    def __init__(self, options: list) -> None:
+        self.model = options[0]
+        super().__init__()
     
-    def get_options(self) -> list:
+    def generate(self, system_prompt: str, chat_prompt: str, max_tokens: int = 512, temperature: float = 1.0, json_mode: bool= False, top_p: float = 1, frequency_penalty: float = 0, presence_penalty: float = 0) -> str:
+        response = openai.chat.completions.create(
+            model=self.model,
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": chat_prompt},
+            ],
+            max_tokens=max_tokens,
+            temperature=temperature,
+            top_p=top_p,
+            frequency_penalty=frequency_penalty,
+            presence_penalty=presence_penalty,
+            response_format={ "type": "json_object" } if json_mode else openai._types.NOT_GIVEN
+        )
+        return response.choices[0].message.content if not json_mode else orjson.loads(response.choices[0].message.content)
+
+
+    @classmethod
+    def get_options(cls) -> list:
         return [
             gr.Dropdown(
                 label="Model",

src/engines/LLMEngine/__init__.py

@@ -1 +1,2 @@
 from .BaseLLMEngine import BaseLLMEngine
+from .OpenaiLLMEngine import OpenaiLLMEngine

src/engines/ScriptEngine/BaseScriptEngine.py

@@ -8,3 +8,6 @@ class BaseScriptEngine(BaseEngine):
     @abstractmethod
     def generate(self) -> str:
         pass
+
+    def time_script(self):
+        ...

src/engines/ScriptEngine/CustomScriptEngine.py

@@ -0,0 +1,25 @@
+from .BaseScriptEngine import BaseScriptEngine
+import gradio as gr
+
+
+class CustomScriptEngine(BaseScriptEngine):
+    name = "Custom Script Engine"
+    description = "Generate a script with a custom provided prompt"
+    num_options = 1
+
+    def __init__(self, options: list[list | tuple | str | int | float | bool | None]):
+        self.script = options[0]
+        super().__init__()
+
+    def generate(self, *args, **kwargs) -> str:
+        return self.script
+
+    @classmethod
+    def get_options(cls) -> list:
+        return [
+            gr.Textbox(
+                label="Prompt",
+                placeholder="Enter your prompt here",
+                value="",
+            )
+        ]

src/engines/ScriptEngine/__init__.py

@@ -1,2 +1,3 @@
 from .BaseScriptEngine import BaseScriptEngine
 from .ShowerThoughtsScriptEngine import ShowerThoughtsScriptEngine
+from .CustomScriptEngine import CustomScriptEngine

src/engines/__init__.py

@@ -1,8 +1,10 @@
 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": [ScriptEngine.ShowerThoughtsScriptEngine, ScriptEngine.CustomScriptEngine],
 }

ui/gradio_ui.py

@@ -40,7 +40,7 @@ class GenerateUI:
 
     def get_generate_interface(self) -> gr.Blocks:
         with gr.Blocks() as interface:
-            with gr.Row() as row:
+            with gr.Row(equal_height=False) as row:
                 inputs = []
                 with gr.Blocks() as col1:
                     for engine_type, engines in ENGINES.items():