src/utils/prompting.py

import yaml
import os
from typing import TypedDict


class Prompt(TypedDict):
    system: str
    chat: str


def get_prompt(name, *, location="src/chore/prompts") -> tuple[str, str]:
    path = os.path.join(os.getcwd(), location, f"{name}.yaml")
    if not os.path.exists(path):
        raise FileNotFoundError(f"Prompt file {path} does not exist.")
    with open(path, "r") as file:
        prompt: Prompt = yaml.safe_load(file)
    return prompt["system"], prompt["chat"]
fix(GenerationContext.py): fix typo in variable name powerfulllmengine to powerfulllmengine for better readability feat(GenerationContext.py): add setup_dir method to create a directory for output files with a timestamp feat(GenerationContext.py): call setup_dir method before generating script and synthesizing audio to ensure output directory exists feat(prompts/fix_captions.yaml): add a new prompt file to provide instructions for fixing captions fix(BaseTTSEngine.py): add force_duration method to adjust audio clip duration if it exceeds a specified duration feat(CoquiTTSEngine.py): add options for forcing duration and specifying duration in the UI feat(utils/prompting.py): add get_prompt function to load prompt files from a specified location fix(gradio_ui.py): set equal_height=True for engine_rows to ensure consistent height for engine options 2024-02-15 12:27:13 +01:00			`import yaml`
			`import os`
			`from typing import TypedDict`

Fix dict key 2024-02-15 17:54:21 +01:00
fix(GenerationContext.py): fix typo in variable name powerfulllmengine to powerfulllmengine for better readability feat(GenerationContext.py): add setup_dir method to create a directory for output files with a timestamp feat(GenerationContext.py): call setup_dir method before generating script and synthesizing audio to ensure output directory exists feat(prompts/fix_captions.yaml): add a new prompt file to provide instructions for fixing captions fix(BaseTTSEngine.py): add force_duration method to adjust audio clip duration if it exceeds a specified duration feat(CoquiTTSEngine.py): add options for forcing duration and specifying duration in the UI feat(utils/prompting.py): add get_prompt function to load prompt files from a specified location fix(gradio_ui.py): set equal_height=True for engine_rows to ensure consistent height for engine options 2024-02-15 12:27:13 +01:00			`class Prompt(TypedDict):`
			`system: str`
			`chat: str`

Fix dict key 2024-02-15 17:54:21 +01:00
			`def get_prompt(name, *, location="src/chore/prompts") -> tuple[str, str]:`
fix(GenerationContext.py): fix typo in variable name powerfulllmengine to powerfulllmengine for better readability feat(GenerationContext.py): add setup_dir method to create a directory for output files with a timestamp feat(GenerationContext.py): call setup_dir method before generating script and synthesizing audio to ensure output directory exists feat(prompts/fix_captions.yaml): add a new prompt file to provide instructions for fixing captions fix(BaseTTSEngine.py): add force_duration method to adjust audio clip duration if it exceeds a specified duration feat(CoquiTTSEngine.py): add options for forcing duration and specifying duration in the UI feat(utils/prompting.py): add get_prompt function to load prompt files from a specified location fix(gradio_ui.py): set equal_height=True for engine_rows to ensure consistent height for engine options 2024-02-15 12:27:13 +01:00			`path = os.path.join(os.getcwd(), location, f"{name}.yaml")`
			`if not os.path.exists(path):`
			`raise FileNotFoundError(f"Prompt file {path} does not exist.")`
			`with open(path, "r") as file:`
Fix dict key 2024-02-15 17:54:21 +01:00			`prompt: Prompt = yaml.safe_load(file)`
			`return prompt["system"], prompt["chat"]`