services/thermal_service.py

from ultralytics import YOLO

thermal_model = YOLO("thermal_model.pt")

def detect_thermal_anomalies(image_path):
    results = thermal_model(image_path)
    flagged = []
    for r in results:
        if hasattr(r, 'temperature') and r.temperature > 75:
            flagged.append(r)
    return flagged