a0a5cc4c60
Updated all relative imports to use absolute import paths for consistency and improved maintainability. Also replaced bare except clauses with 'except Exception' for better error handling, and made minor code cleanups such as removing unused variables and improving import order. No functional changes were made.
26 lines
836 B
Python
26 lines
836 B
Python
import logging
|
|
import os
|
|
import sys
|
|
|
|
repo_root = os.path.dirname(__file__)
|
|
src_path = os.path.join(repo_root, "src")
|
|
if src_path not in sys.path:
|
|
sys.path.insert(0, src_path)
|
|
|
|
from utils.document_processing import create_document_converter # noqa: E402
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
logger.info("Warming up docling models")
|
|
|
|
try:
|
|
# Use the sample document to warm up docling
|
|
test_file = "/app/warmup_ocr.pdf"
|
|
logger.info(f"Using test file to warm up docling: {test_file}")
|
|
converter = create_document_converter()
|
|
converter.convert(test_file)
|
|
logger.info("Docling models warmed up successfully")
|
|
except Exception as e:
|
|
logger.info(f"Docling warm-up completed with exception: {str(e)}")
|
|
# This is expected - we just want to trigger the model downloads
|