core/__init__.py

"""
Core processing modules for BackgroundFX Pro.
Handles main processing pipeline components.
"""

from .temporal import (
    TemporalStabilizer,
    TemporalConfig,
    FrameBuffer,
    FrameAnomalyDetector,
    OpticalFlowTracker
)

# Only import what actually exists in the new models.py
from .models import ModelManager

# Create dummy classes for backward compatibility
# These are referenced elsewhere but don't exist in the new models.py
SAM2Model = None
MatAnyoneModel = None
ModelConfig = None
ModelCache = None

from .quality import (
    QualityAnalyzer,
    QualityMetrics,
    QualityConfig,
    MetricsTracker,
    QualityReport
)

from .edge import (
    EdgeProcessor,
    EdgeConfig,
    HairSegmentation,
    EdgeRefinement,
    SymmetryCorrector
)

from .hair_segmentation import (
    HairSegmentationPipeline,
    HairConfig,
    HairMaskRefiner,
    AsymmetryDetector,
    HairEdgeEnhancer
)

__version__ = "1.0.0"

__all__ = [
    # Temporal
    'TemporalStabilizer',
    'TemporalConfig',
    'FrameBuffer',
    'FrameAnomalyDetector',
    'OpticalFlowTracker',
    
    # Models
    'ModelManager',
    'SAM2Model',
    'MatAnyoneModel',
    'ModelConfig',
    'ModelCache',
    
    # Quality
    'QualityAnalyzer',
    'QualityMetrics',
    'QualityConfig',
    'MetricsTracker',
    'QualityReport',
    
    # Edge
    'EdgeProcessor',
    'EdgeConfig',
    'HairSegmentation',
    'EdgeRefinement',
    'SymmetryCorrector',
    
    # Hair Segmentation
    'HairSegmentationPipeline',
    'HairConfig',
    'HairMaskRefiner',
    'AsymmetryDetector',
    'HairEdgeEnhancer'
]