52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""Language configuration for prompts"""
|
|
|
|
from enum import Enum
|
|
from typing import Literal
|
|
|
|
class Language(str, Enum):
|
|
CHINESE = "zh"
|
|
ENGLISH = "en"
|
|
|
|
LanguageType = Literal["zh", "en"]
|
|
|
|
# Default categories for each language
|
|
DEFAULT_CATEGORIES = {
|
|
"zh": ["材料", "功能", "用途", "使用族群", "特性"],
|
|
"en": ["Materials", "Functions", "Usages", "User Groups", "Characteristics"],
|
|
}
|
|
|
|
CATEGORY_DESCRIPTIONS = {
|
|
"zh": {
|
|
"材料": "物件由什麼材料組成",
|
|
"功能": "物件能做什麼",
|
|
"用途": "物件在什麼場景使用",
|
|
"使用族群": "誰會使用這個物件",
|
|
"特性": "物件有什麼特徵",
|
|
},
|
|
"en": {
|
|
"Materials": "What materials the object is made of",
|
|
"Functions": "What the object can do",
|
|
"Usages": "In what scenarios the object is used",
|
|
"User Groups": "Who uses this object",
|
|
"Characteristics": "What features the object has",
|
|
},
|
|
}
|
|
|
|
# Category name mappings between languages
|
|
CATEGORY_MAPPING = {
|
|
"zh_to_en": {
|
|
"材料": "Materials",
|
|
"功能": "Functions",
|
|
"用途": "Usages",
|
|
"使用族群": "User Groups",
|
|
"特性": "Characteristics",
|
|
},
|
|
"en_to_zh": {
|
|
"Materials": "材料",
|
|
"Functions": "功能",
|
|
"Usages": "用途",
|
|
"User Groups": "使用族群",
|
|
"Characteristics": "特性",
|
|
},
|
|
}
|