# -*- coding: utf-8 -*-
"""
全局配置文件
集中管理 API Key、文件路径、模型参数等
"""
import os
from pathlib import Path

# ============================================================
# 项目根目录
# ============================================================
PROJECT_ROOT = Path(__file__).parent.resolve()
DATA_DIR = PROJECT_ROOT / "data"

# ============================================================
# DeepSeek API 配置（兼容 OpenAI 接口协议）
# ============================================================
DEEPSEEK_API_KEY = os.environ.get("DEEPSEEK_API_KEY", "")
DEEPSEEK_BASE_URL = "https://api.deepseek.com/v1"
DEEPSEEK_MODEL = "deepseek-chat"

# 备用1：火山引擎方舟（豆包大模型 doubao-seed-2-1-pro-260628）
VOLCENGINE_API_KEY = os.environ.get("VOLCENGINE_API_KEY", os.environ.get("ARK_API_KEY", "836d9bc0-80e8-4e45-90df-7287994d91ec"))
VOLCENGINE_BASE_URL = os.environ.get("VOLCENGINE_BASE_URL", "https://ark.cn-beijing.volces.com/api/v3")
VOLCENGINE_MODEL = os.environ.get("VOLCENGINE_MODEL", "doubao-seed-2-0-lite-260428")

# 备用2：如果用户配置了其他兼容 OpenAI 协议的 API
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "")
OPENAI_BASE_URL = os.environ.get("OPENAI_BASE_URL", "https://api.openai.com/v1")
OPENAI_MODEL = os.environ.get("OPENAI_MODEL", "gpt-4o-mini")

# ============================================================
# LLM 调用参数
# ============================================================
LLM_TEMPERATURE = 0.1          # 低温度保证输出稳定
LLM_MAX_TOKENS = 4096
LLM_TIMEOUT = 60               # 超时秒数
LLM_MAX_RETRIES = 2            # 最大重试次数

# ============================================================
# 风险评估权重配置
# ============================================================
RISK_WEIGHTS = {
    "tech_disruption":       0.20,   # 技术路线颠覆风险
    "talent_loss":           0.15,   # 核心人员流失风险
    "algo_compliance":       0.15,   # 算法/数据合规风险
    "geopolitical":          0.20,   # 地缘政治/出口管制风险
    "rd_capitalization":     0.15,   # 研发资本化操纵风险
    "concentration":         0.15,   # 客户/供应商集中风险
}

# ============================================================
# 保险定价参数
# ============================================================
INSURANCE_PRODUCTS = {
    "ip_lawsuit": {
        "name": "知识产权被诉险",
        "base_premium": 50000,     # 基础保费（元）
        "base_coverage": 5000000,  # 基础保额（元）
        "description": "覆盖因知识产权纠纷（专利、商标、著作权）产生的诉讼费用及赔偿金",
    },
    "exec_departure": {
        "name": "高管离职业务中断险",
        "base_premium": 80000,
        "base_coverage": 10000000,
        "description": "覆盖核心技术人员/高管离职导致的业务中断损失",
    },
    "data_compliance": {
        "name": "数据合规行政处罚险",
        "base_premium": 30000,
        "base_coverage": 3000000,
        "description": "覆盖因非主观恶意的数据合规违规产生的行政罚款",
    },
}

# ============================================================
# Streamlit 页面配置
# ============================================================
PAGE_TITLE = "🛡️ 科创企业智能风控与核保系统"
PAGE_ICON = "🛡️"
LAYOUT = "wide"