backend/test_training.py

import os
import sys

# When running as a module 'backend.test_training', we can use relative imports
try:
    from .model_manager import EmotionClassifier
except ImportError:
    # Fallback if run as script, but this will break relative imports in model_manager
    # So we must fix path to import 'backend.model_manager'
    sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
    from backend.model_manager import EmotionClassifier

def test_training():
    print("Initializing Classifier...")
    classifier = EmotionClassifier()
    
    # Correct data path assuming we are in backend/
    base_dir = os.path.dirname(os.path.abspath(__file__))
    data_dir = os.path.join(base_dir, "../data sets")
    
    print(f"Data directory: {data_dir}")
    
    def log_callback(msg):
        print(f"[TRAIN] {msg}")
        
    print("Starting training test...")
    try:
        classifier.train_model(data_dir, log_callback=log_callback)
        print("Training test passed!")
    except Exception as e:
        print(f"Training test failed: {e}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    test_training()