ai-service/app/utils/celery/worker.py

import sys
from celery import Celery
from app.services.background_services.update_form_data_in_db import update_db
from app.services.background_services.update_preapproval_status_in_db import update_application_status
from app.services.background_services.verify_document import upload_and_verify_documents
from ...config.env import env

sys.path.extend(["./"])

# Initialize Celery
celery_app = Celery('tasks', broker=env.REDIS_URI, backend=env.REDIS_URI)

# Queue
DOC_VERIFY_QUEUE = "doc_verify_queue"
FORM_UPDATE_QUEUE = "form_update_queue"
PRE_APPROVAL_STARTED_QUEUE = "pre_approval_started_queue"

# Define tasks


@celery_app.task(name=FORM_UPDATE_QUEUE)
def form_update_task(property_dict):
    result = update_db(property_dict)
    return {}


@celery_app.task(name=PRE_APPROVAL_STARTED_QUEUE)
def pre_approval_task(application_id):
    result = update_application_status(application_id)
    return {}


@celery_app.task(name=DOC_VERIFY_QUEUE)
def doc_verify_task(doc_dict):
    result = upload_and_verify_documents(doc_dict)
    return {}