116 lines
6.2 KiB
Python
116 lines
6.2 KiB
Python
import os
|
||
import sys
|
||
|
||
from loguru import logger
|
||
from nicegui import app
|
||
|
||
# 1. 处理读取库
|
||
try:
|
||
import tomllib as toml_read # Python 3.11+
|
||
except ImportError:
|
||
try:
|
||
import tomli as toml_read
|
||
except ImportError:
|
||
print("错误: 缺少 TOML 读取库。请运行: pip install tomli")
|
||
sys.exit(1)
|
||
|
||
# 2. 处理写入库 (必须安装 pip install tomli-w)
|
||
try:
|
||
import tomli_w as toml_write
|
||
except ImportError:
|
||
# 如果没安装，提供一个 fallback 提示
|
||
toml_write = None
|
||
|
||
|
||
def get_base_dir():
|
||
if getattr(sys, "frozen", False):
|
||
return os.path.dirname(sys.executable)
|
||
else:
|
||
# 假设当前文件在项目根目录或根目录下的某个文件夹中
|
||
return os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
|
||
def get_resource_path(relative_path):
|
||
base_path = get_base_dir()
|
||
external_path = os.path.join(base_path, relative_path)
|
||
if os.path.exists(external_path):
|
||
return external_path
|
||
if getattr(sys, "frozen", False):
|
||
internal_path = os.path.join(sys._MEIPASS, relative_path)
|
||
if os.path.exists(internal_path):
|
||
return internal_path
|
||
return external_path
|
||
|
||
|
||
def init_storage():
|
||
"""如果 storage 是空的，则注入默认配置"""
|
||
cfg = app.storage.general
|
||
if "class_name" not in cfg:
|
||
logger.info("初始化默认配置...")
|
||
default_config = {
|
||
"data_folder": os.path.join(get_base_dir(), "data"),
|
||
"templates_folder": os.path.join(get_base_dir(), "templates"),
|
||
"source_file": "大四班 幼儿学期发展报告.pptx",
|
||
"output_folder": "output",
|
||
"excel_file": "names.xlsx",
|
||
"image_folder": "images",
|
||
"fonts_dir": "fonts",
|
||
"signature_image": "signature.png",
|
||
"class_name": "K4D",
|
||
"teachers": ["康璐璐", "冯宇阳", "孙继艳"],
|
||
"class_type": 2,
|
||
"default_comment": "暂无评语",
|
||
"age_group": "大班上学期",
|
||
"ai": {
|
||
"api_key": "sk-ccb6a1445126f2e56ba50d7622ff350f",
|
||
"api_url": "https://apis.iflow.cn/v1/chat/completions",
|
||
"model": "deepseek-v3.2",
|
||
"prompt": "# Role \n你是一位拥有20年经验的资深幼儿园主班老师。你的文笔温暖、细腻、充满爱意，擅长发现每个孩子身上独特的闪光点。你的评语风格是“治愈系”的，能让家长读完后感到欣慰并对未来充满希望。\n\n# Goal\n请根据用户提供的【幼儿姓名】、【年龄段/班级】以及【日常表现关键词/评分数据】，撰写一份高质量的学期末成长评语。\n# Constraints & Rules\n1. **严格的格式排版 (Strict Formatting)**:\n- **换行**：正文中间不要随意换行，保持为一段完整的段落。\n2. **称呼处理**:\n- 自动识别用户输入的姓名，去掉姓氏。\n- 例如：“王小明” -> 第一行输出“小明宝贝：”。\n3. **分龄侧重 (根据 Age_Group 调整侧重点)**:\n- **小班 (3-4岁)**：侧重于适应集体生活、情绪稳定性、基本生活自理能力、愿意与老师互动。\n- **中班 (4-5岁)**：侧重于社交互动、分享与合作、动手能力、好奇心、规则意识。\n- **大班 (5-6岁)**：侧重于学习习惯、逻辑思维、领导力、任务意识、幼小衔接准备。\n4. **写作结构 (固定内容)**:\n- **开头**：固定文本必须包含：“{class_type}”\n- **正文**：结合【表现关键词】和【性别】，具体描述进步和优点。\n- **结尾**：委婉地提出期望（“如果你能...老师会更为你骄傲”），并送上祝福。\n5. **语气风格**:\n- 积极正面，多用肯定句。\n- 字数控制在 150-250 字之间。\n# Input Format\n- Name {{name}}\n- Age_Group {{class_name}}\n- Traits {{traits}}\n- Sex {{sex}}\n# Output Example\n(假设输入:Name=张图图, Age_Group=小班, Traits=适应能力强, 爱笑, 挑食,Sex=女)\n亲爱的图图宝贝：{class_type}\n你是一个爱笑的小天使，每天早上都能看到你甜甜的笑脸。从一开始的哭鼻子到现在能开心地参与游戏，你的适应能力让老师感到惊喜。不过，老师发现你在吃饭时偶尔会把不喜欢的青菜挑出来哦。如果你能和青菜宝宝做好朋友，把身体练得棒棒的，那就更完美啦！祝可爱的图图宝贝新年快乐，健康成长！\n",
|
||
},
|
||
}
|
||
cfg.update(default_config)
|
||
|
||
|
||
# ==========================================
|
||
# 1. 配置加载 (Config Loader)
|
||
# ==========================================
|
||
def load_config():
|
||
try:
|
||
config_data = app.storage.general
|
||
|
||
base_dir = get_base_dir()
|
||
config = {
|
||
"root_path": base_dir,
|
||
"data_folder": os.path.join(os.path.join("data")),
|
||
# 扁平化映射
|
||
"source_file": get_resource_path(
|
||
os.path.join("templates", config_data.get("source_file", ""))
|
||
),
|
||
"excel_file": get_resource_path(
|
||
os.path.join("data", config_data.get("excel_file", "names"))
|
||
),
|
||
"image_folder": get_resource_path(
|
||
os.path.join("data", config_data.get("image_folder", ""))
|
||
),
|
||
"fonts_dir": get_resource_path(config_data.get("fonts_dir", "fonts")),
|
||
"output_folder": os.path.join(
|
||
base_dir, config_data.get("output_folder", "output")
|
||
),
|
||
"signature_image": get_resource_path(
|
||
os.path.join("data", config_data.get("signature_image", "signature.png"))
|
||
),
|
||
"class_name": config_data.get("class_name", "未命名班级"),
|
||
"teachers": config_data.get("teachers", []),
|
||
"class_type": config_data.get("class_type", 0),
|
||
"default_comment": config_data.get("default_comment", "暂无评语"),
|
||
"age_group": config_data.get("age_group", "大班上学期"),
|
||
"ai": config_data.get(
|
||
"ai", {"api_key": "", "api_url": "", "model": "", "prompt": ""}
|
||
),
|
||
}
|
||
return config
|
||
|
||
except Exception as e:
|
||
print(f"解析配置文件失败: {e}")
|
||
return {}
|