from remote_command cimport RemoteCommand
from annotation cimport Annotation
from ai_config cimport AIRecognitionConfig

cdef class Inference:
    cdef object model
    cdef object on_annotation
    cdef Annotation _previous_annotation
    cdef AIRecognitionConfig ai_config
    cdef bint stop_signal

    cdef bint is_video(self, str filepath)
    cdef run_inference(self, RemoteCommand cmd, int batch_size=?)
    cdef _process_video(self, RemoteCommand cmd, int batch_size)
    cdef _process_image(self, RemoteCommand cmd)
    cdef stop(self)

    cdef frame_to_annotation(self, long time, frame, boxes: object)
    cdef bint is_valid_annotation(self, Annotation annotation)