add json printing callback

Author: lnotspotl

include/depthai/modelzoo/Zoo.hpp

@@ -101,9 +101,10 @@ std::string getModelFromZoo(const NNModelDescription& modelDescription,
  * DEPTHAI_ZOO_CACHE_PATH environment variable and uses that if set, otherwise the default is used (see getDefaultCachePath).
  * @param apiKey: API key for the model zoo, default is "". If apiKey is set to "", this function checks the DEPTHAI_ZOO_API_KEY environment variable and uses
  * that if set. Otherwise, no API key is used.
+ * @param format: Format to use for output (possible values: pretty, json), default is "pretty"
  * @return bool: True if all models were downloaded successfully, false otherwise
  */
-bool downloadModelsFromZoo(const std::string& path, const std::string& cacheDirectory = "", const std::string& apiKey = "");
+bool downloadModelsFromZoo(const std::string& path, const std::string& cacheDirectory = "", const std::string& apiKey = "", const std::string& format = "pretty");
 
 std::ostream& operator<<(std::ostream& os, const NNModelDescription& modelDescription);
 

src/modelzoo/Zoo.cpp

@@ -332,6 +332,8 @@ nlohmann::json ZooManager::fetchModelDownloadLinks() {
         throw std::runtime_error(generateErrorMessageHub(response));
     }
 
+    std::cout << "Response: " << response.text << std::endl;
+
     // Extract download links from response
     nlohmann::json responseJson = nlohmann::json::parse(response.text);
     return responseJson;
@@ -582,7 +584,7 @@ struct JsonDownloadProgressManager {
     }
 };
 
-bool downloadModelsFromZoo(const std::string& path, const std::string& cacheDirectory, const std::string& apiKey) {
+bool downloadModelsFromZoo(const std::string& path, const std::string& cacheDirectory, const std::string& apiKey, const std::string& format) {
     logger::info("Downloading models from zoo");
     // Make sure 'path' exists
     if(!std::filesystem::exists(path)) throw std::runtime_error("Path does not exist: " + path);

src/modelzoo/zoo_helper.cpp

@@ -33,6 +33,9 @@ int main(int argc, char* argv[]) {
     const std::string DEFAULT_DOWNLOAD_ENDPOINT = dai::modelzoo::getDownloadEndpoint();
     program.add_argument("--download_endpoint").default_value(DEFAULT_DOWNLOAD_ENDPOINT).help("Endpoint to use for downloading models");
 
+    const std::string FORMAT_DEFAULT = "pretty";
+    program.add_argument("--format").default_value(FORMAT_DEFAULT).help("Format to use for output (possible values: pretty, json)");
+
     program.add_argument("--verbose").default_value(false).implicit_value(true).help("Verbose output");
 
     // Parse arguments
@@ -50,9 +53,10 @@ int main(int argc, char* argv[]) {
     auto apiKey = program.get<std::string>("--api_key");
     auto healthEndpoint = program.get<std::string>("--health_endpoint");
     auto downloadEndpoint = program.get<std::string>("--download_endpoint");
+    auto format = program.get<std::string>("--format");
 
     bool verbose = program.get<bool>("--verbose");
-    if(!dai::utility::isEnvSet("DEPTHAI_LEVEL") && verbose) {
+    if(!dai::utility::isEnvSet("DEPTHAI_LEVEL") && verbose && format == "pretty") {
         dai::Logging::getInstance().logger.set_level(spdlog::level::info);
     }
 
@@ -61,10 +65,12 @@ int main(int argc, char* argv[]) {
     dai::modelzoo::setDownloadEndpoint(downloadEndpoint);
 
     // Print arguments
-    std::cout << "Downloading models defined in yaml files in folder: " << yamlFolder << std::endl;
-    std::cout << "Downloading models into cache folder: " << cacheFolder << std::endl;
-    if(!apiKey.empty()) {
-        std::cout << "Using API key: " << apiKey << std::endl;
+    if(format == "pretty") {
+        std::cout << "Downloading models defined in yaml files in folder: " << yamlFolder << std::endl;
+        std::cout << "Downloading models into cache folder: " << cacheFolder << std::endl;
+        if(!apiKey.empty()) {
+            std::cout << "Using API key: " << apiKey << std::endl;
+        }
     }
 
     // Download models
@@ -74,6 +80,8 @@ int main(int argc, char* argv[]) {
         return EXIT_FAILURE;
     }
 
-    std::cout << "Successfully downloaded all models from " << yamlFolder << std::endl;
+    if(format == "pretty") {
+        std::cout << "Successfully downloaded all models from " << yamlFolder << std::endl;
+    }
     return EXIT_SUCCESS;
 }