#ifndef AAPGSMODELCLASSIFIER_H
#define AAPGSMODELCLASSIFIER_H

#include <memory>

#include <QImage>
#include <QString>

class AapgsModelClassifier
{
public:
    struct Classification
    {
        QString modelType;
        double confidence = 0.0;
    };

    AapgsModelClassifier();
    ~AapgsModelClassifier();

    AapgsModelClassifier(const AapgsModelClassifier&) = delete;
    AapgsModelClassifier& operator=(const AapgsModelClassifier&) = delete;

    bool Classify(const QImage& frame,
                  Classification& result,
                  QString& errorMessage);

private:
    class Impl;
    std::unique_ptr<Impl> m_impl;
};

#endif // AAPGSMODELCLASSIFIER_H