58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
"""Optional OCR for cropped image/table regions."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import io
|
||
from pathlib import Path
|
||
|
||
import fitz
|
||
|
||
|
||
def ocr_pixmap(pix: fitz.Pixmap) -> str:
|
||
"""Run OCR on a pixmap; returns empty string when OCR is unavailable."""
|
||
try:
|
||
import pytesseract
|
||
from PIL import Image
|
||
except ImportError:
|
||
return ""
|
||
|
||
try:
|
||
image = Image.open(io.BytesIO(pix.tobytes("png")))
|
||
return _ocr_pil(image)
|
||
except Exception:
|
||
return ""
|
||
|
||
|
||
def ocr_image_file(path: Path) -> str:
|
||
try:
|
||
import pytesseract # noqa: F401
|
||
from PIL import Image
|
||
except ImportError:
|
||
return ""
|
||
|
||
try:
|
||
return _ocr_pil(Image.open(path))
|
||
except Exception:
|
||
return ""
|
||
|
||
|
||
def _ocr_pil(image) -> str:
|
||
import pytesseract
|
||
|
||
for lang in ("chi_tra+eng", "chi_sim+eng", "eng"):
|
||
try:
|
||
text = pytesseract.image_to_string(image, lang=lang)
|
||
cleaned = " ".join(text.split())
|
||
if cleaned:
|
||
return cleaned
|
||
except Exception:
|
||
continue
|
||
return ""
|
||
|
||
|
||
def describe_visual(ocr_text: str, kind: str, label: str) -> str:
|
||
"""Lightweight image/table description without an external vision model."""
|
||
if ocr_text:
|
||
return f"{kind}：{ocr_text[:300]}"
|
||
return f"{kind}：{label}"
|