musev/auto_prompt/human.py

"""负责按照人相关的属性转化成提词
"""
from typing import List

from .attributes.human import PortraitMultiAttr2Text
from .attributes.attributes import BaseAttribute2Text
from .attributes.attr2template import MultiAttr2PromptTemplate


class PortraitAttr2PromptTemplate(MultiAttr2PromptTemplate):
    """可以将任务字典转化为形象提词模板类
        template class for converting task dictionaries into image prompt templates
    Args:
        MultiAttr2PromptTemplate (_type_): _description_
    """

    templates = "a portrait of {}"

    def __init__(
        self, templates: str = None, attr2text: List = None, name: str = "portrait"
    ) -> None:
        """

        Args:
            templates (str, optional): 形象提词模板，若为None，则使用默认的类属性. Defaults to None.
                portrait prompt template, if None, the default class attribute is used.
            attr2text (List, optional): 形象类需要新增、更新的属性列表，默认使用PortraitMultiAttr2Text中定义的形象属性. Defaults to None.
                the list of attributes that need to be added or updated in the image class, by default, the image attributes defined in PortraitMultiAttr2Text are used.
            name (str, optional): 该形象类的名字. Defaults to "portrait".
                class name of this class instance
        """
        if (
            attr2text is None
            or isinstance(attr2text, list)
            or isinstance(attr2text, BaseAttribute2Text)
        ):
            attr2text = PortraitMultiAttr2Text(funcs=attr2text)
        if templates is None:
            templates = self.templates
        super().__init__(templates, attr2text, name=name)