CscDetectorAiAPI.h

Go to the documentation of this file.

#ifndef CscDetectorAiAPI_h
#define CscDetectorAiAPI_h
 
#include "CoreObjects/CscWorldObject.h"
#include "CscCommon.h"
#include "DetectorEngine/CscDetectorEngine.h"
#include <iostream>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/objdetect.hpp>
#include <vector>
 
using std::string;
 
namespace conscience_core::detector_engine {
 
class CscDetectorAiAPI : public CscDetectorEngine {
public:
    static constexpr const char *DETECTOR_PARAM_KEY_WEBHOOKPATH = "webhookPath";
    static constexpr const char *DETECTOR_PARAM_KEY_IMAGE_DESCRIPTION_MODEL = "imageDescriptionModel";
    static constexpr const char *DETECTOR_PARAM_KEY_IMAGE_DESCRIPTION_PROMPT = "imageDescriptionPrompt";
    static constexpr const char *DETECTOR_PARAM_KEY_QUESTIONS_ABOUT_DESCRIPTION_MODEL = "questionsAboutDescriptionModel";
    static constexpr const char *DETECTOR_PARAM_KEY_QUESTIONS_ABOUT_DESCRIPTION_PROMPT = "questionsAboutDescriptionPrompt";
    static constexpr const char *DETECTOR_PARAM_KEY_INTERVAL_TO_CHECK_IMAGE_IN_MILLIS = "intervalToCheckImageInMillis";
    CscDetectorAiAPI(optional<string> aiApiWebhookPath, const map<string, string> &additionWebhookParams);
    ptr<DetectionResult> detectOnImage(const DetectorSourceImage &image, ptr<DetectionParameters> parameters) override;
    virtual bool equals(CscDetectorEngine *) const override;
 
private:
    const int DEFAULT_INTERVAL_TO_CHECK_IMAGE_IN_MILLIS = 80000;
    int intervalToCheckImageInMillis;
    bool callInProgress = false;
    optional<string> aiApiWebhookPath;
    map<string, string> additionWebhookParams;
    unique_ptr<CscLogger> logger = CscLogger::getForCategory("CscDetectorAiAPI");
    optional<unsigned long long> lastTimeCheck = {};
 
    CscWorldObject *createObjectForDetectionResult(int modelId, string modelName);
 
protected:
    std::string extractCleanJson(const std::string &rawInput);
};
 
}
 
#endif /* CscDetectorAiAPI_hpp */