docs: apply fleet-template (16-artifact scaffold)
Adds missing standard artifacts: - README.md (if missing) - AGENTS.md (AI agent contract) - PLAN.md (current sprint) - STATUS.md (where we are) - DEVELOPMENT.md (dev workflow) - DEPLOYMENT.md (deploy procedure) - TESTING.md (test strategy) - DECISIONS.md (ADR index + templates) - .github/CODEOWNERS - .github/workflows/ci.yml Preserves all existing artifacts. Refs: RugMunchMedia/fleet-template
This commit is contained in:
commit
47ba268131
310 changed files with 38429 additions and 0 deletions
77
ocr_extractor.py
Normal file
77
ocr_extractor.py
Normal file
|
|
@ -0,0 +1,77 @@
|
|||
"""Pry — Image OCR using Tesseract.
|
||||
Extract text from images on web pages. Uses pytesseract + Pillow."""
|
||||
|
||||
import io
|
||||
import logging
|
||||
import os
|
||||
from typing import Any, ClassVar
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
try:
|
||||
import pytesseract
|
||||
from PIL import Image
|
||||
|
||||
_has_tesseract = True
|
||||
except ImportError:
|
||||
_has_tesseract = False
|
||||
|
||||
|
||||
class ImageOCR:
|
||||
"""Extract text from images using Tesseract OCR."""
|
||||
|
||||
SUPPORTED_LANGUAGES: ClassVar[list[str]] = [
|
||||
"eng", "chi_sim", "chi_tra", "spa", "fra", "deu", "ita",
|
||||
"por", "rus", "jpn", "kor", "ara",
|
||||
]
|
||||
|
||||
def __init__(self, language: str = "eng"):
|
||||
self.language = language if language in self.SUPPORTED_LANGUAGES else "eng"
|
||||
|
||||
def extract_from_bytes(self, image_bytes: bytes, config: str = "") -> dict[str, Any]:
|
||||
"""Extract text from image bytes."""
|
||||
if not _has_tesseract:
|
||||
return {
|
||||
"success": False,
|
||||
"error": "pytesseract not installed. Run: pip install pytesseract pillow",
|
||||
}
|
||||
try:
|
||||
img = Image.open(io.BytesIO(image_bytes))
|
||||
text = pytesseract.image_to_string(img, lang=self.language, config=config)
|
||||
data = pytesseract.image_to_data(
|
||||
img, lang=self.language, output_type=pytesseract.Output.DICT
|
||||
)
|
||||
confidences = [c for c in data["conf"] if isinstance(c, int) and 0 <= c <= 100]
|
||||
avg_confidence = round(sum(confidences) / len(confidences), 1) if confidences else 0
|
||||
return {
|
||||
"success": True,
|
||||
"text": text.strip(),
|
||||
"confidence": avg_confidence,
|
||||
"word_count": len(text.split()),
|
||||
"language": self.language,
|
||||
}
|
||||
except Exception as e:
|
||||
return {"success": False, "error": str(e)[:200]}
|
||||
|
||||
def extract_from_file(self, image_path: str) -> dict[str, Any]:
|
||||
"""Extract text from an image file."""
|
||||
if not os.path.exists(image_path):
|
||||
return {"success": False, "error": f"File not found: {image_path}"}
|
||||
with open(image_path, "rb") as f:
|
||||
return self.extract_from_bytes(f.read())
|
||||
|
||||
async def extract_from_url(self, url: str) -> dict[str, Any]:
|
||||
"""Download image from URL and extract text."""
|
||||
from client import get_client
|
||||
|
||||
client = await get_client()
|
||||
try:
|
||||
resp = await client.get(url, timeout=30)
|
||||
if resp.is_success:
|
||||
return self.extract_from_bytes(resp.content)
|
||||
except Exception as e:
|
||||
return {"success": False, "error": str(e)[:200]}
|
||||
return {"success": False, "error": "Failed to download image"}
|
||||
|
||||
def is_available(self) -> bool:
|
||||
return _has_tesseract
|
||||
Loading…
Add table
Add a link
Reference in a new issue