Create tasks.py

This commit is contained in:
Christian Krakau-Louis
2025-02-09 16:22:39 +01:00
committed by GitHub
parent 527c6c86ff
commit 22f8f60f12
+27
View File
@@ -0,0 +1,27 @@
#!/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}