mirror of
https://github.com/Paillat-dev/viralfactory.git
synced 2026-01-02 01:06:19 +00:00
31 lines
860 B
Python
31 lines
860 B
Python
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 []
|