pryscraper/ocr_extractor.py
cryptorugmunch bb77eb5f35 chore(license): re-license to dual MIT (core) + BSL 1.1 (stealth)
Re-license Pry from full Proprietary to a dual-license model:

- Core engine, extraction, templates (80+), MCP server, x402 payment rail,
  CLI, SDK, browser extension, WordPress plugin, Shopify app, and
  llm_providers: MIT (see LICENSE)
- Anti-detection / stealth subset (15 files): BSL 1.1 with Change Date
  2029-01-01 (see LICENSE-BSL-STEALTH)

BSL files (anti-detection moat):
  ultimate_scraper.py, stealth_engine.py, stealth_scripts/*.js (6),
  camoufox_integration.py, tls_fingerprint.py, cookie_warmer.py,
  behavioral_biometrics.py, adaptive.py, browser_pool.py, network.py,
  captcha_solver.py, shadow_dom.py, lazy_load.py, signup_automator.py,
  auth_connector.py

This enables community contributions to the core engine (templates,
integrations, MCP tools) while protecting the anti-detection techniques
that constitute the actual competitive moat. BSL Additional Use Grant
permits free non-production use; production deployment requires a
commercial license from enterprise@rugmunch.io.

Changes:
- Replace proprietary LICENSE with MIT LICENSE + new LICENSE-BSL-STEALTH
- Add SPDX-License-Identifier headers to 300+ source files
- Add docs/adr/0002-dual-licensing.md (ADR documenting the decision)
- Update README.md: new License section with BSL Additional Use Grant
- Update LICENSING_PRICING_STRATEGY.md: Section 3 (PryScraper) for dual license
- Update AGENTS.md: license line in header + new rule 8 (PRs touching BSL rejected)
- Update pyproject.toml: license = "MIT AND BSL-1.1" + classifiers + license-files
- Update DECISIONS.md index with ADR-0002
- Update STATUS.md (2026-07-03) and PLAN.md sprint goals

Refs: ADR-0002
2026-07-02 19:49:21 +02:00

83 lines
2.9 KiB
Python

"""Pry — Image OCR using Tesseract.
Extract text from images on web pages. Uses pytesseract + Pillow."""
# SPDX-License-Identifier: MIT
# Copyright (c) 2026 Rug Munch Media LLC
#
# Part of Pry — https://git.rugmunch.io/RugMunchMedia/pryscraper
# Licensed under MIT. See LICENSE.
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