188 lines
5.4 KiB
Python
188 lines
5.4 KiB
Python
|
import os
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
DEFAULT_CONFIG: dict[str, Any] = {
|
|||
|
|
# 应用展示名称，同时作为 AppData 默认目录名的一部分
|
|||
|
|
"APP_NAME": "AiShiliu",
|
|||
|
|
# Flask 后端监听地址
|
|||
|
|
"APP_HOST": "127.0.0.1",
|
|||
|
|
# Flask 后端监听端口
|
|||
|
|
"APP_PORT": 5000,
|
|||
|
|
# 应用默认时区
|
|||
|
|
"APP_TIMEZONE": "Asia/Shanghai",
|
|||
|
|
# 已构建前端静态资源目录，留空时自动回退到项目默认目录
|
|||
|
|
"FRONTEND_STATIC_DIR": "",
|
|||
|
|
|
|||
|
|
# 日志总开关
|
|||
|
|
"LOG_ENABLED": True,
|
|||
|
|
# 日志级别
|
|||
|
|
"LOG_LEVEL": "INFO",
|
|||
|
|
# 是否写文本日志
|
|||
|
|
"LOG_TEXT_ENABLED": True,
|
|||
|
|
# 是否写JSON日志
|
|||
|
|
"LOG_JSON_ENABLED": True,
|
|||
|
|
# 单文件滚动大小（MB）
|
|||
|
|
"LOG_ROTATE_MB": 5,
|
|||
|
|
# 滚动保留份数
|
|||
|
|
"LOG_BACKUP_COUNT": 7,
|
|||
|
|
|
|||
|
|
# MySQL 主机地址（预留给后续鉴权等在线能力）
|
|||
|
|
"DB_HOST": "127.0.0.1",
|
|||
|
|
# MySQL 端口（预留）
|
|||
|
|
"DB_PORT": 3306,
|
|||
|
|
# MySQL 数据库名（预留）
|
|||
|
|
"DB_NAME": "ai_shiliu",
|
|||
|
|
# MySQL 用户名（预留）
|
|||
|
|
"DB_USER": "ai_shiliu",
|
|||
|
|
# MySQL 密码（预留）
|
|||
|
|
"DB_PASS": "",
|
|||
|
|
# MySQL 字符集（预留）
|
|||
|
|
"DB_CHARSET": "utf8mb4",
|
|||
|
|
|
|||
|
|
# AI 提供方（openai/deepseek/dify）
|
|||
|
|
"AI_PROVIDER": "dify",
|
|||
|
|
# OpenAI 接口密钥
|
|||
|
|
"OPENAI_API_KEY": "YOUR_OPENAI_API_KEY_HERE",
|
|||
|
|
# OpenAI API 基础地址
|
|||
|
|
"OPENAI_API_BASE": "https://api.openai.com/v1",
|
|||
|
|
# OpenAI 默认模型
|
|||
|
|
"OPENAI_MODEL": "gpt-4.1-mini",
|
|||
|
|
# DeepSeek 接口密钥
|
|||
|
|
"DEEPSEEK_API_KEY": "",
|
|||
|
|
# DeepSeek API 基础地址
|
|||
|
|
"DEEPSEEK_API_BASE": "https://api.deepseek.com",
|
|||
|
|
# DeepSeek 默认模型
|
|||
|
|
"DEEPSEEK_MODEL": "deepseek-chat",
|
|||
|
|
# Dify 接口密钥
|
|||
|
|
"DIFY_API_KEY": "app-a9dofsiQi4e157uDYTx8Lrja",
|
|||
|
|
# Dify API 基础地址
|
|||
|
|
"DIFY_API_BASE": "http://47.92.48.126/v1",
|
|||
|
|
# Dify 用户标识
|
|||
|
|
"DIFY_USER": "wechat_user",
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 百度 OCR API Key
|
|||
|
|
"BAIDU_API_KEY": "ElIQN30iAqpEGi9zv0VlrtQX",
|
|||
|
|
# 百度 OCR Secret Key
|
|||
|
|
"BAIDU_SECRET_KEY": "7wrO2wDTx7FehuelgG0NCBDFOklnqSz0",
|
|||
|
|
# OCR 引擎（baidu/rapid）
|
|||
|
|
"OCR_PROVIDER": "rapid",
|
|||
|
|
|
|||
|
|
# 微信机器人上报后端消息接口
|
|||
|
|
"BACKEND_URL": "http://127.0.0.1:5000/api/messages/receive",
|
|||
|
|
# 轮询主循环间隔（秒）
|
|||
|
|
"BOT_LOOP_INTERVAL": 3,
|
|||
|
|
# 点击联系人后等待（秒）
|
|||
|
|
"BOT_CLICK_AFTER_DELAY": 1.2,
|
|||
|
|
# 切换会话后读取标题等待（秒）
|
|||
|
|
"BOT_TITLE_AFTER_DELAY": 1.0,
|
|||
|
|
# 联系人切换节流等待（秒）
|
|||
|
|
"BOT_CONTACT_SWITCH_DELAY": 1.0,
|
|||
|
|
# 主循环异常后的重试等待（秒）
|
|||
|
|
"BOT_LOOP_ERROR_DELAY": 3.0,
|
|||
|
|
|
|||
|
|
# 微信窗口目标宽度
|
|||
|
|
"WECHAT_WINDOW_TARGET_WIDTH": 1080,
|
|||
|
|
# 微信窗口目标高度
|
|||
|
|
"WECHAT_WINDOW_TARGET_HEIGHT": 820,
|
|||
|
|
# 微信窗口目标左侧坐标
|
|||
|
|
"WECHAT_WINDOW_TARGET_LEFT": 120,
|
|||
|
|
# 微信窗口目标顶部坐标
|
|||
|
|
"WECHAT_WINDOW_TARGET_TOP": 80,
|
|||
|
|
|
|||
|
|
# 是否保存 OCR 调试截图
|
|||
|
|
"OCR_SAVE_IMAGES": True,
|
|||
|
|
|
|||
|
|
# 联系人行高（用于定位）
|
|||
|
|
"CONTACT_ROW_HEIGHT": 64,
|
|||
|
|
# 联系人列宽（用于定位）
|
|||
|
|
"CONTACT_ROW_WIDTH": 240,
|
|||
|
|
# 联系人列表左偏移
|
|||
|
|
"CONTACT_LIST_LEFT_OFFSET": 68,
|
|||
|
|
# 联系人列表上偏移
|
|||
|
|
"CONTACT_LIST_TOP_OFFSET": 82,
|
|||
|
|
# 联系人列表下偏移
|
|||
|
|
"CONTACT_LIST_BOTTOM_OFFSET": 0,
|
|||
|
|
|
|||
|
|
# 会话名区域左偏移
|
|||
|
|
"SESSION_NAME_LEFT_OFFSET": 48,
|
|||
|
|
# 会话名区域上偏移
|
|||
|
|
"SESSION_NAME_TOP_OFFSET": 8,
|
|||
|
|
# 会话名区域宽度
|
|||
|
|
"SESSION_NAME_WIDTH": 140,
|
|||
|
|
# 会话名区域高度
|
|||
|
|
"SESSION_NAME_HEIGHT": 24,
|
|||
|
|
# 会话名 OCR 放大倍数
|
|||
|
|
"SESSION_NAME_OCR_SCALE": 4,
|
|||
|
|
# 会话名 OCR 额外放大倍数
|
|||
|
|
"SESSION_NAME_OCR_EXTRA_SCALE": 6,
|
|||
|
|
|
|||
|
|
# 聊天区截图左偏移
|
|||
|
|
"CHAT_CAPTURE_LEFT_OFFSET": 310,
|
|||
|
|
# 聊天区截图上偏移
|
|||
|
|
"CHAT_CAPTURE_TOP_OFFSET": 70,
|
|||
|
|
# 聊天区截图宽度
|
|||
|
|
"CHAT_CAPTURE_WIDTH": 750,
|
|||
|
|
# 聊天区截图高度
|
|||
|
|
"CHAT_CAPTURE_HEIGHT": 550,
|
|||
|
|
|
|||
|
|
# 顶部惩罚权重比例
|
|||
|
|
"OCR_TOP_PENALTY_RATIO": 0.18,
|
|||
|
|
# 顶部惩罚二值化系数
|
|||
|
|
"OCR_TOP_PENALTY_BIN_FACTOR": 2.0,
|
|||
|
|
# 顶部惩罚颜色系数
|
|||
|
|
"OCR_TOP_PENALTY_COLOR_FACTOR": 2.2,
|
|||
|
|
|
|||
|
|
# 标题 OCR 区域左偏移（保留较大的方案）
|
|||
|
|
"TITLE_OCR_AREA_LEFT_OFFSET": 310,
|
|||
|
|
# 标题 OCR 区域上偏移（保留较大的方案）
|
|||
|
|
"TITLE_OCR_AREA_TOP_OFFSET": 0,
|
|||
|
|
# 标题 OCR 区域宽度（保留较大的方案）
|
|||
|
|
"TITLE_OCR_AREA_WIDTH": 600,
|
|||
|
|
# 标题 OCR 区域高度（保留较大的方案）
|
|||
|
|
"TITLE_OCR_AREA_HEIGHT": 70,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _raw_value(key: str) -> Any:
|
|||
|
|
env_value = os.getenv(key)
|
|||
|
|
if env_value is not None:
|
|||
|
|
return env_value
|
|||
|
|
return DEFAULT_CONFIG.get(key)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_str(key: str, default: str = "") -> str:
|
|||
|
|
value = _raw_value(key)
|
|||
|
|
if value is None:
|
|||
|
|
return default
|
|||
|
|
return str(value)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_int(key: str, default: int = 0) -> int:
|
|||
|
|
value = _raw_value(key)
|
|||
|
|
if value is None or value == "":
|
|||
|
|
return default
|
|||
|
|
return int(value)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_float(key: str, default: float = 0.0) -> float:
|
|||
|
|
value = _raw_value(key)
|
|||
|
|
if value is None or value == "":
|
|||
|
|
return default
|
|||
|
|
return float(value)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_bool(key: str, default: bool = False) -> bool:
|
|||
|
|
value = _raw_value(key)
|
|||
|
|
if value is None:
|
|||
|
|
return default
|
|||
|
|
if isinstance(value, bool):
|
|||
|
|
return value
|
|||
|
|
normalized = str(value).strip().lower()
|
|||
|
|
if normalized in {"1", "true", "yes", "on"}:
|
|||
|
|
return True
|
|||
|
|
if normalized in {"0", "false", "no", "off"}:
|
|||
|
|
return False
|
|||
|
|
return default
|