from pydantic import BaseSettings

class Settings(BaseSettings):
    dataset: str "senga-ml/dnotes-dataset-v6"
    base_config: str = "naver-clova-ix/donut-base"
    base_processor: str = "naver-clova-ix/donut-base"
    base_model: str = "naver-clova-ix/donut-base"

    def get_model_config(self, shipper_id: str):
        if shipper_id in ["61", "81", "139"]:
            model = "senga-ml/donut-16"
            processor = "senga-ml/donut-16"
        elif shipper_id in ["165", "127", "145"]:
            model = "senga-ml/donut-17"
            processor = "senga-ml/donut-17"
        else:
            model = self.base_model
            processor = self.base_processor
        return model, processor

settings = Settings()