viralfactory/src/engines/MetadataEngine/ShortsMetadataEngine.py

from . import BaseMetadataEngine

from ...utils.prompting import get_prompt


class ShortsMetadataEngine(BaseMetadataEngine):
    name = "ShortsMetadata"
    description = "Generate metadata for YouTube Shorts / TikTok format videos"

    num_options = 0

    def __init__(self, **kwargs) -> None:
        super().__init__(**kwargs)
        ...

    def get_metadata(self):
        sytsem_prompt, chat_prompt = get_prompt(
            "ShortsMetadata", by_file_location=__file__
        )
        chat_prompt = chat_prompt.replace("{script}", self.ctx.script)

        result = self.ctx.simplellmengine.generate(
            chat_prompt=chat_prompt, system_prompt=sytsem_prompt, json_mode=True
        )
        self.ctx.title = result["title"]
        self.ctx.description = result["description"]

    @classmethod
    def get_options(cls):
        return []