Comprehensive PII Detection & Anonymization
Intelligent Engine Selection • Lightweight • Production Ready
DataFog is a comprehensive open-source library for detecting and anonymizing personally identifiable information (PII) in unstructured data. Built for production workloads, it delivers intelligent engine selection to handle both structured identifiers and contextual entities across different industries and use cases.
đź§ Intelligent Engine Selection
- Automatically chooses the best detection approach for your data
- Pattern-based engine for structured PII (emails, phones, SSNs, credit cards)
- NLP-based engine for contextual entities (names, organizations, locations)
- Industry-optimized detection across financial, healthcare, legal, and enterprise domains
📦 Lightweight & Modular
- Core package under 2MB (vs 800MB+ alternatives)
- Install only what you need:
datafog[nlp]
,datafog[ocr]
,datafog[all]
- Zero ML model downloads for basic usage
🎯 Production Ready
- Comprehensive PII coverage for diverse enterprise needs
- Battle-tested detection patterns with high precision
- Comprehensive test suite with 99.4% coverage
- CLI tools and Python SDK for any workflow
đź”§ Developer Friendly
- Simple API:
detect("Contact john@example.com")
- Multiple anonymization methods: redact, replace, hash
- OCR support for images and documents
DataFog can be installed via pip:
pip install datafog
Command | Description |
---|---|
scan-text |
Analyze text for PII |
scan-image |
Extract and analyze text from images |
redact-text |
Redact PII in text |
replace-text |
Replace PII with anonymized values |
hash-text |
Hash PII in text |
health |
Check service status |
show-config |
Display current settings |
download-model |
Get a specific spaCy model |
list-spacy-models |
Show available models |
list-entities |
View supported PII entities |
To scan and annotate text for PII entities:
datafog scan-text "Your text here"
Example:
datafog scan-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
To extract text from images and optionally perform PII annotation:
datafog scan-image "path/to/image.png" --operations extract
Example:
datafog scan-image "nokia-statement.png" --operations extract
To extract text and annotate PII:
datafog scan-image "nokia-statement.png" --operations scan
To redact PII in text:
datafog redact-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
which should output:
[REDACTED] is the CEO of [REDACTED] and is based out of [REDACTED], [REDACTED]
To replace detected PII:
datafog replace-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
which should return something like:
[PERSON_B86CACE6] is the CEO of [UNKNOWN_445944D7] and is based out of [UNKNOWN_32BA5DCA], [UNKNOWN_B7DF4969]
Note: a unique randomly generated identifier is created for each detected entity
You can select from SHA256, SHA3-256, and MD5 hashing algorithms to hash detected PII. Currently the hashed output does not match the length of the original entity, for privacy-preserving purposes. The default is SHA256.
datafog hash-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
generating an output which looks like this:
5738a37f0af81594b8a8fd677e31b5e2cabd6d7791c89b9f0a1c233bb563ae39 is the CEO of f223faa96f22916294922b171a2696d868fd1f9129302eb41a45b2a2ea2ebbfd and is based out of ab5f41f04096cf7cd314357c4be26993eeebc0c094ca668506020017c35b7a9c, cad0535decc38b248b40e7aef9a1cfd91ce386fa5c46f05ea622649e7faf18fb
datafog health
datafog show-config
datafog download-model en_core_web_sm
datafog show-spacy-model-directory en_core_web_sm
datafog list-spacy-models
datafog list-entities
- For
scan-image
andscan-text
commands, use--operations
to specify different operations. Default isscan
. - Process multiple images or text strings in a single command by providing multiple arguments.
- Ensure proper permissions and configuration of the DataFog service before running commands.
đź’ˇ Tip: For more detailed information on each command, use the --help
option, e.g., datafog scan-text --help
.
To use DataFog, you'll need to create a DataFog client with the desired operations. Here's a basic setup:
from datafog import DataFog
# For text annotation
client = DataFog(operations="scan")
# For OCR (Optical Character Recognition)
ocr_client = DataFog(operations="extract")
DataFog now supports multiple annotation engines through the TextService
class. You can choose between different engines for PII detection:
from datafog.services.text_service import TextService
# Use fast engine only (fastest, pattern-based detection)
fast_service = TextService(engine="regex")
# Use spaCy engine only (more comprehensive NLP-based detection)
spacy_service = TextService(engine="spacy")
# Use auto mode (default) - tries fast engine first, falls back to spaCy if no entities found
auto_service = TextService() # engine="auto" is the default
Each engine targets different PII detection needs:
- regex: Pattern-based detection optimized for structured identifiers like emails, phone numbers, credit cards, SSNs, and IP addresses
- spacy: NLP-based entity recognition for contextual entities like names, organizations, locations, dates, and monetary amounts
- auto: Intelligent selection - tries pattern-based detection first, falls back to NLP for comprehensive contextual analysis
Here's an example of how to annotate PII in a text document:
import requests
# Fetch sample medical record
doc_url = "https://gist.githubusercontent.com/sidmohan0/b43b72693226422bac5f083c941ecfdb/raw/b819affb51796204d59987893f89dee18428ed5d/note1.txt"
response = requests.get(doc_url)
text_lines = [line for line in response.text.splitlines() if line.strip()]
# Run annotation
annotations = client.run_text_pipeline_sync(str_list=text_lines)
print(annotations)
For OCR capabilities, you can use the following:
import asyncio
import nest_asyncio
nest_asyncio.apply()
async def run_ocr_pipeline_demo():
image_url = "https://s3.amazonaws.com/thumbnails.venngage.com/template/dc377004-1c2d-49f2-8ddf-d63f11c8d9c2.png"
results = await ocr_client.run_ocr_pipeline(image_urls=[image_url])
print("OCR Pipeline Results:", results)
loop = asyncio.get_event_loop()
loop.run_until_complete(run_ocr_pipeline_demo())
Note: The DataFog library uses asynchronous programming for OCR, so make sure to use the async
/await
syntax when calling the appropriate methods.
DataFog provides various anonymization techniques to protect sensitive information. Here are examples of how to use them:
To redact PII in text:
from datafog import DataFog
from datafog.config import OperationType
client = DataFog(operations=[OperationType.SCAN, OperationType.REDACT])
text = "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
redacted_text = client.run_text_pipeline_sync([text])[0]
print(redacted_text)
Output:
[REDACTED] is the CEO of [REDACTED] and is based out of [REDACTED], [REDACTED]
To replace detected PII with unique identifiers:
from datafog import DataFog
from datafog.config import OperationType
client = DataFog(operations=[OperationType.SCAN, OperationType.REPLACE])
text = "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
replaced_text = client.run_text_pipeline_sync([text])[0]
print(replaced_text)
Output:
[PERSON_B86CACE6] is the CEO of [UNKNOWN_445944D7] and is based out of [UNKNOWN_32BA5DCA], [UNKNOWN_B7DF4969]
To hash detected PII:
from datafog import DataFog
from datafog.config import OperationType
from datafog.models.anonymizer import HashType
client = DataFog(operations=[OperationType.SCAN, OperationType.HASH], hash_type=HashType.SHA256)
text = "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
hashed_text = client.run_text_pipeline_sync([text])[0]
print(hashed_text)
Output:
5738a37f0af81594b8a8fd677e31b5e2cabd6d7791c89b9f0a1c233bb563ae39 is the CEO of f223faa96f22916294922b171a2696d868fd1f9129302eb41a45b2a2ea2ebbfd and is based out of ab5f41f04096cf7cd314357c4be26993eeebc0c094ca668506020017c35b7a9c, cad0535decc38b248b40e7aef9a1cfd91ce386fa5c46f05ea622649e7faf18fb
You can choose from SHA256 (default), SHA3-256, and MD5 hashing algorithms by specifying the hash_type
parameter
DataFog provides multiple annotation engines designed for different PII detection scenarios:
The TextService
class supports three engine modes:
# Use regex engine for structured identifiers
regex_service = TextService(engine="regex")
# Use spaCy engine for contextual entities
spacy_service = TextService(engine="spacy")
# Use auto mode (default) - intelligent engine selection
auto_service = TextService() # engine="auto" is the default
Different engines excel at detecting different types of personally identifiable information:
Engine | PII Types Detected | Best For |
---|---|---|
Regex | EMAIL, PHONE, SSN, CREDIT_CARD, IP_ADDRESS, DOB, ZIP | Financial services, healthcare, compliance |
SpaCy | PERSON, ORG, GPE, CARDINAL, DATE, TIME, MONEY, PRODUCT | Legal documents, communication monitoring, general text |
Auto | All of the above (context-dependent) | Mixed data sources, unknown content types |
Financial Services & Healthcare:
- Primary need: Structured identifiers (SSNs, credit cards, account numbers)
- Recommended:
regex
engine for high precision on regulatory requirements - Common PII: ~60% structured identifiers, ~40% names/addresses
Legal & Document Review:
- Primary need: Names, organizations, locations in unstructured text
- Recommended:
spacy
engine for comprehensive entity recognition - Common PII: ~30% structured identifiers, ~70% contextual entities
Enterprise Communication & Mixed Content:
- Primary need: Both structured and contextual PII detection
- Recommended:
auto
engine for intelligent selection - Benefits from both engines depending on content type
Regex Engine: Choose when you need to detect specific, well-formatted identifiers:
- Processing structured databases or forms
- Compliance scanning for specific regulatory requirements (GDPR, HIPAA)
- High-volume processing where deterministic results are important
- Financial data with credit cards, SSNs, account numbers
SpaCy Engine: Choose when you need contextual understanding:
- Analyzing unstructured documents, emails, or communications
- Legal eDiscovery where names and organizations are key
- Content where entities don't follow standard patterns
- Multi-language support requirements
Auto Engine: Choose for general-purpose PII detection:
- Unknown or mixed content types
- Applications serving multiple industries
- When you want comprehensive coverage without manual engine selection
- Default choice for most production applications
You can test the different engines locally using pytest:
pip install pytest-benchmark
pytest tests/benchmark_text_service.py -v
For more detailed examples, check out our Jupyter notebooks in the examples/
directory:
text_annotation_example.ipynb
: Demonstrates text PII annotationimage_processing.ipynb
: Shows OCR capabilities and text extraction from images
These notebooks provide step-by-step guides on how to use DataFog for various tasks.
For local development:
- Clone the repository.
- Navigate to the project directory:
cd datafog-python
- Create a new virtual environment (using
.venv
is recommended as it is hardcoded in the justfile):python -m venv .venv
- Activate the virtual environment:
- On Windows:
.venv\Scripts\activate
- On macOS/Linux:
source .venv/bin/activate
- On Windows:
- Install the package in editable mode:
pip install -r requirements-dev.txt
- Set up the project:
just setup
Now, you can develop and run the project locally.
- Format the code:
This runs
just format
isort
to sort imports. - Lint the code:
This runs
just lint
flake8
to check for linting errors. - Generate coverage report:
This runs
just coverage-html
pytest
and generates a coverage report in thehtmlcov/
directory.
We use pre-commit to run checks locally before committing changes. Once installed, you can run:
pre-commit run --all-files
For OCR, we use Tesseract, which is incorporated into the build step. You can find the relevant configurations under .github/workflows/
in the following files:
dev-cicd.yml
feature-cicd.yml
main-cicd.yml
- Python 3.10
This software is published under the MIT license.