Christian Krakau-Louis
28 lines
818 B
Python
28 lines
818 B
Python
#!/usr/bin/env python3
|
|
|
|
import boto3
|
|
from .celery_worker import celery
|
|
from .config import settings
|
|
|
|
textract_client = boto3.client(
|
|
"textract",
|
|
aws_access_key_id=settings.aws_access_key_id,
|
|
aws_secret_access_key=settings.aws_secret_access_key,
|
|
region_name=settings.aws_region,
|
|
)
|
|
|
|
@celery.task
|
|
def process_document(file_path: str):
|
|
"""Process document: OCR, metadata extraction, upload"""
|
|
|
|
# Send to AWS Textract
|
|
with open(file_path, "rb") as document:
|
|
response = textract_client.analyze_document(
|
|
Document={"Bytes": document.read()},
|
|
FeatureTypes=["TABLES", "FORMS"]
|
|
)
|
|
|
|
extracted_text = " ".join([block["Text"] for block in response["Blocks"] if block["BlockType"] == "WORD"])
|
|
|
|
return {"file": file_path, "text": extracted_text}
|