from celery import Celery
from app.core.config import settings
from app.handlers.toxicity_handler import ToxicityHandler

# Инициализация Celery
celery_app = Celery(
    'tasks',
    broker=settings.CELERY_BROKER_URL,
    backend=settings.CELERY_RESULT_BACKEND
)

# Конфигурация Celery
celery_app.conf.update(
    task_serializer='json',
    result_serializer='json',
    accept_content=['json'],
    timezone='UTC',
    enable_utc=True,
)

# Инициализация обработчика токсичности
toxicity_handler = ToxicityHandler()

@celery_app.task
def assess_toxicity_task(text: str) -> float:
    """
    Задача для оценки токсичности текста.
    
    Args:
        text (str): Входной текст.
    
    Returns:
        float: Оценка токсичности.
    """
    return toxicity_handler.text_to_toxicity(text)