src/models/mock_ensemble.py

import numpy as np

class EnsembleDetector:
    def __init__(self):
        pass
    
    def predict(self, image):
        # Mock prediction for setup without model weights
        return [
            {
                "box": [100, 100, 200, 200],
                "class_name": "pothole",
                "class_id": 0,
                "confidence": 0.85,
                "votes": 3
            },
            {
                "box": [300, 150, 400, 250],
                "class_name": "crack",
                "class_id": 1,
                "confidence": 0.92,
                "votes": 2
            }
        ]

class SeverityClassifier:
    def __init__(self):
        pass
        
    def classify(self, area_ratio, confidence):
        if area_ratio > 0.1:
            return "heavy"
        elif area_ratio > 0.05:
            return "medium"
        return "light"

class TTAProcessor:
    def __init__(self):
        pass
        
    def predict(self, image):
        # Mock TTA
        return []