test_setup.py

#!/usr/bin/env python3
"""
Test script to verify the Memory Assistant setup.
Run this to check if all components are working correctly.
"""

import sys
from pathlib import Path
import logging

# Add the project root to Python path
sys.path.append(str(Path(__file__).parent))

def test_imports():
    """Test if all modules can be imported."""
    print("🔍 Testing imports...")
    
    try:
        from config.settings import settings
        print("✅ Settings module imported successfully")
    except Exception as e:
        print(f"❌ Failed to import settings: {e}")
        return False
    
    try:
        from capture.camera import Camera
        print("✅ Camera module imported successfully")
    except Exception as e:
        print(f"❌ Failed to import camera: {e}")
        return False
    
    try:
        from vision.detector import ObjectDetector
        print("✅ Object detector module imported successfully")
    except Exception as e:
        print(f"❌ Failed to import object detector: {e}")
        return False
    
    try:
        from memory.database import MemoryDatabase
        print("✅ Database module imported successfully")
    except Exception as e:
        print(f"❌ Failed to import database: {e}")
        return False
    
    try:
        from api.openai_client import OpenAIClient
        print("✅ OpenAI client module imported successfully")
    except Exception as e:
        print(f"❌ Failed to import OpenAI client: {e}")
        return False
    
    return True

def test_settings():
    """Test settings configuration."""
    print("\n🔧 Testing settings...")
    
    try:
        from config.settings import settings
        
        print(f"✅ Base directory: {settings.BASE_DIR}")
        print(f"✅ Database path: {settings.DATABASE_PATH}")
        print(f"✅ Images path: {settings.IMAGES_PATH}")
        print(f"✅ Capture interval: {settings.CAPTURE_INTERVAL} seconds")
        
        # Test directory creation
        settings.create_directories()
        print("✅ Directories created successfully")
        
        return True
        
    except Exception as e:
        print(f"❌ Settings test failed: {e}")
        return False

def test_camera():
    """Test camera functionality."""
    print("\n📸 Testing camera...")
    
    try:
        from capture.camera import Camera
        
        camera = Camera()
        print("✅ Camera object created")
        
        # Test initialization
        if camera.initialize():
            print("✅ Camera initialized successfully")
            camera.release()
            return True
        else:
            print("⚠️ Camera initialization failed (this is normal if no webcam is available)")
            return True  # Don't fail the test for camera issues
            
    except Exception as e:
        print(f"❌ Camera test failed: {e}")
        return False

def test_database():
    """Test database functionality."""
    print("\n🗄️ Testing database...")
    
    try:
        from memory.database import MemoryDatabase
        
        db = MemoryDatabase()
        print("✅ Database initialized successfully")
        
        # Test statistics
        stats = db.get_statistics()
        print(f"✅ Database statistics: {stats}")
        
        return True
        
    except Exception as e:
        print(f"❌ Database test failed: {e}")
        return False

def test_openai():
    """Test OpenAI API connection."""
    print("\n🤖 Testing OpenAI API...")
    
    try:
        from api.openai_client import OpenAIClient
        
        client = OpenAIClient()
        print("✅ OpenAI client created")
        
        # Test connection
        if client.validate_api_connection():
            print("✅ OpenAI API connection successful")
            return True
        else:
            print("⚠️ OpenAI API connection failed (check your API key)")
            return False
            
    except Exception as e:
        print(f"❌ OpenAI test failed: {e}")
        return False

def main():
    """Run all tests."""
    print("🧠 Memory Assistant Setup Test")
    print("=" * 40)
    
    tests = [
        ("Imports", test_imports),
        ("Settings", test_settings),
        ("Camera", test_camera),
        ("Database", test_database),
        ("OpenAI", test_openai)
    ]
    
    passed = 0
    total = len(tests)
    
    for test_name, test_func in tests:
        try:
            if test_func():
                passed += 1
        except Exception as e:
            print(f"❌ {test_name} test crashed: {e}")
    
    print("\n" + "=" * 40)
    print(f"📊 Test Results: {passed}/{total} tests passed")
    
    if passed == total:
        print("🎉 All tests passed! Your Memory Assistant is ready to use.")
        print("\n🚀 To start the application, run:")
        print("   streamlit run app.py")
    else:
        print("⚠️ Some tests failed. Please check the errors above.")
        print("\n💡 Common issues:")
        print("   - Make sure you have a .env file with your OpenAI API key")
        print("   - Install all dependencies: pip install -r requirements.txt")
        print("   - Ensure you have a webcam available for camera tests")

if __name__ == "__main__":
    main()