diff --git a/config.py b/config.py new file mode 100644 index 0000000..d2011e1 --- /dev/null +++ b/config.py @@ -0,0 +1,57 @@ +""" +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, +}