from modules.ai_model import run_inference  # Import the model inference function

def detect_faults(video_frame):
    # Process the frame (convert to suitable format if needed)
    outputs = run_inference(video_frame)

    # Post-process output (bounding boxes, fault detection)
    faults = process_model_output(outputs)
    return faults

def process_model_output(outputs):
    # Example: extract faults based on AI output
    faults = []
    if "hotspot" in outputs:
        faults.append("Hotspot Detected")
    if "shading" in outputs:
        faults.append("Shading Detected")
    return faults