import sys, os

# Same logic as video.py
ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))
sys.path.append(ROOT_DIR)

print(f"ROOT_DIR: {ROOT_DIR}")
print(f"Looking for ai_models at: {os.path.join(ROOT_DIR, 'ai_models')}")
print(f"ai_models exists: {os.path.exists(os.path.join(ROOT_DIR, 'ai_models'))}")
print(f"assets exists: {os.path.exists(os.path.join(ROOT_DIR, 'ai_models/prism/assets'))}")

try:
    from ai_models.prism.feature_engine import PrismFeatureExtractor
    from ai_models.prism.router import PrismRouter

    prism_extractor = PrismFeatureExtractor()
    prism_router = PrismRouter(
        artifacts_dir=os.path.join(ROOT_DIR, "ai_models/prism/assets")
    )
    print("SUCCESS: AI models loaded OK")
except Exception as e:
    import traceback
    print(f"ERROR: {type(e).__name__}: {e}")
    traceback.print_exc()