Configure EasyOCR model storage with enhanced error handling and logging

- Set up a writable directory for EasyOCR model storage, with a primary path in a temporary directory and fallbacks to /tmp and the current working directory.
- Improved logging to provide feedback on the model directory creation process and any issues encountered.

Dockerfile

@@ -10,9 +10,10 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # Create necessary directories with proper permissions
-RUN mkdir -p /app/.streamlit /tmp/docling_temp && \
+RUN mkdir -p /app/.streamlit /tmp/docling_temp /tmp/easyocr_models && \
     chmod 755 /app/.streamlit && \
-    chmod 777 /tmp/docling_temp
+    chmod 777 /tmp/docling_temp && \
+    chmod 777 /tmp/easyocr_models
 
 COPY requirements.txt ./
 COPY src/ ./src/

src/streamlit_app.py

@@ -63,10 +63,30 @@ os.environ['STREAMLIT_BROWSER_GATHER_USAGE_STATS'] = 'false'
 os.environ['STREAMLIT_SERVER_ENABLE_CORS'] = 'false'
 os.environ['STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION'] = 'false'
 
+# Configure EasyOCR to use writable directory for model storage
+os.environ['EASYOCR_MODULE_PATH'] = os.path.join(TEMP_DIR, 'easyocr_models')
+try:
+    os.makedirs(os.environ['EASYOCR_MODULE_PATH'], exist_ok=True)
+    logging.info(f"EasyOCR model directory: {os.environ['EASYOCR_MODULE_PATH']}")
+except Exception as e:
+    logging.warning(f"Could not create EasyOCR model directory: {e}")
+    # Fallback to /tmp if the temp directory fails
+    os.environ['EASYOCR_MODULE_PATH'] = '/tmp/easyocr_models'
+    try:
+        os.makedirs(os.environ['EASYOCR_MODULE_PATH'], exist_ok=True)
+        logging.info(f"Using fallback EasyOCR model directory: {os.environ['EASYOCR_MODULE_PATH']}")
+    except Exception as e2:
+        logging.error(f"Failed to create EasyOCR model directory: {e2}")
+        # Last resort: use current directory
+        os.environ['EASYOCR_MODULE_PATH'] = os.path.join(os.getcwd(), 'easyocr_models')
+        os.makedirs(os.environ['EASYOCR_MODULE_PATH'], exist_ok=True)
+        logging.warning(f"Using current directory for EasyOCR models: {os.environ['EASYOCR_MODULE_PATH']}")
+
 # Log startup information
 logging.info("=" * 50)
 logging.info("Docling Streamlit App Starting")
 logging.info(f"Temp directory: {TEMP_DIR}")
+logging.info(f"EasyOCR model directory: {os.environ.get('EASYOCR_MODULE_PATH', 'NOT_SET')}")
 logging.info(f"Current working directory: {os.getcwd()}")
 logging.info(f"Python version: {sys.version}")
 logging.info("=" * 50)