import sys
import os

# Add the LightRAG-main directory to path
lightrag_dir = os.path.join(os.path.dirname(os.path.abspath('.')), 'LightRAG-main')
if lightrag_dir not in sys.path:
    sys.path.insert(0, lightrag_dir)

# Test the document processor imports
try:
    from lightrag.document_processor import get_document_processor
    print('✅ Document processor import successful')
    
    # Test OCR processor import
    from simple_ocr_processor import get_simple_ocr_processor
    print('✅ Simple OCR processor import successful')
    
    # Test image classifier import
    from fast_image_classifier import FastImageClassifier
    print('✅ Fast image classifier import successful')
    
    print('🎉 All Web UI imports are working correctly!')
    
    # Test document processor initialization
    try:
        processor = get_document_processor()
        print('✅ Document processor initialization successful')
        
        # Check if OCR processor is available
        if processor.ocr_processor and processor.ocr_processor.ocr_available:
            print('✅ OCR processor is available and ready')
        else:
            print('❌ OCR processor not available')
            
        # Check if image classifier is available
        if processor.image_classifier and processor.image_classifier.available:
            print('✅ Image classifier is available and ready')
        else:
            print('❌ Image classifier not available')
            
    except Exception as e:
        print(f'❌ Document processor initialization failed: {e}')
        import traceback
        traceback.print_exc()
    
except Exception as e:
    print(f'❌ Import error: {e}')
    import traceback
    traceback.print_exc()