miralabs-backend/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 app.services.background_services.update_eligibility_status_in_db import update_eligibility

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"
ELIGIBILITY_QUEUE = "update_eligibility"


# 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 {}

@celery_app.task(name=ELIGIBILITY_QUEUE)
def eligible_task(application_id, status):
    result = update_eligibility(application_id,status)
    return {}