import json import logging logger = logging.getLogger(__name__) DEFAULT_CONFIG = { "output_format": "PNG", # PNG or JPG "output_quality": 95, # JPG quality 1-95 "output_dir": "", # empty = use app output/ dir "last_garment_dir": "", "last_print_dir": "", "last_template": "", # name of the last-selected template "last_batch_mode": "full_combo", # BatchMode value of the last-used mode "update_source": "", # HTTP(S) manifest source (empty = no update check) "update_user": "", # HTTP Basic Auth user (empty = anonymous) "update_pass": "", # HTTP Basic Auth password (empty = anonymous) # AI 穿搭(docs/11 §11):上次路径与批量设置并入 app_config(密钥不在此,见 ai_models.json) "outfit_excel": "", "outfit_output_dir": "", "outfit_model": "", # last-selected model name "outfit_concurrency": 1, "outfit_request_interval": 2.0, "outfit_task_cooldown": 1.0, "outfit_retry_count": 2, "outfit_resolution": "1K", "outfit_quality": "均衡", "outfit_retry_failed": False, } _CONFIG_FILENAME = "app_config.json" _AI_MODELS_FILENAME = "ai_models.json" _OUTFIT_PROMPT_FILENAME = "outfit_prompt.txt" # Default outfit prompt (docs/11 §7). Persisted to outfit_prompt.txt on first save. DEFAULT_OUTFIT_PROMPT = ( "为商品「{title}」生成人物上身实穿效果图:真人模特正面穿着这件衣服," "完整保留款式、版型、颜色与印花图案,自然光、纯色棚拍背景," "电商主图风格,不加文字与促销标签。" ) def load_config(): """ 从 JSON 文件加载应用配置。 返回默认配置与文件配置的合并字典。如果配置文件不存在、损坏或无法读取, 函数不会抛出异常,而是记录日志并返回默认配置。 Returns: dict: 合并后的配置字典 """ from services.file_service import get_config_path config_file = get_config_path(_CONFIG_FILENAME) if not config_file.exists(): logger.info("Config file not found, using defaults: %s", config_file) return dict(DEFAULT_CONFIG) try: with open(str(config_file), encoding="utf-8") as f: data = json.load(f) if not isinstance(data, dict): raise ValueError("Config root is not a JSON object") merged = dict(DEFAULT_CONFIG) merged.update(data) logger.info("Config loaded from %s", config_file) return merged except (json.JSONDecodeError, ValueError) as exc: logger.warning("Config file damaged (%s), using defaults: %s", exc, config_file) return dict(DEFAULT_CONFIG) except OSError as exc: logger.warning("Config file unreadable (%s), using defaults: %s", exc, config_file) return dict(DEFAULT_CONFIG) def save_config(data): """Save config dict to JSON. Logs error on failure, does not raise.""" from services.file_service import get_config_path config_file = get_config_path(_CONFIG_FILENAME) try: config_file.parent.mkdir(parents=True, exist_ok=True) with open(str(config_file), "w", encoding="utf-8") as f: json.dump(data, f, ensure_ascii=False, indent=2) logger.info("Config saved to %s", config_file) except OSError as exc: logger.error("Failed to save config to %s: %s", config_file, exc) def load_ai_models(): """Load AI model configs from ai_models.json. Returns a list of dicts (each with a ``name`` plus AiModelConfig fields). Missing/corrupt file → empty list (the UI then prompts the admin to add one). Keys live only in ~/.cmbot and are never committed (docs/11 §11). """ from services.file_service import get_config_path models_file = get_config_path(_AI_MODELS_FILENAME) if not models_file.exists(): logger.info("AI models file not found: %s", models_file) return [] try: with open(str(models_file), encoding="utf-8-sig") as f: data = json.load(f) except (json.JSONDecodeError, ValueError, OSError) as exc: logger.warning("AI models file unreadable (%s): %s", exc, models_file) return [] models = data.get("models") if isinstance(data, dict) else data if not isinstance(models, list): logger.warning("AI models file has no model list: %s", models_file) return [] return [m for m in models if isinstance(m, dict)] def load_outfit_prompt(): """Return the saved outfit prompt template, or the built-in default.""" from services.file_service import get_config_path prompt_file = get_config_path(_OUTFIT_PROMPT_FILENAME) if not prompt_file.exists(): return DEFAULT_OUTFIT_PROMPT try: with open(str(prompt_file), encoding="utf-8-sig") as f: return f.read() except OSError as exc: logger.warning("Outfit prompt unreadable (%s): %s", exc, prompt_file) return DEFAULT_OUTFIT_PROMPT def save_outfit_prompt(text): """Persist the outfit prompt template (utf-8, no BOM). Does not raise.""" from services.file_service import get_config_path prompt_file = get_config_path(_OUTFIT_PROMPT_FILENAME) try: prompt_file.parent.mkdir(parents=True, exist_ok=True) with open(str(prompt_file), "w", encoding="utf-8") as f: f.write(text) logger.info("Outfit prompt saved to %s", prompt_file) except OSError as exc: logger.error("Failed to save outfit prompt to %s: %s", prompt_file, exc)