- Add LLM enhancement modes: None, Local LLM, API (Premium) - Add generation modes: 1-Shot and Multi-Pass - Local LLM: Polish assembled prompt with Qwen text models - API: Deep analysis with Claude/GPT/Gemini + image support - Multi-Pass: Generate each category step by step - Add word composition analysis comparing tags vs prompts - Visual coverage report with category breakdown - Track words gained from enhancement and missing words - Store analysis stats in metadata 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
73 lines
1.7 KiB
Python
73 lines
1.7 KiB
Python
"""Prompt composition module."""
|
|
|
|
from .base import (
|
|
ComposeMode,
|
|
ComposeConfig,
|
|
ComposeResult,
|
|
ComposeAnalytics,
|
|
StyleConfig,
|
|
)
|
|
from .standard import StandardGenerator
|
|
from .llm import LLMGenerator
|
|
from .pipeline import ComposePipeline
|
|
from .canonical_structurer import (
|
|
CanonicalStructurer,
|
|
CanonicalClassifier,
|
|
NLPTagMatcher,
|
|
get_nlp_matcher,
|
|
normalize_tag,
|
|
)
|
|
from .enhanced_categories import (
|
|
Category as EnhancedCategory,
|
|
CATEGORY_KEYWORDS as ENHANCED_CATEGORY_KEYWORDS,
|
|
CATEGORY_ORDER as ENHANCED_CATEGORY_ORDER,
|
|
CATEGORY_PATTERNS,
|
|
TAG_SYNONYMS,
|
|
EnhancedCategoryClassifier,
|
|
get_classifier as get_enhanced_classifier,
|
|
classify_tags,
|
|
get_uncategorized_tags,
|
|
)
|
|
from .zimage_assembler import (
|
|
ZImageAssembler,
|
|
assemble_zimage_prompt,
|
|
CATEGORY_ORDER as ZIMAGE_CATEGORY_ORDER,
|
|
CATEGORY_HEADERS as ZIMAGE_CATEGORY_HEADERS,
|
|
)
|
|
|
|
__all__ = [
|
|
# Enums and configs
|
|
"ComposeMode",
|
|
"ComposeConfig",
|
|
"ComposeResult",
|
|
"ComposeAnalytics",
|
|
"StyleConfig",
|
|
# Generators
|
|
"StandardGenerator",
|
|
"LLMGenerator",
|
|
# Pipeline
|
|
"ComposePipeline",
|
|
# Canonical structurer
|
|
"CanonicalStructurer",
|
|
"CanonicalClassifier",
|
|
# NLP tag matching
|
|
"NLPTagMatcher",
|
|
"get_nlp_matcher",
|
|
"normalize_tag",
|
|
# Enhanced category system
|
|
"EnhancedCategory",
|
|
"ENHANCED_CATEGORY_KEYWORDS",
|
|
"ENHANCED_CATEGORY_ORDER",
|
|
"CATEGORY_PATTERNS",
|
|
"TAG_SYNONYMS",
|
|
"EnhancedCategoryClassifier",
|
|
"get_enhanced_classifier",
|
|
"classify_tags",
|
|
"get_uncategorized_tags",
|
|
# Z-Image assembler
|
|
"ZImageAssembler",
|
|
"assemble_zimage_prompt",
|
|
"ZIMAGE_CATEGORY_ORDER",
|
|
"ZIMAGE_CATEGORY_HEADERS",
|
|
]
|