"""
PregoPal - 全局配置
====================
所有路径、常量、营养数据库、食谱模板集中管理。
"""

from pathlib import Path

# ============================================================
# 目录路径
# ============================================================
DATA_DIR = Path("data")
VOICE_DIR = DATA_DIR / "voices"
LOGS_DIR = DATA_DIR / "logs"
REPORTS_DIR = DATA_DIR / "reports"
FAMILY_FILE = DATA_DIR / "family.json"
DIET_LOG_FILE = DATA_DIR / "diet_logs.json"
NUTRITION_DB_FILE = DATA_DIR / "nutrition_db.json"

# 确保目录存在
for d in [DATA_DIR, VOICE_DIR, LOGS_DIR, REPORTS_DIR]:
    d.mkdir(parents=True, exist_ok=True)

# ============================================================
# 家庭成员身份枚举
# ============================================================
FAMILY_ROLES = ["孕妇", "丈夫", "婆婆", "妈妈", "爸爸", "其他家人"]

# ============================================================
# 孕期阶段
# ============================================================
TRIMESTERS = ["孕早期", "孕中期", "孕晚期"]

# ============================================================
# 营养数据库（内置推荐）
# 注意：此为基于通用知识的估算，后续将替换为中国官方标准
# ============================================================
DEFAULT_NUTRITION_DB = {
    "叶酸": {"category": "维生素", "daily_recommend_mg": 0.4, "foods": ["菠菜", "西兰花", "芦笋", "牛肝", "豆类"], "benefit": "预防胎儿神经管畸形"},
    "铁": {"category": "矿物质", "daily_recommend_mg": 27, "foods": ["红肉", "动物肝脏", "菠菜", "黑木耳", "红枣"], "benefit": "预防孕期贫血"},
    "钙": {"category": "矿物质", "daily_recommend_mg": 1000, "foods": ["牛奶", "酸奶", "豆腐", "芝麻", "小鱼干"], "benefit": "促进胎儿骨骼发育"},
    "DHA": {"category": "脂肪酸", "daily_recommend_mg": 200, "foods": ["三文鱼", "鳕鱼", "核桃", "亚麻籽", "藻油"], "benefit": "促进胎儿大脑发育"},
    "蛋白质": {"category": "宏量营养素", "daily_recommend_g": 70, "foods": ["鸡蛋", "鸡肉", "鱼肉", "豆腐", "牛奶"], "benefit": "胎儿生长发育的基础"},
    "维生素C": {"category": "维生素", "daily_recommend_mg": 85, "foods": ["橙子", "猕猴桃", "草莓", "番茄", "青椒"], "benefit": "增强免疫力，促进铁吸收"},
    "维生素D": {"category": "维生素", "daily_recommend_mcg": 10, "foods": ["蛋黄", "肝脏", "三文鱼", "蘑菇", "晒太阳"], "benefit": "促进钙吸收"},
    "膳食纤维": {"category": "宏量营养素", "daily_recommend_g": 25, "foods": ["燕麦", "全麦面包", "红薯", "蔬菜", "水果"], "benefit": "预防孕期便秘"},
    "锌": {"category": "矿物质", "daily_recommend_mg": 11, "foods": ["牡蛎", "瘦肉", "坚果", "全谷物", "豆类"], "benefit": "促进胎儿生长发育"},
    "碘": {"category": "矿物质", "daily_recommend_mcg": 220, "foods": ["海带", "紫菜", "碘盐", "海鱼", "贝类"], "benefit": "促进胎儿智力发育"},
}

# ============================================================
# 孕期推荐食谱模板
# ============================================================
MEAL_TEMPLATES = {
    "早餐": ["全麦面包+鸡蛋+牛奶", "燕麦粥+坚果+水果", "小米粥+包子+煮鸡蛋", "豆浆+全麦馒头+蔬菜沙拉"],
    "午餐": ["清蒸鱼+糙米饭+炒时蔬", "番茄牛腩+杂粮饭+凉拌黄瓜", "鸡肉沙拉+藜麦饭+紫菜蛋花汤", "豆腐煲+红薯饭+清炒西兰花"],
    "晚餐": ["鲫鱼豆腐汤+小馒头+清炒菠菜", "蒸蛋羹+小米粥+炒青菜", "虾仁西兰花+糙米饭+番茄汤", "瘦肉粥+蒸南瓜+凉拌木耳"],
    "加餐": ["酸奶+坚果", "水果拼盘", "红枣枸杞茶+全麦饼干", "牛奶+燕麦能量棒"],
}

# ============================================================
# 孕期阶段调整建议
# ============================================================
TRIMESTER_ADJUSTMENTS = {
    "孕早期": {"focus": "补充叶酸，缓解孕吐", "avoid": ["油腻", "辛辣"]},
    "孕中期": {"focus": "补充蛋白质、钙、铁", "avoid": []},
    "孕晚期": {"focus": "控制体重，补充DHA", "avoid": ["高糖", "高盐"]},
}

TRIMESTER_TIPS = {
    "孕早期": "🌱 孕早期建议：多补充叶酸，少食多餐，避免空腹。可以吃一些苏打饼干缓解孕吐。",
    "孕中期": "🌿 孕中期建议：胎儿快速发育期，注意补充优质蛋白和钙质。建议每天喝一杯牛奶。",
    "孕晚期": "🌾 孕晚期建议：控制体重增长，减少高糖高盐食物。多补充DHA促进胎儿大脑发育。",
}

# ============================================================
# 数据存储 Schema 版本
# ============================================================
DIET_LOG_SCHEMA_VERSION = "1.0"

# ============================================================
# 声纹识别阈值
# ============================================================
VOICEPRINT_SIMILARITY_THRESHOLD = 0.7