import cv2
import numpy as np
import os
import sys
from sign_utils import MediaPipeLandmarksModel, SignLanguagePredictor

def test_camera():
    """Test if camera is working."""
    print("Testing camera...")
    cap = cv2.VideoCapture(0)
    
    if not cap.isOpened():
        print("❌ Camera not accessible")
        return False
    
    ret, frame = cap.read()
    if not ret:
        print("❌ Cannot read from camera")
        cap.release()
        return False
    
    print("✅ Camera working")
    cap.release()
    return True

def test_mediapipe():
    """Test MediaPipe hand detection."""
    print("Testing MediaPipe...")
    try:
        landmark_model = MediaPipeLandmarksModel()
        
        # Create a dummy image
        dummy_image = np.zeros((480, 640, 3), dtype=np.uint8)
        landmarks = landmark_model.predict_landmarks(dummy_image)
        
        print("✅ MediaPipe working")
        return True
    except Exception as e:
        print(f"❌ MediaPipe error: {e}")
        return False

def test_model_files():
    """Test if model files exist."""
    print("Testing model files...")
    
    model_path = "models/model.tflite"
    labels_path = "models/labels.txt"
    
    if not os.path.exists(model_path):
        print(f"❌ Model file not found: {model_path}")
        return False
    
    if not os.path.exists(labels_path):
        print(f"❌ Labels file not found: {labels_path}")
        return False
    
    print("✅ Model files found")
    return True

def test_predictor():
    """Test the sign language predictor."""
    print("Testing predictor...")
    
    if not test_model_files():
        return False
    
    try:
        predictor = SignLanguagePredictor(
            model_path="models/model.tflite",
            labels_path="models/labels.txt"
        )
        
        # Create a dummy image
        dummy_image = np.zeros((480, 640, 3), dtype=np.uint8)
        prediction, confidence = predictor.predict(dummy_image)
        
        print(f"✅ Predictor working - Sample prediction: {prediction} ({confidence:.2f})")
        return True
    except Exception as e:
        print(f"❌ Predictor error: {e}")
        return False

def test_dependencies():
    """Test if all dependencies are installed."""
    print("Testing dependencies...")
    
    required_modules = [
        'cv2', 'numpy', 'tensorflow', 'mediapipe', 
        'streamlit', 'sklearn'
    ]
    
    missing_modules = []
    
    for module in required_modules:
        try:
            __import__(module)
        except ImportError:
            missing_modules.append(module)
    
    if missing_modules:
        print(f"❌ Missing modules: {missing_modules}")
        return False
    
    print("✅ All dependencies installed")
    return True

def run_full_test():
    """Run complete system test."""
    print("🤟 Sign Language Interpreter - System Test")
    print("=" * 50)
    
    tests = [
        ("Dependencies", test_dependencies),
        ("Camera", test_camera),
        ("MediaPipe", test_mediapipe),
        ("Model Files", test_model_files),
        ("Predictor", test_predictor)
    ]
    
    passed = 0
    total = len(tests)
    
    for test_name, test_func in tests:
        print(f"\n{test_name}:")
        if test_func():
            passed += 1
        else:
            print(f"  Fix this issue before proceeding")
    
    print(f"\n{'='*50}")
    print(f"Test Results: {passed}/{total} tests passed")
    
    if passed == total:
        print("🎉 All tests passed! System is ready to use.")
        print("\nNext steps:")
        print("1. Run 'streamlit run app.py' to start the application")
        print("2. Or run 'python data_collection.py' to collect custom data")
    else:
        print("❌ Some tests failed. Please fix the issues above.")
        
        if not test_model_files():
            print("\n💡 Tip: Run 'python create_sample_model.py' to create a sample model")

def main():
    if len(sys.argv) > 1 and sys.argv[1] == "--quick":
        # Quick test - just check if basic components work
        if test_dependencies() and test_mediapipe():
            print("✅ Quick test passed!")
        else:
            print("❌ Quick test failed!")
    else:
        run_full_test()

if __name__ == "__main__":
    main()