slahiriandClaude Opus 4.5
7f1186d412
New Features: - SID_PromptCompose node with Standard and Enhance with AI modes - SID_PromptSynthesis node for VLM-based prompt generation - SID_ImageAnalysis node for comprehensive image analysis - Text LLM support (Qwen2.5 0.5B-7B) for prompt enhancement - PixAI tagger integration Documentation: - docs/image_tokenization_spec.md - Full technical specification for tokenization → classification → assembly pipeline - docs/implementation_plan.md - 8-phase implementation roadmap - docs/SID_PromptCompose.md - Node usage documentation Infrastructure: - core/compose/ module with base, pipeline, standard, llm generators - core/log.py centralized logging - Config files for Qwen2.5 text models (YAML) - Web assets for ComfyUI integration This commit establishes the foundation for rewriting the compose pipeline based on the tokenization specification. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
"""Configuration loader for SID Photography Toolkit."""
|
|
|
|
import json
|
|
from functools import lru_cache
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import yaml
|
|
|
|
# Config directory path
|
|
_CONFIG_DIR = Path(__file__).parent.parent / "config"
|
|
|
|
|
|
@lru_cache(maxsize=16)
|
|
def load_config(name: str) -> dict[str, Any]:
|
|
"""
|
|
Load a config file by name (supports JSON and YAML).
|
|
|
|
Args:
|
|
name: Config name (without extension)
|
|
|
|
Returns:
|
|
Parsed config as dict
|
|
|
|
Raises:
|
|
FileNotFoundError: If config file doesn't exist
|
|
"""
|
|
# Try JSON first, then YAML
|
|
json_path = _CONFIG_DIR / f"{name}.json"
|
|
yaml_path = _CONFIG_DIR / f"{name}.yaml"
|
|
|
|
if json_path.exists():
|
|
with open(json_path, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
elif yaml_path.exists():
|
|
with open(yaml_path, "r", encoding="utf-8") as f:
|
|
return yaml.safe_load(f)
|
|
else:
|
|
raise FileNotFoundError(f"Config not found: {name} (tried .json and .yaml)")
|
|
|
|
|
|
def get_model_config(model_name: str) -> dict[str, Any]:
|
|
"""Get configuration for a specific model."""
|
|
return load_config(model_name)
|
|
|
|
|
|
def get_prompt(model_name: str, mode: str) -> str:
|
|
"""
|
|
Get prompt for a model and mode.
|
|
|
|
Args:
|
|
model_name: Model name (e.g., "florence")
|
|
mode: Caption mode (e.g., "detailed", "tags")
|
|
|
|
Returns:
|
|
Prompt string
|
|
"""
|
|
config = load_config(model_name)
|
|
prompts = config.get("prompts", {})
|
|
return prompts.get(mode, prompts.get("detailed", ""))
|
|
|
|
|
|
def reload_configs() -> None:
|
|
"""Clear config cache and reload from disk."""
|
|
load_config.cache_clear()
|