test_setup.py

from PIL import Image
import torch
import sys
sys.path.insert(0, 'src')

from models.face_model import FaceRecognitionModel
from models.signature_model import SignatureModel
from models.fusion import MultimodalFusion

print("="*60)
print("TESTING YOUR MULTIMODAL BIOMETRICS SETUP")
print("="*60)

# Test 1: Face model
print("\n1. Testing Face Model (facenet_pytorch)...")
try:
    face_model = FaceRecognitionModel(device='cpu')
    print("   ✓ Face model loaded successfully")
except Exception as e:
    print(f"   ✗ Error: {e}")
    sys.exit(1)

# Test 2: Signature model
print("\n2. Testing Signature Model...")
try:
    sig_model = SignatureModel('models/checkpoints/siamese_signature_model.pth', device='cpu')
    print("   ✓ Signature model loaded successfully")
except Exception as e:
    print(f"   ⚠ Warning: {e}")

# Test 3: Fusion
print("\n3. Testing Multimodal Fusion...")
try:
    fusion = MultimodalFusion(face_model, sig_model)
    print("   ✓ Fusion initialized successfully")
except Exception as e:
    print(f"   ✗ Error: {e}")
    sys.exit(1)

print("\n" + "="*60)
print("✓ ALL SYSTEMS READY!")
print("="*60)
print("\nNext steps:")
print("1. Place test images in: data/raw/faces/ and data/raw/signatures/")
print("2. Run: python verify_sample.py")
print("3. Or run: python backend/main.py")