"""
ParamHub 配置文件
"""
import secrets
from pathlib import Path

# 项目根目录
BASE_DIR = Path(__file__).parent

# 数据目录
DATA_DIR = BASE_DIR / 'data'
DATA_DIR.mkdir(exist_ok=True)

# 图片上传目录
IMAGES_DIR = BASE_DIR / 'static' / 'uploads'
IMAGES_DIR.mkdir(parents=True, exist_ok=True)

# 允许的图片格式
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'webp'}

# 数据文件路径
MODELS_FILE = DATA_DIR / 'models.json'
GPUS_FILE = DATA_DIR / 'gpus.json'
CPUS_FILE = DATA_DIR / 'cpus.json'
CATEGORIES_FILE = DATA_DIR / 'categories.json'
KNOWLEDGE_FILE = DATA_DIR / 'knowledge.json'
CONFIG_FILE = DATA_DIR / 'config.json'

# Flask secret key (每次启动随机生成，重启后需重新登录)
SECRET_KEY = secrets.token_hex(32)

# 默认管理员密码（首次使用，之后可从 config.json 修改）
DEFAULT_ADMIN_PASSWORD = 'admin123'

# 大模型配置（默认值）
LLM_CONFIG = {
    'base_url': 'http://192.168.2.17:19007/v1',
    'api_key': 'xxxx',
    'model': 'auto',
    'vision_model': 'gpt-4-vision-preview',
}

# 默认网站配置（包含LLM配置）
DEFAULT_CONFIG = {
    'site_name': 'ParamHub',
    'site_subtitle': '参数百科',
    'footer_text': 'ParamHub - AI大模型与硬件参数速查平台',
    'icp_number': '',
    'copyright_year': '2024',
    'contact_email': '',
    'github_url': '',
    'llm_base_url': 'http://192.168.2.17:19007/v1',
    'llm_api_key': '',
    'llm_model': 'auto',
    'llm_vision_model': 'gpt-4-vision-preview',
    'admin_password': DEFAULT_ADMIN_PASSWORD,
}