models/property_relation.py

# models/property_relation.py

from .model_loader import load_model
from .logging_config import logger

def check_if_property_related(text):
    try:
        classifier = load_model("zero-shot-classification", "typeform/mobilebert-uncased-mnli")
        result = classifier(text[:1000], ["property-related", "non-property-related"])
        is_related = result['labels'][0] == "property-related"
        return {
            'is_related': is_related,
            'confidence': float(result['scores'][0])
        }
    except Exception as e:
        logger.error(f"Error checking property relation: {str(e)}")
        return {
            'is_related': False,
            'confidence': 0.0
        }