models/__init__.py

"""
ShortSmith v2 - Models Package

AI model wrappers for:
- Visual analysis (Qwen2-VL)
- Audio analysis (Librosa + Wav2Vec 2.0)
- Face recognition (InsightFace)
- Body recognition (OSNet)
- Motion detection (RAFT)
- Object tracking (ByteTrack)
"""

from models.audio_analyzer import AudioAnalyzer, AudioFeatures, AudioSegmentScore
from models.visual_analyzer import VisualAnalyzer, VisualFeatures
from models.face_recognizer import FaceRecognizer, FaceDetection, FaceMatch
from models.body_recognizer import BodyRecognizer, BodyDetection
from models.motion_detector import MotionDetector, MotionScore
from models.tracker import ObjectTracker, TrackedObject

__all__ = [
    "AudioAnalyzer",
    "AudioFeatures",
    "AudioSegmentScore",
    "VisualAnalyzer",
    "VisualFeatures",
    "FaceRecognizer",
    "FaceDetection",
    "FaceMatch",
    "BodyRecognizer",
    "BodyDetection",
    "MotionDetector",
    "MotionScore",
    "ObjectTracker",
    "TrackedObject",
]