# -*- coding: utf-8 -*-


class Translate:
    def __init__(self, src_lang: str, target_lang: str) -> None:
        """
        :param str src_lang: source language
        :param str target_lang: target language

        **Options for source & target language**
            * *th* - *en* - Thai to English
            * *en* - *th* - English to Thai
            * *th* - *zh* - Thai to Chinese
            * *zh* - *th* - Chinese to Thai
        
        :Example:

        Translate text from Thai to English::

            from pythainlp.translate import Translate
            th2en = Translate('th', 'en')

            th2en.translate("ฉันรักแมว")
            # output: I love cat.
        """
        self.model = None
        self.load_model(src_lang, target_lang)

    def load_model(self, src_lang: str, target_lang: str):
        if src_lang == "th" and target_lang == "en":
            from pythainlp.translate.en_th import ThEnTranslator
            self.model = ThEnTranslator()
        elif src_lang == "en" and target_lang == "th":
            from pythainlp.translate.en_th import EnThTranslator
            self.model = EnThTranslator()
        elif src_lang == "th" and target_lang == "zh":
            from pythainlp.translate.zh_th import ThZhTranslator
            self.model = ThZhTranslator()
        elif src_lang == "zh" and target_lang == "th":
            from pythainlp.translate.zh_th import ZhThTranslator
            self.model = ZhThTranslator()
        else:
            raise ValueError("Not support language!")

    def translate(self, text) -> str:
        """
        Translate text

        :param str text: input text in source language
        :return: translated text in target language
        :rtype: str
        """
        return self.model.translate(text)

# -*- coding: utf-8 -*-


class Translate:
    def __init__(self, src_lang: str, target_lang: str) -> None:
        """
        :param str src_lang: source language
        :param str target_lang: target language

        **Options for source & target language**
            * *th* - *en* - Thai to English
            * *en* - *th* - English to Thai
            * *th* - *zh* - Thai to Chinese
            * *zh* - *th* - Chinese to Thai

        :Example:

        Translate text from Thai to English::

            from pythainlp.translate import Translate
            th2en = Translate('th', 'en')

            th2en.translate("ฉันรักแมว")
            # output: I love cat.
        """
        self.model = None
        self.load_model(src_lang, target_lang)

    def load_model(self, src_lang: str, target_lang: str):
        if src_lang == "th" and target_lang == "en":
            from pythainlp.translate.en_th import ThEnTranslator
            self.model = ThEnTranslator()
        elif src_lang == "en" and target_lang == "th":
            from pythainlp.translate.en_th import EnThTranslator
            self.model = EnThTranslator()
        elif src_lang == "th" and target_lang == "zh":
            from pythainlp.translate.zh_th import ThZhTranslator
            self.model = ThZhTranslator()
        elif src_lang == "zh" and target_lang == "th":
            from pythainlp.translate.zh_th import ZhThTranslator
            self.model = ZhThTranslator()
        else:
            raise ValueError("Not support language!")

    def translate(self, text) -> str:
        """
        Translate text

        :param str text: input text in source language
        :return: translated text in target language
        :rtype: str
        """
        return self.model.translate(text)

Update core.py