49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
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() |