From 430cdf32cf07d883ea6be83405f9546af89cd84a Mon Sep 17 00:00:00 2001
From: MangoPig <Ronaldng1588@gmail.com>
Date: Sat, 21 Feb 2026 23:04:39 +0000
Subject: [PATCH] feat(docling): add Chinese OCR support (simplified +
 traditional)

- Add entrypoint script to auto-download OCR models on first run
- Include ch_sim (Simplified) and ch_tra (Traditional) for HK law firm use
- Add docling_models volume for model persistence
- Bump start_period to 120s for first-run download time
---
 Code/docker-compose.yaml | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/Code/docker-compose.yaml b/Code/docker-compose.yaml
index f479a26..c426229 100644
--- a/Code/docker-compose.yaml
+++ b/Code/docker-compose.yaml
@@ -3,20 +3,34 @@ services:
     docling:
         # https://github.com/docling-project/docling-serve
         # 5001: Web UI + API
+        # OCR: EasyOCR with English + Chinese Simplified + Traditional (auto-downloaded on first run)
 
         image: ghcr.io/docling-project/docling-serve-cpu:v1.13.0
         container_name: jt-docling
         restart: unless-stopped
         environment:
             DOCLING_SERVE_ENABLE_UI: "true"
+        volumes:
+            - docling_models:/opt/app-root/src/.cache/docling/models
         ports:
             - "5001:5001"
+        entrypoint: ["/bin/sh", "-c"]
+        command:
+            - |
+              # Download Chinese OCR models if missing (first run only)
+              if [ ! -f /opt/app-root/src/.cache/docling/models/EasyOcr/zh_sim_g2.pth ] || [ ! -f /opt/app-root/src/.cache/docling/models/EasyOcr/zh_tra_g2.pth ]; then
+                echo "Downloading Chinese OCR models (Simplified + Traditional)..."
+                python3 -c "import easyocr; easyocr.Reader(['en','ch_sim','ch_tra'], gpu=False)"
+                # Copy downloaded models to correct location
+                cp -n /opt/app-root/src/.EasyOCR/model/*.pth /opt/app-root/src/.cache/docling/models/EasyOcr/ 2>/dev/null || true
+              fi
+              exec uvicorn docling_serve.app:app --host 0.0.0.0 --port 5001
         healthcheck:
             test: ["CMD", "curl", "-sf", "http://localhost:5001/health"]
             interval: 30s
             timeout: 10s
             retries: 3
-            start_period: 60s
+            start_period: 120s
 
     # Ollama
     ollama:
@@ -55,5 +69,7 @@ services:
     #         retries: 3
 
 volumes:
+    docling_models:
+        name: jt-docling-models
     ollama_data:
         name: jt-ollama-data